Skip to content

Commit 5c480d6

Browse files
committed
Add reference to CPUID
1 parent 4f73e3f commit 5c480d6

File tree

3 files changed

+8
-11
lines changed

3 files changed

+8
-11
lines changed

paddle/utils/CpuId.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ static InitFunction __init_simd_flags(
3434

3535
SIMDFlags::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;

paddle/utils/CpuId.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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()

paddle/utils/tests/test_SIMDFlags.cpp

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,17 +20,14 @@ using namespace paddle; // NOLINT
2020

2121
TEST(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

0 commit comments

Comments
 (0)