Skip to content

Commit 30a99fa

Browse files
committed
fix is_thor missing issue
1 parent e2fa2dd commit 30a99fa

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
@@ -859,3 +859,9 @@ def is_tegra_platform() -> bool:
859859
if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]:
860860
return True
861861
return False
862+
863+
864+
def is_thor() -> bool:
865+
if torch.cuda.get_device_capability() in [(11, 0)]:
866+
return True
867+
return False

0 commit comments

Comments
 (0)