Skip to content

Commit 3115036

Browse files
committed
Skip test_moe_matmul_ogs on older cards
1 parent c8d863f commit 3115036

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
@@ -1429,6 +1429,10 @@ def _jagged_dense_add_2d_make_precompiler(x_data: torch.Tensor, x_offsets: torch
14291429
return make_precompiler(_jagged_dense_add_2d_kernel)(x_offsets, x_data, y, out, out.size(0), out.size(1), x_offsets.size(0), y.size(0), y.size(1), out.stride(0), out.stride(1), x_data.stride(0), x_offsets.stride(0), y.stride(0), y.stride(1), _BLOCK_SIZE_1, _BLOCK_SIZE_2, num_warps=8, num_stages=4)""",
14301430
)
14311431

1432+
@unittest.skipIf(
1433+
torch.cuda.get_device_capability(0) < (9, 0),
1434+
"Triton internal error on RTX 3090",
1435+
)
14321436
def test_moe_matmul_ogs(self):
14331437
mod = import_path(examples_dir / "moe_matmul_ogs.py")
14341438

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

0 commit comments

Comments
 (0)