From 269a62a2b5df26904c5bf74f109a28e3c46908f5 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Wed, 16 May 2018 18:56:14 +0800 Subject: [PATCH] add mkldnn to paddle lib --- cmake/inference_lib.cmake | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/cmake/inference_lib.cmake b/cmake/inference_lib.cmake index 06a7ae5682..8b7266823d 100644 --- 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 -- GitLab