diff --git a/test/test_matmul_cuda.py b/test/test_matmul_cuda.py index 19a09ec96e5be..6bdb4fdc6e0bc 100644 --- a/test/test_matmul_cuda.py +++ b/test/test_matmul_cuda.py @@ -1382,6 +1382,7 @@ def test_blockwise_mxfp8_nvfp4_mxfp4_numerics(self, test_case_name, fast_accum, sqnr = compute_error(C_ref, C) assert sqnr.item() > approx_match_sqnr_target + @skipIfRocm @unittest.skipIf(not PLATFORM_SUPPORTS_FP8 or IS_WINDOWS, f8_msg) @parametrize("recipe", ["mxfp8", "nvfp4"]) def test_blockwise_mxfp8_nvfp4_error_messages(self, device, recipe) -> None: @@ -1615,6 +1616,7 @@ def test_blockwise_mxfp8_compile(self) -> None: ) torch.testing.assert_close(C, C_ref, atol=0, rtol=0) + @skipIfRocm @unittest.skipIf(not PLATFORM_SUPPORTS_MX_GEMM, mx_skip_msg) def test_blockwise_nvfp4_compile(self) -> None: