Disabling SDL with MKL threading on Windows

apache · Jul 30, 2021 · b3c66c6 · b3c66c6
1 parent d147df0
commit b3c66c6
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/initialize.cc b/src/initialize.cc
@@ -95,13 +95,17 @@ LibraryInitializer::LibraryInitializer()
     cpu_worker_nthreads_(dmlc::GetEnv("MXNET_CPU_WORKER_NTHREADS", 1)),
     mp_cv_num_threads_(dmlc::GetEnv("MXNET_MP_OPENCV_NUM_THREADS", 0)) {
   dmlc::InitLogging("mxnet");
+
+#if !(defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__))
 #if MKL_USE_SINGLE_DYNAMIC_LIBRARY
-  #if defined( __INTEL_LLVM_COMPILER)
-    mkl_set_threading_layer(MKL_THREADING_INTEL);
-  #else
-    mkl_set_threading_layer(MKL_THREADING_GNU);
-  #endif
+#if defined( __INTEL_LLVM_COMPILER)
+  mkl_set_threading_layer(MKL_THREADING_INTEL);
+#else
+  mkl_set_threading_layer(MKL_THREADING_GNU);
+#endif
 #endif
+#endif
+
   engine::OpenMP::Get();   // force OpenMP initialization
   install_pthread_atfork_handlers();
 }