trunk/36739e0856ebe768e44f9d411f535d68f51e38a5: [ROCm][Inductor][UT] Enable the FP8 scaled_mm swizzle v2 tests (#198600)
- PyTorch: 1502 events in the last 90 days
- PyTorch: 1483th Release in the last 90 days
- Previous: earlier the same day · ciflow/torchtitan/198697
What happened
Swizzled MX scales are not handled by any inductor template, so test_scaled_mm_v2_swizzle_compile checks that compile falls back to the ATen _scaled_mm_v2 kernel. That path was skipped on ROCm because SWIZZLE_32_4_4 has no hipBLASLt scale mode. gfx950 on ROCm 7.14 packs MX FP8 scales as SWIZZLE_32_8, so drop skipIfRocm and pick the layout with rocm_mx_swizzle. Older ROCm and non-gfx950 still skip. Test Plan: python…
Summary assembled by rule from the sources below