Skip to content

Commit 7a13eb0

Browse files
[mxfp8] skip mxfp8 test on rocm (#2681)
* skip mxfp8 test on rocm * lint
1 parent 0fb99a6 commit 7a13eb0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/prototype/moe_training/test_scaled_grouped_mm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,7 @@ def test_emulate_mxfp8_grouped_gemm_2d_2d(M, N, num_experts):
304304
assert sqnr >= min_sqnr, f"sqnr {sqnr} is too low, must be >= {min_sqnr}"
305305

306306

307+
@skip_if_rocm("ROCm not supported")
307308
@pytest.mark.parametrize("M,K,N", [(1024, 1024, 1024), (1024, 2048, 4096)])
308309
@pytest.mark.parametrize("num_experts", (1, 8, 16))
309310
def test_mxfp8_grouped_gemm_with_dq_fwd_bwd(M, K, N, num_experts):

0 commit comments

Comments
 (0)