diff options
| -rw-r--r-- | src/utils/cpu.cpp | 12 | ||||
| -rw-r--r-- | src/utils/cpu.h | 4 |
2 files changed, 15 insertions, 1 deletions
diff --git a/src/utils/cpu.cpp b/src/utils/cpu.cpp index 6737b5261..e57956c06 100644 --- a/src/utils/cpu.cpp +++ b/src/utils/cpu.cpp @@ -51,6 +51,10 @@ void Cpu::detect() mCpuFlags |= FEATURE_SSE4; if (__builtin_cpu_supports ("sse4.2")) mCpuFlags |= FEATURE_SSE42; + if (__builtin_cpu_supports ("avx")) + mCpuFlags |= FEATURE_AVX; + if (__builtin_cpu_supports ("avx2")) + mCpuFlags |= FEATURE_AVX2; printFlags(); #elif defined(__linux__) || defined(__linux) FILE *file = fopen("/proc/cpuinfo", "r"); @@ -84,6 +88,10 @@ void Cpu::detect() mCpuFlags |= FEATURE_SSE4; else if (flag == "sse4_2") mCpuFlags |= FEATURE_SSE42; + else if (flag == "avx") + mCpuFlags |= FEATURE_AVX; + else if (flag == "avx2") + mCpuFlags |= FEATURE_AVX2; } fclose(file); printFlags(); @@ -114,5 +122,9 @@ void Cpu::printFlags() str.append(" sse4"); if (mCpuFlags & FEATURE_SSE42) str.append(" sse4_2"); + if (mCpuFlags & FEATURE_AVX) + str.append(" avx"); + if (mCpuFlags & FEATURE_AVX2) + str.append(" avx2"); logger->log(str); } diff --git a/src/utils/cpu.h b/src/utils/cpu.h index 1a81ca788..2780795bd 100644 --- a/src/utils/cpu.h +++ b/src/utils/cpu.h @@ -33,7 +33,9 @@ namespace Cpu FEATURE_SSE2 = 4, FEATURE_SSSE3 = 8, FEATURE_SSE4 = 16, - FEATURE_SSE42 = 32 + FEATURE_SSE42 = 32, + FEATURE_AVX = 64, + FEATURE_AVX2 = 128 }; void detect(); |