fix(pt): add libtorch directory to rpath

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Apr 2, 2024 · d42bd8e · d42bd8e
1 parent 8e0cc90
commit d42bd8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -194,6 +194,10 @@ if(ENABLE_PYTORCH AND NOT DEEPMD_C_ROOT)
       add_definitions(-D_GLIBCXX_USE_CXX11_ABI=${OP_CXX_ABI})
     endif()
   endif()
+  # get torch directory
+  set(PyTorch_LIBRARY_PATH ${TORCH_INCLUDE_DIRS}/../lib)
+  # trick: use TensorFlow_LIBRARY_PATH as general backend library paths
+  list(APPEND TensorFlow_LIBRARY_PATH ${PyTorch_LIBRARY_PATH})
 endif()
 # log enabled backends
 if(NOT DEEPMD_C_ROOT)

diff --git a/source/ipi/CMakeLists.txt b/source/ipi/CMakeLists.txt
@@ -44,13 +44,13 @@ if(APPLE)
     ${ipiname}
     PROPERTIES LINK_FLAGS "${extra_link_flags}"
                INSTALL_RPATH
-               "@loader_path/../${LIB_DIR}:${TensorFlow_LIBRARY_PATH}")
+               "@loader_path/../${LIB_DIR};${TensorFlow_LIBRARY_PATH}")
 else()
   set_target_properties(
     ${ipiname}
     PROPERTIES LINK_FLAGS
                "-Wl,-rpath,'$ORIGIN'/../${LIB_DIR} ${extra_link_flags}"
-               INSTALL_RPATH "$ORIGIN/../${LIB_DIR}:${TensorFlow_LIBRARY_PATH}")
+               INSTALL_RPATH "$ORIGIN/../${LIB_DIR};${TensorFlow_LIBRARY_PATH}")
 endif()
 
 if(CMAKE_TESTING_ENABLED)