File tree 5 files changed +4
-3
lines changed
5 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 16
16
[submodule "3rdparty/mkldnn "]
17
17
path = 3rdparty/mkldnn
18
18
url = https://github.com/intel/mkl-dnn.git
19
- branch = master
20
19
[submodule "3rdparty/tvm "]
21
20
path = 3rdparty/tvm
22
21
url = https://github.com/apache/incubator-tvm.git
Original file line number Diff line number Diff line change @@ -284,6 +284,7 @@ if(USE_MKLDNN)
284
284
set (MKLDNN_ARCH_OPT_FLAGS "" CACHE INTERNAL "" FORCE)
285
285
set (MKLDNN_ENABLE_JIT_PROFILING OFF CACHE INTERNAL "" FORCE)
286
286
set (MKLDNN_LIBRARY_TYPE STATIC CACHE INTERNAL "" FORCE)
287
+ set (DNNL_ENABLE_CONCURRENT_EXEC ON CACHE INTERNAL "" FORCE)
287
288
288
289
if (NOT USE_OPENMP)
289
290
set (MKLDNN_CPU_RUNTIME SEQ CACHE INTERNAL "" FORCE)
Original file line number Diff line number Diff line change @@ -31,6 +31,7 @@ mkldnn_FLAGS += -DMKLDNN_BUILD_TESTS=OFF
31
31
mkldnn_FLAGS += -DMKLDNN_BUILD_EXAMPLES=OFF
32
32
mkldnn_FLAGS += -DMKLDNN_ENABLE_JIT_PROFILING=OFF
33
33
mkldnn_FLAGS += -DMKLDNN_LIBRARY_TYPE=STATIC
34
+ mkldnn_FLAGS += -DDNNL_ENABLE_CONCURRENT_EXEC=ON
34
35
35
36
ifneq ($(USE_OPENMP ) , 1)
36
37
mkldnn_FLAGS += -DMKLDNN_CPU_RUNTIME=SEQ
Original file line number Diff line number Diff line change @@ -100,7 +100,7 @@ static void VerifyDefMem(const mkldnn::memory &mem) {
100
100
101
101
TEST (MKLDNN_UTIL_FUNC, MemFormat) {
102
102
// Check whether the number of format is correct.
103
- CHECK_EQ (mkldnn_format_tag_last, 131 );
103
+ CHECK_EQ (mkldnn_format_tag_last, 152 );
104
104
CHECK_EQ (mkldnn_nchw, 5 );
105
105
CHECK_EQ (mkldnn_oihw, 5 );
106
106
}
You can’t perform that action at this time.
0 commit comments