Skip to content

Commit 2b06b2c

Browse files
authored
[TEST] Update test_load_scope_sem_coop_grid_cta_not_one to be skipped if no NV GPUs on system (#5538)
1 parent 4d7dafd commit 2b06b2c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/language/test_core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1654,7 +1654,7 @@ def change_value(X, BLOCK_SIZE: tl.constexpr, sem: tl.constexpr):
16541654

16551655

16561656
@pytest.mark.interpreter
1657-
@pytest.mark.skipif(torch.cuda.get_device_capability()[0] < 9 or is_hip(),
1657+
@pytest.mark.skipif(not is_cuda() or torch.cuda.get_device_capability()[0] < 9,
16581658
reason="Requires compute capability >= 9 for NV")
16591659
def test_load_scope_sem_coop_grid_cta_not_one(device):
16601660

0 commit comments

Comments
 (0)