Skip to content

Commit ef0974f

Browse files
TaoLvanirudh2290
authored andcommitted
Update 3rdparty/mkldnn to v1.2.2 (apache#17313)
* fix cpp test * update to dnnl v1.2-rc * pin rls-v1.2 * build dnnl with DNNL_ENABLE_CONCURRENT_EXEC=ON * update rls-v1.2 * update to formal 1.2 release * try patch * fix rnn * pin rls-v1.2 * dnnl v1.2.1 * dnnl v1.2.2
1 parent 88adc0a commit ef0974f

File tree

5 files changed

+4
-3
lines changed

5 files changed

+4
-3
lines changed

.gitmodules

-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
[submodule "3rdparty/mkldnn"]
1717
path = 3rdparty/mkldnn
1818
url = https://github.com/intel/mkl-dnn.git
19-
branch = master
2019
[submodule "3rdparty/tvm"]
2120
path = 3rdparty/tvm
2221
url = https://github.com/apache/incubator-tvm.git

3rdparty/mkldnn

CMakeLists.txt

+1
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,7 @@ if(USE_MKLDNN)
284284
set(MKLDNN_ARCH_OPT_FLAGS "" CACHE INTERNAL "" FORCE)
285285
set(MKLDNN_ENABLE_JIT_PROFILING OFF CACHE INTERNAL "" FORCE)
286286
set(MKLDNN_LIBRARY_TYPE STATIC CACHE INTERNAL "" FORCE)
287+
set(DNNL_ENABLE_CONCURRENT_EXEC ON CACHE INTERNAL "" FORCE)
287288

288289
if(NOT USE_OPENMP)
289290
set(MKLDNN_CPU_RUNTIME SEQ CACHE INTERNAL "" FORCE)

mkldnn.mk

+1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ mkldnn_FLAGS += -DMKLDNN_BUILD_TESTS=OFF
3131
mkldnn_FLAGS += -DMKLDNN_BUILD_EXAMPLES=OFF
3232
mkldnn_FLAGS += -DMKLDNN_ENABLE_JIT_PROFILING=OFF
3333
mkldnn_FLAGS += -DMKLDNN_LIBRARY_TYPE=STATIC
34+
mkldnn_FLAGS += -DDNNL_ENABLE_CONCURRENT_EXEC=ON
3435

3536
ifneq ($(USE_OPENMP), 1)
3637
mkldnn_FLAGS += -DMKLDNN_CPU_RUNTIME=SEQ

tests/cpp/operator/mkldnn_test.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ static void VerifyDefMem(const mkldnn::memory &mem) {
100100

101101
TEST(MKLDNN_UTIL_FUNC, MemFormat) {
102102
// Check whether the number of format is correct.
103-
CHECK_EQ(mkldnn_format_tag_last, 131);
103+
CHECK_EQ(mkldnn_format_tag_last, 152);
104104
CHECK_EQ(mkldnn_nchw, 5);
105105
CHECK_EQ(mkldnn_oihw, 5);
106106
}

0 commit comments

Comments
 (0)