Skip to content

Commit b1f5c66

Browse files
authored
[AMD] Fix typo in disabling AMD tests (#7178)
Actually calls `is_hip()` so the tests are properly skipped on AMD.
1 parent ba94c21 commit b1f5c66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/triton_kernels/tests/test_routing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def init_data(n_tokens, n_expts_tot, dtype=torch.float32, device="cuda"):
1919
@pytest.mark.parametrize("n_expts_tot, n_expts_act", [(128, 32), (1500, 8)])
2020
@pytest.mark.parametrize("use_expt_indx", [False, True])
2121
@pytest.mark.parametrize("sm_first", [True, False])
22-
@pytest.mark.skipif(is_hip, reason="Tests are currently broken on AMD")
22+
@pytest.mark.skipif(is_hip(), reason="Tests are currently broken on AMD")
2323
def test_op(n_tokens_pad, n_tokens_raw, n_expts_tot, n_expts_act, sm_first, use_expt_indx, device):
2424
torch.manual_seed(2)
2525
if n_tokens_raw is None:

0 commit comments

Comments
 (0)