Skip to content

Commit 2e6843e

Browse files
fix ci broken issue: psutil not found error (#3903)
This is broken the DLFW, I need to get this in dlfw so I merged in without approval. It is verified in CI.
1 parent 98f54fe commit 2e6843e

File tree

2 files changed

+1
-12
lines changed

2 files changed

+1
-12
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -859,18 +859,6 @@ def get_output_dtypes(output: Any, truncate_double: bool = False) -> List[dtype]
859859
return output_dtypes
860860

861861

862-
def is_tegra_platform() -> bool:
863-
if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]:
864-
return True
865-
return False
866-
867-
868-
def is_thor() -> bool:
869-
if torch.cuda.get_device_capability() in [(11, 0)]:
870-
return True
871-
return False
872-
873-
874862
def get_cpu_memory_usage() -> Any:
875863
return psutil.Process().memory_info().rss / 1024 / 1024
876864

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -729,6 +729,7 @@ def run(self):
729729
"packaging>=23",
730730
"typing-extensions>=4.7.0",
731731
"dllist",
732+
"psutil",
732733
# dummy package as a WAR for the tensorrt dependency on nvidia-cuda-runtime-cu13
733734
"nvidia-cuda-runtime-cu13==0.0.0a0",
734735
]

0 commit comments

Comments
 (0)