Direction for extending MXGemmKernelChoice/MXLinearRecipeName for XPU support #3457
Unanswered
slabhs-aws
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hi team,
TorchAO today provides 3 MXGemm kernel paths:
For users running MX workloads on XPUs, it’s unclear how future MX kernel selection is intended to evolve.
Questions:
MXGemmKernelChoice, or is there a different path to onboard new XPU architectures?MXLinearRecipeNameThis context affects how upstream frameworks (e.g., torchtitan) gate MX execution.
Cross-referencing with the related torchtitan discussion: pytorch/torchtitan#2120
Thank you!
Beta Was this translation helpful? Give feedback.
All reactions