fix(//py): Lib path incorrect

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jun 1, 2020 · ff2b13c · ff2b13c
1 parent db20098
commit ff2b13c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/py/trtorch/__init__.py b/py/trtorch/__init__.py
@@ -14,4 +14,6 @@
 
 def _register_with_torch():
     trtorch_dir = os.path.dirname(__file__)
-    torch.ops.load_library(trtorch_dir + '/lib/trtorch.so')
+    torch.ops.load_library(trtorch_dir + '/lib/libtrtorch.so')
+
+_register_with_torch()