common/cpu_detect: Add AVX512 detection

5 years ago · d6474b4aca
parent 7d1dca4c98
commit d6474b4aca
2 changed files with 6 additions and 0 deletions
--- a/src/common/x64/cpu_detect.cpp
+++ b/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;
+            }
        }
    }

--- a/src/common/x64/cpu_detect.h
+++ b/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;