Fix pytorch issue 29722 wrong MKLDNN_THREADING

MKLDNN determines by itself if intel openmp is present or not. Setting this variable here prevents it from using intel openmp. This results in the behavior described in pytorch#29722
CeadeS · Nov 21, 2019 · ee97481 · ee97481
1 parent f7b12a9
commit ee97481
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/Modules/FindMKLDNN.cmake b/cmake/Modules/FindMKLDNN.cmake
@@ -49,7 +49,7 @@ ENDIF(MKL_FOUND)
 
 SET(MKL_cmake_included TRUE)
 IF (NOT MKLDNN_THREADING)
-  SET(MKLDNN_THREADING "OMP:COMP" CACHE STRING "")
+  SET(MKLDNN_THREADING "OMP" CACHE STRING "")
 ELSEIF (MKLDNN_THREADING STREQUAL "TBB")
   IF (USE_TBB)
     MESSAGE(STATUS "MKL-DNN is using TBB")