Browse Source

!15854 mkldnn support multi-thread

From: @limingqi107
Reviewed-by: @cristoval,@wilfchen
Signed-off-by: @wilfchen
pull/15854/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
4cde4fa92f
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      cmake/external_libs/mkl_dnn.cmake

+ 2
- 1
cmake/external_libs/mkl_dnn.cmake View File

@@ -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})


Loading…
Cancel
Save