Merge pull request #10705 from jacquesqiao/add-mkldnn-to-paddle-lib

add mkldnn to paddle lib

Merge pull request #10705 from jacquesqiao/add-mkldnn-to-paddle-lib
add mkldnn to paddle lib
ca0dce70 · Qiao Longfei · GitHub · a06efd29 · 07642119 · ca0dce70
隐藏空白更改
内联并排

Showing with 8 addition and 0 deletion

cmake/inference_lib.cmake cmake/inference_lib.cmake +8 -0

未找到文件。
--- a/cmake/inference_lib.cmake
+++ b/cmake/inference_lib.cmake
@@ -98,6 +98,14 @@ elseif (WITH_MKLML)
    )
 endif()

+if(WITH_MKLDNN)
+  set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/mkldnn")
+  copy(mkldnn_lib
+    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
+    DSTS ${dst_dir} ${dst_dir}/lib
+  )
+endif()
+
 if(NOT MOBILE_INFERENCE AND NOT RPI)
  set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/snappy")
  copy(snappy_lib