Skip to content

Commit 377ae7f

Browse files
authored
[rocm7.0_internal_testing] mx fpx: Skip test_blockwise_mxfloatx (#2439)
Issue is tracked by SWDEV-535267 This PR skips two test methods (test_blockwise_mxfp8_nvfp4_error_messages and test_blockwise_nvfp4_compile) when running on ROCm platform to address an issue tracked by SWDEV-535267. - Adds @skipIfRocm decorators to skip tests that are failing on ROCm 7.0 internal testing - Targets specific matrix multiplication tests related to MX float formats Signed-off-by: Jagadish Krishnamoorthy <jagadish.krishnamoorthy@amd.com>
1 parent 8e49451 commit 377ae7f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/test_matmul_cuda.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1385,6 +1385,7 @@ def test_blockwise_mxfp8_nvfp4_mxfp4_numerics(self, test_case_name, fast_accum,
13851385
sqnr = compute_error(C_ref, C)
13861386
assert sqnr.item() > approx_match_sqnr_target
13871387

1388+
@skipIfRocm
13881389
@unittest.skipIf(not PLATFORM_SUPPORTS_FP8 or IS_WINDOWS, f8_msg)
13891390
@parametrize("recipe", ["mxfp8", "nvfp4"])
13901391
def test_blockwise_mxfp8_nvfp4_error_messages(self, device, recipe) -> None:
@@ -1618,6 +1619,7 @@ def test_blockwise_mxfp8_compile(self) -> None:
16181619
)
16191620
torch.testing.assert_close(C, C_ref, atol=0, rtol=0)
16201621

1622+
@skipIfRocm
16211623
@unittest.skipIf(not PLATFORM_SUPPORTS_MX_GEMM, mx_skip_msg)
16221624
def test_blockwise_nvfp4_compile(self) -> None:
16231625

0 commit comments

Comments
 (0)