From 14a5431526aa47ea31bb1903e74354d6c05f98f8 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Tue, 9 Sep 2025 13:58:01 -0700 Subject: [PATCH 1/2] cherry pick is_thor --- py/torch_tensorrt/dynamo/utils.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/py/torch_tensorrt/dynamo/utils.py b/py/torch_tensorrt/dynamo/utils.py index b2ca8faa30..2416f343d6 100644 --- a/py/torch_tensorrt/dynamo/utils.py +++ b/py/torch_tensorrt/dynamo/utils.py @@ -852,3 +852,9 @@ def is_tegra_platform() -> bool: if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]: return True return False + + +def is_thor_platform() -> bool: + if torch.cuda.get_device_capability() in [(11, 0)]: + return True + return False From 50ed30bb7475403d395f4028dd015adde9b3d011 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Tue, 9 Sep 2025 14:03:13 -0700 Subject: [PATCH 2/2] typo --- py/torch_tensorrt/dynamo/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/utils.py b/py/torch_tensorrt/dynamo/utils.py index 2416f343d6..564250e5ae 100644 --- a/py/torch_tensorrt/dynamo/utils.py +++ b/py/torch_tensorrt/dynamo/utils.py @@ -854,7 +854,7 @@ def is_tegra_platform() -> bool: return False -def is_thor_platform() -> bool: +def is_thor() -> bool: if torch.cuda.get_device_capability() in [(11, 0)]: return True return False