Skip to content

Commit 886a8fe

Browse files
authored
[KERNELS] Skip idle_sms on AMD (#7217)
As it's not currently supported
1 parent 2892466 commit 886a8fe

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/triton_kernels/tests/test_matmul.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -445,6 +445,8 @@ def round_x(x, idx):
445445

446446

447447
def test_set_idle_sms():
448+
if not is_cuda():
449+
pytest.skip("Only supported on CUDA")
448450
from triton_kernels.matmul_ogs_details.opt_flags import make_opt_flags
449451
num_idle_sms = 24
450452
matmul_ogs_set_idle_sms(num_idle_sms)

0 commit comments

Comments
 (0)