add mkldnn to paddle lib

8 years ago · 269a62a2b5
parent 9707aa6bde
commit 269a62a2b5
1 changed files with 8 additions and 0 deletions
--- 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