Skip to content

Commit 226ed04

Browse files
committed
addressing the review comments- comments and error message
1 parent a028601 commit 226ed04

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

py/torch_tensorrt/_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def is_platform_supported_for_trtllm() -> bool:
8181
Unsupported:
8282
- Windows platforms
8383
- Jetson/Orin/Xavier (aarch64 architecture + 'tegra' in platform release)
84+
- Thor devices
8485
- CUDA 13 not supported
8586
"""
8687
system = platform.system().lower()
@@ -110,7 +111,7 @@ def is_platform_supported_for_trtllm() -> bool:
110111
major, minor = map(int, cuda_version.split("."))
111112
if major != 12:
112113
logger.error(
113-
"CUDA 13 is not supported for TRT-LLM plugins. Please install pytorch with CUDA 12.x support"
114+
"CUDA 13 is not currently supported for TRT-LLM plugins. Please install pytorch with CUDA 12.x support"
114115
)
115116
return False
116117

0 commit comments

Comments
 (0)