[mxfp8 moe training] wrap 3d quantize tensor in custom ops and integrate it

[danielvegamyhre]

Stacked PRs:

• ->[mxfp8 moe training] wrap 3d quantize tensor in custom ops and integrate it #3003
• [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] wrap 3d quantize tensor in custom ops and integrate it

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 1 New Failure

As of commit a665638 with merge base 66384a9 ():

NEW FAILURE - The following job has failed:

• PR Label Check / Check PR Labels (gh)
  Process completed with exit code 1.

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