Skip to content

Commit 6bbd852

Browse files
committed
excluding thor from the supported platform of TRTLLM wheel
1 parent cee5c7a commit 6bbd852

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -898,9 +898,9 @@ def is_platform_supported_for_trtllm() -> bool:
898898
)
899899
return False
900900

901-
if machine == "aarch64" and "tegra" in release:
901+
if machine == "aarch64" and "tegra" in release or is_thor():
902902
logger.info(
903-
"TensorRT-LLM plugins for NCCL backend are not supported on Jetson/Orin/Xavier (Tegra) devices."
903+
"TensorRT-LLM plugins for NCCL backend are not supported on Jetson/Orin/Xavier (Tegra) or Thor devices."
904904
)
905905
return False
906906

0 commit comments

Comments
 (0)