Skip to content

Commit 7b0c12e

Browse files
LuLu
authored andcommitted
remove shared library LIB_DEEPMD_OP_CUDA, instead use static library
1 parent e4e1ac7 commit 7b0c12e

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

source/CMakeLists.txt

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -184,11 +184,6 @@ include_directories(${TensorFlow_INCLUDE_DIRS})
184184
if (BUILD_CPP_IF)
185185
set (LIB_DEEPMD "deepmd")
186186
set (LIB_DEEPMD_OP "deepmd_op")
187-
if (USE_CUDA_TOOLKIT)
188-
set (LIB_DEEPMD_OP_CUDA "deepmd_op_cuda")
189-
else()
190-
set (LIB_DEEPMD_OP_CUDA "deepmd_op")
191-
endif()
192187
if (CMAKE_CXX_COMPILER_VERSION VERSION_GREATER_EQUAL 4.9)
193188
set (LIB_DEEPMD_NATIVE "deepmd_native_md")
194189
set (LIB_DEEPMD_IPI "deepmd_ipi")

source/lmp/env.sh.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,4 @@ TF_RPATH=`echo $TENSORFLOW_LIBRARY_PATH | sed "s/;/ -Wl,-rpath=/g"`
88

99
NNP_INC=" -std=c++11 @PREC_DEF@ @TTM_DEF@ @OLD_LMP_PPPM_DEF@ -I$TF_INCLUDE_DIRS -I$DEEPMD_ROOT/include/deepmd "
1010
NNP_PATH=" -L$TF_LIBRARY_PATH -L$DEEPMD_ROOT/lib"
11-
NNP_LIB=" -Wl,--no-as-needed -l@LIB_DEEPMD_OP@ -l@LIB_DEEPMD_OP_CUDA@ -l@LIB_DEEPMD@ -ltensorflow_cc -ltensorflow_framework -Wl,-rpath=$TF_RPATH -Wl,-rpath=$DEEPMD_ROOT/lib"
11+
NNP_LIB=" -Wl,--no-as-needed -l@LIB_DEEPMD_OP@ -l@LIB_DEEPMD@ -ltensorflow_cc -ltensorflow_framework -Wl,-rpath=$TF_RPATH -Wl,-rpath=$DEEPMD_ROOT/lib"

0 commit comments

Comments
 (0)