Fix dll locations

Author: gs-olive

py/torch_tensorrt/__init__.py

@@ -90,21 +90,28 @@ def _find_lib(name: str, paths: List[str]) -> str:
 
 def _register_with_torch() -> None:
     trtorch_dir = os.path.dirname(__file__)
-    linked_file = (
-        "/torchtrt.dll" if sys.platform.startswith("win") else "/lib/libtorchtrt.so"
+    linked_file = os.path.join(
+        "lib", ("torchtrt.dll" if sys.platform.startswith("win") else "libtorchtrt.so")
     )
-    linked_file_runtime = (
-        "/torchtrt_runtime.dll"
-        if sys.platform.startswith("win")
-        else "/lib/libtorchtrt_runtime.so"
+    linked_file_runtime = os.path.join(
+        "lib",
+        (
+            "torchtrt_runtime.dll"
+            if sys.platform.startswith("win")
+            else "libtorchtrt_runtime.so"
+        ),
     )
-    if os.path.isfile(trtorch_dir + linked_file):
+    linked_file_full_path = os.path.join(trtorch_dir, linked_file)
+    linked_file_runtime_full_path = os.path.join(trtorch_dir, linked_file_runtime)
+
+    if os.path.isfile(linked_file_full_path):
         assert ENABLED_FEATURES.torchscript_frontend
         assert ENABLED_FEATURES.torch_tensorrt_runtime
-        torch.ops.load_library(trtorch_dir + linked_file)
-    elif os.path.isfile(trtorch_dir + linked_file_runtime):
+        torch.ops.load_library(linked_file_full_path)
+
+    elif os.path.isfile(linked_file_runtime_full_path):
         assert ENABLED_FEATURES.torch_tensorrt_runtime
-        torch.ops.load_library(trtorch_dir + linked_file_runtime)
+        torch.ops.load_library(linked_file_runtime_full_path)
 
 
 _register_with_torch()