fix(//py): Fix python setup.py with new libtrtorch.so location

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Aug 23, 2021 · 68ba63c · 68ba63c
1 parent 87e7efc
commit 68ba63c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/setup.py b/py/setup.py
@@ -72,7 +72,7 @@ def is_exe(fpath):
 
 def build_libtrtorch_pre_cxx11_abi(develop=True, use_dist_dir=True, cxx11_abi=False):
     cmd = [BAZEL_EXE, "build"]
-    cmd.append("//cpp/api/lib:libtrtorch.so")
+    cmd.append("//cpp/lib:libtrtorch.so")
     if develop:
         cmd.append("--compilation_mode=dbg")
     else:
@@ -115,7 +115,7 @@ def copy_libtrtorch(multilinux=False):
     if multilinux:
         copyfile(dir_path + "/build/libtrtorch_build/libtrtorch.so", dir_path + '/trtorch/lib/libtrtorch.so')
     else:
-        copyfile(dir_path + "/../bazel-bin/cpp/api/lib/libtrtorch.so", dir_path + '/trtorch/lib/libtrtorch.so')
+        copyfile(dir_path + "/../bazel-bin/cpp/lib/libtrtorch.so", dir_path + '/trtorch/lib/libtrtorch.so')
 
 
 class DevelopCommand(develop):