Support LLVM OpenMP runtime for MSVC (#5370)

with `/openmp:llvm` compile option
Tencent · Mar 14, 2024 · 5a11c38 · 5a11c38
1 parent 18b374e
commit 5a11c38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -2837,7 +2837,7 @@ int get_omp_thread_num()
 
 int get_kmp_blocktime()
 {
-#if defined(_OPENMP) && __clang__
+#if defined(_OPENMP) && (__clang__ || defined(_OPENMP_LLVM_RUNTIME))
     return kmp_get_blocktime();
 #else
     return 0;
@@ -2846,7 +2846,7 @@ int get_kmp_blocktime()
 
 void set_kmp_blocktime(int time_ms)
 {
-#if defined(_OPENMP) && __clang__
+#if defined(_OPENMP) && (__clang__ || defined(_OPENMP_LLVM_RUNTIME))
     kmp_set_blocktime(time_ms);
 #else
     (void)time_ms;