Skip to content

Commit aa8e45f

Browse files
cherry pick is_thor from ngc/release/25.09 branch to main (#3813)
1 parent c75f9bb commit aa8e45f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -852,3 +852,9 @@ def is_tegra_platform() -> bool:
852852
if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]:
853853
return True
854854
return False
855+
856+
857+
def is_thor() -> bool:
858+
if torch.cuda.get_device_capability() in [(11, 0)]:
859+
return True
860+
return False

0 commit comments

Comments
 (0)