mkldnn support multi-thread

2021-04-28 19:27:17 +08:00 · 2021-04-28 19:27:17 +08:00 · a00b5ac42c
parent 57a14f55e8
commit a00b5ac42c
1 changed files with 2 additions and 1 deletions
--- 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})