[mxfp8 moe training] fix kernel test for per group blocked format conversion

[danielvegamyhre]

Stacked PRs:

• ->[mxfp8 moe training] fix kernel test for per group blocked format conversion #3008
• [mxfp8 moe training] wrap 3d quantize tensor in custom ops and integrate it #3004
• [mxfp8 moe training] add CUDA kernel to quantize 3d tensor colwise #3002
• [moe training] add benchmarks for dsv3 236b, 671b shapes; reorganize benchmarks dir #2999
• [mxfp8 moe training] use dim1 cast cuda kernel for 3d weights by reshaping to 2d #2998
• [mxfp8 moe training] add compile support #2990

---

[mxfp8 moe training] fix kernel test for per group blocked format conversion

Summary

• e2e training unit tests confirm numerics are good with new "upper bound" based padding strategy, but individual kernel unit test still needs to be updated.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3008

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.