summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/utils/cpu.cpp12
-rw-r--r--src/utils/cpu.h4
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();