Fix wrong register constraints produced for some types in WGMMAWaitGroupOpPattern
Fix wrong register constraints produced for some types in WGMMAWaitGroupOpPattern

This is a minimal patch to unblock folks. I'll work on polishing it up and upstreaming a proper fix to OpenAI.
