Use dllist on linux to check cross compile feature

Author: apbose

py/requirements.txt

@@ -5,4 +5,5 @@ pybind11==2.6.2
 torch>=2.8.0.dev,<2.9.0
 torchvision>=0.22.0.dev,<0.23.0
 --extra-index-url https://pypi.ngc.nvidia.com
-pyyaml
+pyyaml
+dllist

py/torch_tensorrt/_utils.py

@@ -4,7 +4,6 @@
 from typing import Any
 
 import torch
-from torch_tensorrt import _find_lib
 
 
 def sanitized_torch_version() -> Any:
@@ -16,24 +15,13 @@ def sanitized_torch_version() -> Any:
 
 
 def check_cross_compile_trt_win_lib():
-    if sys.platform.startswith("linux"):
-        LINUX_PATHS = ["/usr/local/cuda-12.8/lib64", "/usr/lib", "/usr/lib64"]
-
-    if platform.uname().processor == "x86_64":
-        LINUX_PATHS += [
-            "/usr/lib/x86_64-linux-gnu",
-        ]
-    elif platform.uname().processor == "aarch64":
-        LINUX_PATHS += ["/usr/lib/aarch64-linux-gnu"]
+    # cross compile feature is only available on linux
+    # build engine on linux and run on windows
+    import dllist
 
-    LINUX_LIBS = [
-        f"libnvinfer_builder_resource_win.so.*",
-    ]
-
-    for lib in LINUX_LIBS:
-        try:
-            ctypes.CDLL(_find_lib(lib, LINUX_PATHS))
+    if sys.platform.startswith("linux"):
+        loaded_libs = dllist.dllist()
+        target_lib = "libnvinfer_builder_resource_win.so.*"
+        if target_lib in loaded_libs:
             return True
-        except:
-            continue
     return False

pyproject.toml

@@ -13,6 +13,7 @@ requires = [
     "pybind11==2.6.2",
     "numpy",
     "sympy",
+    "dllist",
 ]
 build-backend = "setuptools.build_meta"
 
@@ -63,6 +64,7 @@ dependencies = [
     "packaging>=23",
     "numpy",
     "typing-extensions>=4.7.0",
+    "dllist",
 ]
 
 dynamic = ["version"]

tests/py/dynamo/runtime/test_003_cross_compile_for_windows.py

@@ -46,10 +46,8 @@ def forward(self, a, b):
         "Cross compile for windows can only be enabled on linux x86-64 platform",
     )
     @unittest.skipIf(
-        not (
-            check_cross_compile_trt_win_lib(),
-            "TRT windows lib for cross compile not found",
-        ),
+        not (check_cross_compile_trt_win_lib()),
+        "TRT windows lib for cross compile not found",
     )
     @pytest.mark.unit
     def test_dynamo_cross_compile_for_windows(self):
@@ -79,6 +77,10 @@ def forward(self, a, b):
         platform.system() != "Linux" or platform.architecture()[0] != "64bit",
         "Cross compile for windows can only be enabled on linux x86-64 platform",
     )
+    @unittest.skipIf(
+        not (check_cross_compile_trt_win_lib()),
+        "TRT windows lib for cross compile not found",
+    )
     @pytest.mark.unit
     def test_dynamo_cross_compile_for_windows_multiple_output(self):
         class Add(torch.nn.Module):