Skip to content

Commit a012080

Browse files
authored
[flaky test] Skip bf16xint16 GEMM test under Pallas CPU interpret (#2683)
1 parent bf24032 commit a012080

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

test/test_examples.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from helion._testing import skipIfFn
2929
from helion._testing import skipIfNotCUDA
3030
from helion._testing import skipIfPallas
31+
from helion._testing import skipIfPallasInterpret
3132
from helion._testing import skipIfRefEager
3233
from helion._testing import skipIfRocm
3334
from helion._testing import skipIfTileIR
@@ -666,7 +667,11 @@ def test_low_mem_dropout(self):
666667
block_sizes=[block_size],
667668
)
668669

669-
@xfailIfPallas("precision differences with bf16xint16 operations on pallas")
670+
@skipIfPallasInterpret(
671+
"65536x1024x1280 GEMM is too slow under CPU interpret -- it exceeds the "
672+
"300s per-test timeout and (thread timeout method) kills the whole job"
673+
)
674+
@xfailIfPallasTpu("precision differences with bf16xint16 operations on pallas")
670675
@skipIfTileIR("precision differences with bf16xint16 operations on tileir")
671676
@skipIfRocm("precision differences with bf16xint16 operations on rocm")
672677
@skipIfXPU("precision differences with bf16xint16 operations on xpu")

0 commit comments

Comments
 (0)