fix(cmake): fix PyTorch_LIBRARY_PATH

The previous `PyTorch_LIBRARY_PATH` in deepmodeling#3636 is wrong: `BACKEND_INCLUDE_DIRS` is a list of paths instead of a single path. Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jun 20, 2024 · 6d150ab · 6d150ab
1 parent 0c472d1
commit 6d150ab
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -199,8 +199,9 @@ 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)
+  # get torch directory get the directory of the target "torch"
+  get_target_property(_TORCH_LOCATION torch LOCATION)
+  get_filename_component(PyTorch_LIBRARY_PATH ${_TORCH_LOCATION} DIRECTORY)
   list(APPEND BACKEND_LIBRARY_PATH ${PyTorch_LIBRARY_PATH})
   list(APPEND BACKEND_INCLUDE_DIRS ${TORCH_INCLUDE_DIRS})
 endif()