cmake: check USE_OPENMP and pass proper MKL-DNN build flags (apache#1…

…7356)
szhengac · Jan 21, 2020 · f6d69e8 · f6d69e8
1 parent 6cd2ee7
commit f6d69e8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -301,6 +301,10 @@ if(USE_MKLDNN)
   set(MKLDNN_ENABLE_JIT_PROFILING OFF CACHE INTERNAL "" FORCE)
   set(MKLDNN_LIBRARY_TYPE STATIC CACHE INTERNAL "" FORCE)
 
+  if(NOT USE_OPENMP)
+      set(MKLDNN_CPU_RUNTIME SEQ CACHE INTERNAL "" FORCE)
+  endif()
+
   add_subdirectory(3rdparty/mkldnn)
 
   include_directories(3rdparty/mkldnn/include)