Skip to content

Commit f9e8646

Browse files
authored
[rocm7.0_internal_testing] unskip rowwise tests (#2567)
Signed-off-by: Jagadish Krishnamoorthy <[email protected]>
1 parent 41814a3 commit f9e8646

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

test/test_matmul_cuda.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@
4646
parametrize,
4747
run_tests,
4848
skipIfRocm,
49-
skipIfRocmVersionAndArch,
5049
skipIfRocmVersionLessThan,
5150
TEST_CUDA,
5251
TEST_WITH_ROCM,
@@ -908,7 +907,6 @@ def test_float8_scale_fast_accum(self, device) -> None:
908907
out_fp8_s = torch._scaled_mm(x, y, scale_a=scale_a, scale_b=scale_b, use_fast_accum=True)
909908
self.assertEqual(out_fp8, out_fp8_s)
910909

911-
@skipIfRocmVersionAndArch((7, 1), "gfx950")
912910
@unittest.skipIf(not PLATFORM_SUPPORTS_FP8 or IS_WINDOWS, f8_msg)
913911
@unittest.skipIf(not SM89OrLater, "rowwise implementation is currently sm89+ specific")
914912
@parametrize("use_fast_accum", [True, False])
@@ -1014,7 +1012,6 @@ def test_float8_error_messages(self, device) -> None:
10141012
out_dtype=torch.bfloat16,
10151013
)
10161014

1017-
@skipIfRocmVersionAndArch((7, 1), "gfx950")
10181015
@unittest.skipIf(not PLATFORM_SUPPORTS_FP8 or IS_WINDOWS, f8_msg)
10191016
@unittest.skipIf(not SM89OrLater, "rowwise implementation is currently sm89+ specific")
10201017
@parametrize("base_dtype", [torch.bfloat16])

0 commit comments

Comments
 (0)