File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
tests/py/dynamo/distributed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change 2626)
2727from parameterized import parameterized
2828from torch .testing ._internal .common_utils import run_tests
29+ from torch_tensorrt ._features import ENABLED_FEATURES
2930
3031
3132def is_distributed_nccl_available ():
@@ -90,11 +91,15 @@ def forward(self, x):
9091
9192class 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 ))
You can’t perform that action at this time.
0 commit comments