Skip to content

Commit 53506b5

Browse files
committed
Skip test_moe_matmul_ogs on older cards
stack-info: PR: #121, branch: jansel/stack/18
1 parent 13e6e7e commit 53506b5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

test/test_examples.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1430,6 +1430,10 @@ def _jagged_dense_add_2d_make_precompiler(x_data: torch.Tensor, x_offsets: torch
14301430
)
14311431

14321432
@unittest.skip("TODO(yf225): fix occasional numerical error")
1433+
@unittest.skipIf(
1434+
torch.cuda.get_device_capability(0) < (9, 0),
1435+
"Triton internal error on RTX 3090",
1436+
)
14331437
def test_moe_matmul_ogs(self):
14341438
mod = import_path(examples_dir / "moe_matmul_ogs.py")
14351439

@@ -1451,7 +1455,6 @@ def test_moe_matmul_ogs(self):
14511455
helion_kernel_args,
14521456
mod.moe_matmul_ogs_reference(*args),
14531457
block_sizes=[[16, 16], 16],
1454-
l2_grouping=4,
14551458
),
14561459
"""\
14571460
from __future__ import annotations

0 commit comments

Comments
 (0)