diff --git a/cmake/inference_lib.cmake b/cmake/inference_lib.cmake index 807a48a41f72f17944dc1be5b793b0ca7d70c527..1144ca7f43b9ffd288b06672dbce82d5fae26347 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/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