add mkldnn to paddle lib

269a62a2 · qiaolongfei · 9707aa6b · 269a62a2
显示空白变更内容
内联并排

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/")
+  copy(mkldnn_lib
+    SRCS ${MKLDNN_INSTALL_DIR}
+    DSTS ${dst_dir}
+  )
+endif()
 if(NOT MOBILE_INFERENCE AND NOT RPI)
  set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/snappy")
  copy(snappy_lib