make try_initialize_global_cpu_info thread safe

Tencent · Sep 8, 2024 · decbf28 · decbf28
1 parent 0ca79c8
commit decbf28
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -1932,15 +1932,10 @@ static void initialize_global_cpu_info()
 #endif // defined __ANDROID__ || defined __linux__
 }
 
-static int g_cpu_info_initialized = 0;
-
 static inline void try_initialize_global_cpu_info()
 {
-    if (!g_cpu_info_initialized)
-    {
-        initialize_global_cpu_info();
-        g_cpu_info_initialized = 1;
-    }
+    static ncnn::OnceFlag flag = OnceFlagInit;
+    ncnn::CallOnce(flag, &initialize_global_cpu_info);
 }
 
 namespace ncnn {