From ff2b13cdc9b5c0d7bd89e25e9ae6c07e9cf3bcea Mon Sep 17 00:00:00 2001
From: Naren Dasan <naren@narendasan.com>
Date: Mon, 1 Jun 2020 09:18:51 -0700
Subject: [PATCH] fix(//py): Lib path incorrect

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>
---
 py/trtorch/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/py/trtorch/__init__.py b/py/trtorch/__init__.py
index db46947cb6..88e1ca6db9 100644
--- 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')
\ No newline at end of file
+    torch.ops.load_library(trtorch_dir + '/lib/libtrtorch.so')
+
+_register_with_torch()