diff --git a/tests/comm/test_trtllm_alltoall.py b/tests/comm/test_trtllm_alltoall.py index 17d236bba6..d275d0de25 100644 --- a/tests/comm/test_trtllm_alltoall.py +++ b/tests/comm/test_trtllm_alltoall.py @@ -18,6 +18,14 @@ import torch import flashinfer.comm.trtllm_alltoall as tllm_alltoall +from flashinfer.utils import get_compute_capability + +# Skip all tests on SM110 (Thor) devices - these tests hang indefinitely on this architecture +pytestmark = pytest.mark.skipif( + torch.cuda.is_available() + and get_compute_capability(torch.device("cuda:0"))[0] == 11, + reason="Tests hang indefinitely on SM110 (Thor) devices", +) has_setup_max_sm_count = False