Skip to content

Commit 091c2e4

Browse files
committed
addressing L2 CI errors
1 parent 88315b5 commit 091c2e4

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/py/dynamo/distributed/test_nccl_ops.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
)
2727
from parameterized import parameterized
2828
from torch.testing._internal.common_utils import run_tests
29+
from torch_tensorrt._features import ENABLED_FEATURES
2930

3031

3132
def is_distributed_nccl_available():
@@ -90,11 +91,15 @@ def forward(self, x):
9091

9192
class TestNcclOpsConverter(DispatchTestCase):
9293
# 1. Skip if NCCL backend is not available (e.g., Windows, Jetson) - hard requirement
93-
# 2. Don't skip if TRTLLM is unavailable (e.g., CUDA 13) - falls back to PyTorch
94+
# 2. Skip if TRTLLM is unavailable (e.g., CUDA 13) - no converters registered
9495
@unittest.skipIf(
9596
not is_distributed_nccl_available(),
9697
"Skipped: NCCL backend is not available (Windows/Jetson Orin not supported).",
9798
)
99+
@unittest.skipIf(
100+
not ENABLED_FEATURES.trtllm_for_nccl,
101+
"Skipped: TensorRT-LLM plugin for NCCL is not available (e.g., CUDA 13).",
102+
)
98103
@classmethod
99104
def setUpClass(cls):
100105
cls.world_size = int(os.environ.get("OMPI_COMM_WORLD_SIZE", 1))

0 commit comments

Comments
 (0)