Skip to content

Commit c4a6228

Browse files
[TEST][test_matmul::test_mxfp] Skip tests failing due to elements mismatch (#3837)
#3677 (comment)
1 parent 75473cf commit c4a6228

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

python/test/unit/language/test_matmul.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -339,7 +339,12 @@ def fp8e8m0_to_float32(scale):
339339
@pytest.mark.parametrize("nonKDim", ([0, 16, 32] if is_hip_cdna() else [0]))
340340
def test_mxfp(M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, NUM_STAGES, nonKDim, NUM_WARPS, device):
341341
if is_xpu():
342-
if (BLOCK_M, BLOCK_N, BLOCK_K) in {(128, 128, 64), (128, 64, 128)}:
342+
if (nonKDim == 0 and NUM_WARPS == 4 and (M, N, K, BLOCK_M, BLOCK_N, BLOCK_K) in {
343+
(1024, 512, 256, 128, 64, 128),
344+
(1024, 512, 256, 128, 128, 64),
345+
(128, 256, 256, 128, 128, 64),
346+
(128, 128, 128, 128, 128, 64),
347+
}):
343348
pytest.skip("https://github.com/intel/intel-xpu-backend-for-triton/issues/3677")
344349
elif (BLOCK_M, BLOCK_N, BLOCK_K) == (128, 256, 256) and \
345350
triton.runtime.driver.active.utils.get_device_properties(

0 commit comments

Comments
 (0)