Skip to content

Commit ca0dce7

Browse files
authored
Merge pull request #10705 from jacquesqiao/add-mkldnn-to-paddle-lib
add mkldnn to paddle lib
2 parents a06efd2 + 0764211 commit ca0dce7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

cmake/inference_lib.cmake

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,14 @@ elseif (WITH_MKLML)
9898
)
9999
endif()
100100

101+
if(WITH_MKLDNN)
102+
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/mkldnn")
103+
copy(mkldnn_lib
104+
SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
105+
DSTS ${dst_dir} ${dst_dir}/lib
106+
)
107+
endif()
108+
101109
if(NOT MOBILE_INFERENCE AND NOT RPI)
102110
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/snappy")
103111
copy(snappy_lib

0 commit comments

Comments
 (0)