1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/cpu.h"
6 #include "build/build_config.h"
7
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 #if _MSC_VER >= 1700
11 // C4752: found Intel(R) Advanced Vector Extensions; consider using /arch:AVX.
12 #pragma warning(disable: 4752)
13 #endif
14
15 // Tests whether we can run extended instructions represented by the CPU
16 // information. This test actually executes some extended instructions (such as
17 // MMX, SSE, etc.) supported by the CPU and sees we can run them without
18 // "undefined instruction" exceptions. That is, this test succeeds when this
19 // test finishes without a crash.
TEST(CPU,RunExtendedInstructions)20 TEST(CPU, RunExtendedInstructions) {
21 #if defined(ARCH_CPU_X86_FAMILY)
22 // Retrieve the CPU information.
23 base::CPU cpu;
24
25 ASSERT_TRUE(cpu.has_mmx());
26 ASSERT_TRUE(cpu.has_sse());
27 ASSERT_TRUE(cpu.has_sse2());
28
29 // GCC and clang instruction test.
30 #if defined(COMPILER_GCC)
31 // Execute an MMX instruction.
32 __asm__ __volatile__("emms\n" : : : "mm0");
33
34 // Execute an SSE instruction.
35 __asm__ __volatile__("xorps %%xmm0, %%xmm0\n" : : : "xmm0");
36
37 // Execute an SSE 2 instruction.
38 __asm__ __volatile__("psrldq $0, %%xmm0\n" : : : "xmm0");
39
40 if (cpu.has_sse3()) {
41 // Execute an SSE 3 instruction.
42 __asm__ __volatile__("addsubpd %%xmm0, %%xmm0\n" : : : "xmm0");
43 }
44
45 if (cpu.has_ssse3()) {
46 // Execute a Supplimental SSE 3 instruction.
47 __asm__ __volatile__("psignb %%xmm0, %%xmm0\n" : : : "xmm0");
48 }
49
50 if (cpu.has_sse41()) {
51 // Execute an SSE 4.1 instruction.
52 __asm__ __volatile__("pmuldq %%xmm0, %%xmm0\n" : : : "xmm0");
53 }
54
55 if (cpu.has_sse42()) {
56 // Execute an SSE 4.2 instruction.
57 __asm__ __volatile__("crc32 %%eax, %%eax\n" : : : "eax");
58 }
59
60 if (cpu.has_popcnt()) {
61 // Execute a POPCNT instruction.
62 __asm__ __volatile__("popcnt %%eax, %%eax\n" : : : "eax");
63 }
64
65 if (cpu.has_avx()) {
66 // Execute an AVX instruction.
67 __asm__ __volatile__("vzeroupper\n" : : : "xmm0");
68 }
69
70 if (cpu.has_avx2()) {
71 // Execute an AVX 2 instruction.
72 __asm__ __volatile__("vpunpcklbw %%ymm0, %%ymm0, %%ymm0\n" : : : "xmm0");
73 }
74
75 // Visual C 32 bit and ClangCL 32/64 bit test.
76 #elif defined(COMPILER_MSVC) && (defined(ARCH_CPU_32_BITS) || \
77 (defined(ARCH_CPU_64_BITS) && defined(__clang__)))
78
79 // Execute an MMX instruction.
80 __asm emms;
81
82 // Execute an SSE instruction.
83 __asm xorps xmm0, xmm0;
84
85 // Execute an SSE 2 instruction.
86 __asm psrldq xmm0, 0;
87
88 if (cpu.has_sse3()) {
89 // Execute an SSE 3 instruction.
90 __asm addsubpd xmm0, xmm0;
91 }
92
93 if (cpu.has_ssse3()) {
94 // Execute a Supplimental SSE 3 instruction.
95 __asm psignb xmm0, xmm0;
96 }
97
98 if (cpu.has_sse41()) {
99 // Execute an SSE 4.1 instruction.
100 __asm pmuldq xmm0, xmm0;
101 }
102
103 if (cpu.has_sse42()) {
104 // Execute an SSE 4.2 instruction.
105 __asm crc32 eax, eax;
106 }
107
108 if (cpu.has_popcnt()) {
109 // Execute a POPCNT instruction.
110 __asm popcnt eax, eax;
111 }
112
113 // Visual C 2012 required for AVX.
114 #if _MSC_VER >= 1700
115 if (cpu.has_avx()) {
116 // Execute an AVX instruction.
117 __asm vzeroupper;
118 }
119
120 if (cpu.has_avx2()) {
121 // Execute an AVX 2 instruction.
122 __asm vpunpcklbw ymm0, ymm0, ymm0
123 }
124 #endif // _MSC_VER >= 1700
125 #endif // defined(COMPILER_GCC)
126 #endif // defined(ARCH_CPU_X86_FAMILY)
127 }
128