cmake: check USE_OPENMP and pass proper MKL-DNN build flags (#17356)

apache · Jan 19, 2020 · d95d74f · d95d74f
1 parent bef2c9d
commit d95d74f
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)