File tree Expand file tree Collapse file tree 3 files changed +8
-11
lines changed Expand file tree Collapse file tree 3 files changed +8
-11
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ static InitFunction __init_simd_flags(
3434
3535SIMDFlags::SIMDFlags () {
3636 unsigned int cpuInfo[4 ];
37-
37+ // CPUID: https://en.wikipedia.org/wiki/CPUID
3838 CPUID (cpuInfo, 0x00000001 );
3939 simd_flags_ |= cpuInfo[3 ] & (1 << 25 ) ? SIMD_SSE : SIMD_NONE;
4040 simd_flags_ |= cpuInfo[3 ] & (1 << 26 ) ? SIMD_SSE2 : SIMD_NONE;
Original file line number Diff line number Diff line change @@ -61,7 +61,7 @@ class SIMDFlags final {
6161#define HAS_SSE3 SIMDFlags::instance ()->isSSE3()
6262#define HAS_SSSE3 SIMDFlags::instance ()->isSSSE3()
6363#define HAS_SSE41 SIMDFlags::instance ()->isSSE41()
64- #define HAS_SSS42 SIMDFlags::instance ()->isSSE42()
64+ #define HAS_SSE42 SIMDFlags::instance ()->isSSE42()
6565#define HAS_FMA3 SIMDFlags::instance ()->isFMA3()
6666#define HAS_FMA4 SIMDFlags::instance ()->isFMA4()
6767#define HAS_AVX SIMDFlags::instance ()->isAVX()
Original file line number Diff line number Diff line change @@ -20,17 +20,14 @@ using namespace paddle; // NOLINT
2020
2121TEST (SIMDFlags, gccTest) {
2222#if (defined(__GNUC__) || defined(__GNUG__)) && !(defined(__clang__))
23- CHECK (__builtin_cpu_supports (" sse" ) == HAS_SSE);
24- CHECK (__builtin_cpu_supports (" sse2" ) == HAS_SSE2);
25- CHECK (__builtin_cpu_supports (" sse3" ) == HAS_SSE3);
26- CHECK (__builtin_cpu_supports (" ssse3" )== HAS_SSSE3);
23+ CHECK (__builtin_cpu_supports (" sse" ) == HAS_SSE);
24+ CHECK (__builtin_cpu_supports (" sse2" ) == HAS_SSE2);
25+ CHECK (__builtin_cpu_supports (" sse3" ) == HAS_SSE3);
26+ CHECK (__builtin_cpu_supports (" ssse3" ) == HAS_SSSE3);
2727 CHECK (__builtin_cpu_supports (" sse4.1" )== HAS_SSE41);
2828 CHECK (__builtin_cpu_supports (" sse4.2" )== HAS_SSE42);
29- CHECK (__builtin_cpu_supports (" fma3" )== HAS_FMA3);
30- CHECK (__builtin_cpu_supports (" fma4" )== HAS_FMA4);
31- CHECK (__builtin_cpu_supports (" avx" )== HAS_AVX);
32- CHECK (__builtin_cpu_supports (" avx2" )== HAS_AVX2);
33- CHECK (__builtin_cpu_supports (" avx512f" )== HAS_AVX512);
29+ CHECK (__builtin_cpu_supports (" avx" ) == HAS_AVX);
30+ CHECK (__builtin_cpu_supports (" avx2" ) == HAS_AVX2);
3431#endif
3532}
3633
You can’t perform that action at this time.
0 commit comments