fix bestla openmp macro

Signed-off-by: airMeng <[email protected]>
intel · Jan 9, 2024 · 468673d · 468673d
1 parent 3c8246d
commit 468673d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/bestla/bestla/bestla_utils.h b/bestla/bestla/bestla_utils.h
@@ -14,7 +14,14 @@
 #pragma once
 #include <type_traits>
 
-#define BTLA_OPENMP ((_OPENMP) && (BTLA_USE_OPENMP))
+#define BTLA_OPENMP 0
+
+#ifdef BTLA_USE_OPENMP
+#ifdef _OPENMP
+#define BTLA_OPENMP 1
+#endif
+#endif
+
 #if BTLA_OPENMP
 #include <omp.h>
 #endif