Przeglądaj źródła

common/cpu_detect: Add AVX512 detection

Morph 6 lat temu
rodzic
commit
d6474b4aca
2 zmienionych plików z 6 dodań i 0 usunięć
  1. 5 0
      src/common/x64/cpu_detect.cpp
  2. 1 0
      src/common/x64/cpu_detect.h

+ 5 - 0
src/common/x64/cpu_detect.cpp

@@ -110,6 +110,11 @@ static CPUCaps Detect() {
                 caps.bmi1 = true;
             if ((cpu_id[1] >> 8) & 1)
                 caps.bmi2 = true;
+            // Checks for AVX512F, AVX512CD, AVX512VL, AVX512DQ, AVX512BW (Intel Skylake-X/SP)
+            if ((cpu_id[1] >> 16) & 1 && (cpu_id[1] >> 28) & 1 && (cpu_id[1] >> 31) & 1 &&
+                (cpu_id[1] >> 17) & 1 && (cpu_id[1] >> 30) & 1) {
+                caps.avx512 = caps.avx2;
+            }
         }
     }
 

+ 1 - 0
src/common/x64/cpu_detect.h

@@ -19,6 +19,7 @@ struct CPUCaps {
     bool lzcnt;
     bool avx;
     bool avx2;
+    bool avx512;
     bool bmi1;
     bool bmi2;
     bool fma;