fix(//py): Lib path incorrect

narendasan · narendasan · commit ff2b13cdc9b5 · 2020-06-01T09:18:51.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
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()
