fix(cmake): fix USE_PT_PYTHON_LIBS

There is a typo in deepmodeling#3636 Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jul 12, 2024 · f15020f · f15020f
1 parent 37bee25
commit f15020f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -176,10 +176,7 @@ if(BUILD_CPP_IF
         "Cannot determine PyTorch CMake prefix path, error code: $PYTORCH_CMAKE_PREFIX_PATH_RESULT_VAR}, error message: ${PYTORCH_CMAKE_PREFIX_PATH_ERROR_VAR}"
     )
   endif()
-  list(APPEND BACKEND_LIBRARY_PATH ${TensorFlow_LIBRARY_PATH})
-  if(Protobuf_LIBRARY_DIRECTORY)
-    list(APPEND BACKEND_LIBRARY_PATH ${Protobuf_LIBRARY_DIRECTORY})
-  endif()
+  list(APPEND CMAKE_PREFIX_PATH ${PYTORCH_CMAKE_PREFIX_PATH})
 endif()
 if(ENABLE_PYTORCH AND NOT DEEPMD_C_ROOT)
   find_package(Torch REQUIRED)