We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2892466 commit 886a8feCopy full SHA for 886a8fe
python/triton_kernels/tests/test_matmul.py
@@ -445,6 +445,8 @@ def round_x(x, idx):
445
446
447
def test_set_idle_sms():
448
+ if not is_cuda():
449
+ pytest.skip("Only supported on CUDA")
450
from triton_kernels.matmul_ogs_details.opt_flags import make_opt_flags
451
num_idle_sms = 24
452
matmul_ogs_set_idle_sms(num_idle_sms)
0 commit comments