Skip to content

Commit 843992c

Browse files
committed
Skip mxfp on XPU architectures without subgroup mma
1 parent 22ac34f commit 843992c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/test/unit/language/test_matmul.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -316,6 +316,8 @@ def fp8e8m0_to_float32(scale):
316316
@pytest.mark.parametrize("NUM_STAGES", [1, 3])
317317
@pytest.mark.parametrize("NUM_WARPS", [4, 8])
318318
@pytest.mark.parametrize("nonKDim", ([0, 16, 32] if is_hip_cdna() else [0]))
319+
@pytest.mark.skipif(is_xpu() and not torch.xpu.get_device_capability()['has_subgroup_matrix_multiply_accumulate'],
320+
reason="MXFP requires subgroup mma for XPU backend")
319321
def test_mxfp(M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, NUM_STAGES, nonKDim, NUM_WARPS, device):
320322
if is_xpu():
321323
if (BLOCK_M, BLOCK_N, BLOCK_K) in {(128, 128, 64), (128, 64, 128)}:

0 commit comments

Comments
 (0)