From a00b5ac42c33366343341a9b5d067f84940b27e5 Mon Sep 17 00:00:00 2001 From: limingqi107 Date: Wed, 28 Apr 2021 19:27:17 +0800 Subject: [PATCH] mkldnn support multi-thread --- cmake/external_libs/mkl_dnn.cmake | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cmake/external_libs/mkl_dnn.cmake b/cmake/external_libs/mkl_dnn.cmake index c37f0800ff3..1abce2dd316 100644 --- a/cmake/external_libs/mkl_dnn.cmake +++ b/cmake/external_libs/mkl_dnn.cmake @@ -21,7 +21,8 @@ else() LIBS dnnl mkldnn URL ${REQ_URL} MD5 ${MD5} - CMAKE_OPTION -DDNNL_ARCH_OPT_FLAGS='' -DDNNL_BUILD_EXAMPLES=OFF -DDNNL_BUILD_TESTS=OFF) + CMAKE_OPTION -DDNNL_ARCH_OPT_FLAGS='' -DDNNL_BUILD_EXAMPLES=OFF -DDNNL_BUILD_TESTS=OFF + -DDNNL_ENABLE_CONCURRENT_EXEC=ON) endif() include_directories(${onednn_INC})