fix mx triton kernel after PyTorch triton pin change#431

msaroufim merged 1 commit intomainpytorch/ao:mainfrom

20240624_mx_bugfixpytorch/ao:20240624_mx_bugfixCopy head branch name to clipboard