From fdf99bbe6f4921d0e474d3ca21f8be8d90b87627 Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Thu, 17 May 2018 10:25:26 +0800
Subject: [PATCH] follow comment

---
 cmake/inference_lib.cmake | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmake/inference_lib.cmake b/cmake/inference_lib.cmake
index 8b7266823..a92b856f2 100644
--- a/cmake/inference_lib.cmake
+++ b/cmake/inference_lib.cmake
@@ -99,10 +99,10 @@ elseif (WITH_MKLML)
 endif()
 
 if(WITH_MKLDNN)
-  set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/")
+  set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/mkldnn")
   copy(mkldnn_lib
-    SRCS ${MKLDNN_INSTALL_DIR}
-    DSTS ${dst_dir}
+    SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
+    DSTS ${dst} ${dst_dir}/lib
   )
 endif()
 
-- 
GitLab