1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
|
/* vim: set shiftwidth=4 tabstop=8 autoindent cindent expandtab: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/* compile-time and runtime tests for whether to use SSE instructions */
#include "SSE.h"
#include "mozilla/Attributes.h"
#ifdef HAVE_CPUID_H
// cpuid.h is available on gcc 4.3 and higher on i386 and x86_64
# include <cpuid.h>
#elif defined(_MSC_VER) && (defined(_M_IX86) || defined(_M_AMD64))
// MSVC 2005 or newer on x86-32 or x86-64
# include <intrin.h>
#endif
namespace {
// SSE.h has parallel #ifs which declare MOZILLA_SSE_HAVE_CPUID_DETECTION.
// We can't declare these functions in the header file, however, because
// <intrin.h> conflicts with <windows.h> on MSVC 2005, and some files want to
// include both SSE.h and <windows.h>.
#ifdef HAVE_CPUID_H
enum CPUIDRegister { eax = 0, ebx = 1, ecx = 2, edx = 3 };
static bool has_cpuid_bits(unsigned int level, CPUIDRegister reg,
unsigned int bits) {
unsigned int regs[4];
unsigned int eax, ebx, ecx, edx;
unsigned max = __get_cpuid_max(level & 0x80000000u, nullptr);
if (level > max) return false;
__cpuid_count(level, 0, eax, ebx, ecx, edx);
regs[0] = eax;
regs[1] = ebx;
regs[2] = ecx;
regs[3] = edx;
return (regs[reg] & bits) == bits;
}
static bool has_cpuid_bits_ex(unsigned int level, CPUIDRegister reg,
unsigned int bits) {
unsigned int regs[4];
unsigned int eax, ebx, ecx, edx;
unsigned max = __get_cpuid_max(level & 0x80000000u, nullptr);
if (level > max) return false;
__cpuid_count(level, 1, eax, ebx, ecx, edx);
regs[0] = eax;
regs[1] = ebx;
regs[2] = ecx;
regs[3] = edx;
return (regs[reg] & bits) == bits;
}
#elif defined(_MSC_VER) && (defined(_M_IX86) || defined(_M_AMD64))
enum CPUIDRegister { eax = 0, ebx = 1, ecx = 2, edx = 3 };
static bool has_cpuid_bits(unsigned int level, CPUIDRegister reg,
unsigned int bits) {
// Check that the level in question is supported.
int regs[4];
__cpuid_ex(regs, level & 0x80000000u, 1);
if (unsigned(regs[0]) < level) return false;
// "The __cpuid intrinsic clears the ECX register before calling the cpuid
// instruction."
__cpuid_ex(regs, level, 1);
return (unsigned(regs[reg]) & bits) == bits;
}
#elif (defined(__GNUC__) || defined(__SUNPRO_CC)) && \
(defined(__i386) || defined(__x86_64__))
enum CPUIDRegister { eax = 0, ebx = 1, ecx = 2, edx = 3 };
# ifdef __i386
static void moz_cpuid(int CPUInfo[4], int InfoType) {
asm("xchg %esi, %ebx\n"
"xor %ecx, %ecx\n" // ecx is the sub-leaf (we only ever need 0)
"cpuid\n"
"movl %eax, (%edi)\n"
"movl %ebx, 4(%edi)\n"
"movl %ecx, 8(%edi)\n"
"movl %edx, 12(%edi)\n"
"xchg %esi, %ebx\n"
:
: "a"(InfoType), // %eax
"D"(CPUInfo) // %edi
: "%ecx", "%edx", "%esi");
}
static void moz_cpuid_ex(int CPUInfo[4], int InfoType) {
asm("xchg %esi, %ebx\n"
"movl 1, %ecx\n"
"cpuid\n"
"movl %eax, (%edi)\n"
"movl %ebx, 4(%edi)\n"
"movl %ecx, 8(%edi)\n"
"movl %edx, 12(%edi)\n"
"xchg %esi, %ebx\n"
:
: "a"(InfoType), // %eax
"D"(CPUInfo) // %edi
: "%ecx", "%edx", "%esi");
}
# else
static void moz_cpuid(int CPUInfo[4], int InfoType) {
asm("xchg %rsi, %rbx\n"
"xor %ecx, %ecx\n" // ecx is the sub-leaf (we only ever need 0)
"cpuid\n"
"movl %eax, (%rdi)\n"
"movl %ebx, 4(%rdi)\n"
"movl %ecx, 8(%rdi)\n"
"movl %edx, 12(%rdi)\n"
"xchg %rsi, %rbx\n"
:
: "a"(InfoType), // %eax
"D"(CPUInfo) // %rdi
: "%ecx", "%edx", "%rsi");
}
static void moz_cpuid_ex(int CPUInfo[4], int InfoType) {
asm("xchg %rsi, %rbx\n"
"movl 1, %ecx\n"
"cpuid\n"
"movl %eax, (%rdi)\n"
"movl %ebx, 4(%rdi)\n"
"movl %ecx, 8(%rdi)\n"
"movl %edx, 12(%rdi)\n"
"xchg %rsi, %rbx\n"
:
: "a"(InfoType), // %eax
"D"(CPUInfo) // %rdi
: "%ecx", "%edx", "%rsi");
}
# endif
static bool has_cpuid_bits(unsigned int level, CPUIDRegister reg,
unsigned int bits) {
// Check that the level in question is supported.
volatile int regs[4];
moz_cpuid((int*)regs, level & 0x80000000u);
if (unsigned(regs[0]) < level) return false;
moz_cpuid((int*)regs, level);
return (unsigned(regs[reg]) & bits) == bits;
}
static bool has_cpuid_bits_ex(unsigned int level, CPUIDRegister reg,
unsigned int bits) {
// Check that the level in question is supported.
volatile int regs[4];
moz_cpuid_ex((int*)regs, level & 0x80000000u);
if (unsigned(regs[0]) < level) return false;
moz_cpuid_ex((int*)regs, level);
return (unsigned(regs[reg]) & bits) == bits;
}
#endif // end CPUID declarations
} // namespace
namespace mozilla {
namespace sse_private {
#if defined(MOZILLA_SSE_HAVE_CPUID_DETECTION)
# if !defined(MOZILLA_PRESUME_MMX)
MOZ_RUNINIT bool mmx_enabled = has_cpuid_bits(1u, edx, (1u << 23));
# endif
# if !defined(MOZILLA_PRESUME_SSE)
MOZ_RUNINIT bool sse_enabled = has_cpuid_bits(1u, edx, (1u << 25));
# endif
# if !defined(MOZILLA_PRESUME_SSE2)
MOZ_RUNINIT bool sse2_enabled = has_cpuid_bits(1u, edx, (1u << 26));
# endif
# if !defined(MOZILLA_PRESUME_SSE3)
MOZ_RUNINIT bool sse3_enabled = has_cpuid_bits(1u, ecx, (1u << 0));
# endif
# if !defined(MOZILLA_PRESUME_SSSE3)
MOZ_RUNINIT bool ssse3_enabled = has_cpuid_bits(1u, ecx, (1u << 9));
# endif
# if !defined(MOZILLA_PRESUME_SSE4A)
MOZ_RUNINIT bool sse4a_enabled = has_cpuid_bits(0x80000001u, ecx, (1u << 6));
# endif
# if !defined(MOZILLA_PRESUME_SSE4_1)
MOZ_RUNINIT bool sse4_1_enabled = has_cpuid_bits(1u, ecx, (1u << 19));
# endif
# if !defined(MOZILLA_PRESUME_SSE4_2)
MOZ_RUNINIT bool sse4_2_enabled = has_cpuid_bits(1u, ecx, (1u << 20));
# endif
# if !defined(MOZILLA_PRESUME_FMA3)
MOZ_RUNINIT bool fma3_enabled = has_cpuid_bits(1u, ecx, (1u << 12));
# endif
# if !defined(MOZILLA_PRESUME_AVX) || !defined(MOZILLA_PRESUME_AVX2)
static bool has_avx() {
# if defined(MOZILLA_PRESUME_AVX)
return true;
# else
const unsigned AVX = 1u << 28;
const unsigned OSXSAVE = 1u << 27;
const unsigned XSAVE = 1u << 26;
const unsigned XMM_STATE = 1u << 1;
const unsigned YMM_STATE = 1u << 2;
const unsigned AVX_STATE = XMM_STATE | YMM_STATE;
return has_cpuid_bits(1u, ecx, AVX | OSXSAVE | XSAVE) &&
// ensure the OS supports XSAVE of YMM registers
(xgetbv(0) & AVX_STATE) == AVX_STATE;
# endif // MOZILLA_PRESUME_AVX
}
# endif // !MOZILLA_PRESUME_AVX || !MOZILLA_PRESUME_AVX2
# if !defined(MOZILLA_PRESUME_AVX)
MOZ_RUNINIT bool avx_enabled = has_avx();
# endif
# if !defined(MOZILLA_PRESUME_AVX2)
MOZ_RUNINIT bool avx2_enabled = has_avx() && has_cpuid_bits(7u, ebx, (1u << 5));
# endif
# if !defined(MOZILLA_PRESUME_AVXVNNI)
MOZ_RUNINIT bool avxvnni_enabled = has_cpuid_bits_ex(7u, eax, (1u << 4));
# endif
# if !defined(MOZILLA_PRESUME_AES)
MOZ_RUNINIT bool aes_enabled = has_cpuid_bits(1u, ecx, (1u << 25));
# endif
# if !defined(MOZILLA_PRESUME_SHA)
MOZ_RUNINIT bool sha_enabled = has_cpuid_bits_ex(7u, ebx, (1u << 29));
# endif
# if !defined(MOZILLA_PRESUME_SHA512)
MOZ_RUNINIT bool sha512_enabled = has_cpuid_bits_ex(7u, eax, (1u << 0));
# endif
// To accommodate old QEMU, put BMI behind `has_avx`.
// https://searchfox.org/firefox-main/rev/938e8f38c6765875e998d5c2965ad5864f5a5ee2/js/src/jit/x86-shared/Assembler-x86-shared.cpp#380-381
# if !defined(MOZILLA_PRESUME_BMI)
MOZ_RUNINIT bool bmi_enabled = has_avx() && has_cpuid_bits(7u, ebx, (1u << 3));
# endif
# if !defined(MOZILLA_PRESUME_BMI2)
MOZ_RUNINIT bool bmi2_enabled = has_avx() &&
has_cpuid_bits(7u, ebx, (1u << 3)) &&
has_cpuid_bits(7u, ebx, (1u << 8));
# endif
MOZ_RUNINIT bool has_constant_tsc = has_cpuid_bits(0x80000007u, edx, (1u << 8));
#endif
} // namespace sse_private
#ifdef HAVE_CPUID_H
uint64_t xgetbv(uint32_t xcr) {
uint32_t eax, edx;
__asm__(".byte 0x0f, 0x01, 0xd0" : "=a"(eax), "=d"(edx) : "c"(xcr));
return (uint64_t)(edx) << 32 | eax;
}
#endif
} // namespace mozilla
|
