[mxfp8 moe training] fix torch ref impl of SF blocked layout per group along K

[danielvegamyhre]

Stacked PRs:

• [mxfp8 moe training] add bench script for scale conversion to blocked format kernels for groups along K #3604
• ->[mxfp8 moe training] fix torch ref impl of SF blocked layout per group along K #3603

---

[mxfp8 moe training] fix torch ref impl of SF blocked layout per group along K

Part of #3599

[pytorch-bot]

🔗 Helpful Links

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

• 📄 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, 1 Cancelled Job

As of commit da38274 with merge base 3955b6c ():

NEW FAILURE - The following job has failed:

• Run Regression Tests / test-nightly (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/wh... / linux-job (gh)
  test/quantization/pt2e/test_x86inductor_fusion.py::TestPatternMatcher::test_qconv2d_silu_int8_mixed_bf16_cpu

CANCELLED JOB - The following job was cancelled. Please retry:

• Run 1xH100 Tests (gh)

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