From f15020fd9973ddfe208c637a0ec020d459973769 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 12 Jul 2024 19:41:11 -0400 Subject: [PATCH] fix(cmake): fix USE_PT_PYTHON_LIBS There is a typo in #3636 Signed-off-by: Jinzhe Zeng --- source/CMakeLists.txt | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt index cf516b9755..8ba32adb8a 100644 --- 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)