detect more baseline avx512 flags (#3687)

Tencent · Apr 9, 2022 · 32560f4 · 32560f4
1 parent a062148
commit 32560f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -645,7 +645,7 @@ static int get_cpu_support_x86_avx512()
         return 0;
 
     x86_cpuid_sublevel(7, 0, cpu_info);
-    return (cpu_info[1] & (1u << 16)) && (cpu_info[1] & (1u << 31));
+    return (cpu_info[1] & (1u << 16)) && (cpu_info[1] & (1u << 17)) && (cpu_info[1] & (1u << 28)) && (cpu_info[1] & (1u << 30)) && (cpu_info[1] & (1u << 31));
 }
 
 static int get_cpu_support_x86_avx512_vnni()

diff --git a/src/cpu.h b/src/cpu.h
@@ -66,7 +66,7 @@ NCNN_EXPORT int cpu_support_x86_f16c();
 NCNN_EXPORT int cpu_support_x86_avx2();
 // avx_vnni = x86 avx vnni
 NCNN_EXPORT int cpu_support_x86_avx_vnni();
-// avx512 = x86 avx512f + avx512vl
+// avx512 = x86 avx512f + avx512cd + avx512bw + avx512dq + avx512vl
 NCNN_EXPORT int cpu_support_x86_avx512();
 // avx512_vnni = x86 avx512 vnni
 NCNN_EXPORT int cpu_support_x86_avx512_vnni();