make try_initialize_global_cpu_info thread safe

Tencent · Sep 4, 2024 · c6712ee · c6712ee
1 parent fdc42a1
commit c6712ee
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 {