blob: 695683efa6257d7692c40c981b9fefe4082a929f [file] [log] [blame]
Simon Pilgrima271c542017-05-03 15:42:29 +00001//===-- Host.cpp - Implement OS Host Concept --------------------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the operating system Host concept.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/Support/Host.h"
Craig Topperc77d00e2017-11-10 17:10:57 +000015#include "llvm/Support/TargetParser.h"
Simon Pilgrima271c542017-05-03 15:42:29 +000016#include "llvm/ADT/SmallSet.h"
17#include "llvm/ADT/SmallVector.h"
18#include "llvm/ADT/StringRef.h"
19#include "llvm/ADT/StringSwitch.h"
20#include "llvm/ADT/Triple.h"
21#include "llvm/Config/config.h"
22#include "llvm/Support/Debug.h"
23#include "llvm/Support/FileSystem.h"
24#include "llvm/Support/MemoryBuffer.h"
25#include "llvm/Support/raw_ostream.h"
26#include <assert.h>
27#include <string.h>
28
29// Include the platform-specific parts of this class.
30#ifdef LLVM_ON_UNIX
31#include "Unix/Host.inc"
32#endif
33#ifdef LLVM_ON_WIN32
34#include "Windows/Host.inc"
35#endif
36#ifdef _MSC_VER
37#include <intrin.h>
38#endif
39#if defined(__APPLE__) && (defined(__ppc__) || defined(__powerpc__))
40#include <mach/host_info.h>
41#include <mach/mach.h>
42#include <mach/mach_host.h>
43#include <mach/machine.h>
44#endif
45
46#define DEBUG_TYPE "host-detection"
47
48//===----------------------------------------------------------------------===//
49//
50// Implementations of the CPU detection routines
51//
52//===----------------------------------------------------------------------===//
53
54using namespace llvm;
55
56static std::unique_ptr<llvm::MemoryBuffer>
57 LLVM_ATTRIBUTE_UNUSED getProcCpuinfoContent() {
58 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Text =
59 llvm::MemoryBuffer::getFileAsStream("/proc/cpuinfo");
60 if (std::error_code EC = Text.getError()) {
61 llvm::errs() << "Can't read "
62 << "/proc/cpuinfo: " << EC.message() << "\n";
63 return nullptr;
64 }
65 return std::move(*Text);
66}
67
68StringRef sys::detail::getHostCPUNameForPowerPC(
69 const StringRef &ProcCpuinfoContent) {
70 // Access to the Processor Version Register (PVR) on PowerPC is privileged,
71 // and so we must use an operating-system interface to determine the current
72 // processor type. On Linux, this is exposed through the /proc/cpuinfo file.
73 const char *generic = "generic";
74
75 // The cpu line is second (after the 'processor: 0' line), so if this
76 // buffer is too small then something has changed (or is wrong).
77 StringRef::const_iterator CPUInfoStart = ProcCpuinfoContent.begin();
78 StringRef::const_iterator CPUInfoEnd = ProcCpuinfoContent.end();
79
80 StringRef::const_iterator CIP = CPUInfoStart;
81
82 StringRef::const_iterator CPUStart = 0;
83 size_t CPULen = 0;
84
85 // We need to find the first line which starts with cpu, spaces, and a colon.
86 // After the colon, there may be some additional spaces and then the cpu type.
87 while (CIP < CPUInfoEnd && CPUStart == 0) {
88 if (CIP < CPUInfoEnd && *CIP == '\n')
89 ++CIP;
90
91 if (CIP < CPUInfoEnd && *CIP == 'c') {
92 ++CIP;
93 if (CIP < CPUInfoEnd && *CIP == 'p') {
94 ++CIP;
95 if (CIP < CPUInfoEnd && *CIP == 'u') {
96 ++CIP;
97 while (CIP < CPUInfoEnd && (*CIP == ' ' || *CIP == '\t'))
98 ++CIP;
99
100 if (CIP < CPUInfoEnd && *CIP == ':') {
101 ++CIP;
102 while (CIP < CPUInfoEnd && (*CIP == ' ' || *CIP == '\t'))
103 ++CIP;
104
105 if (CIP < CPUInfoEnd) {
106 CPUStart = CIP;
107 while (CIP < CPUInfoEnd && (*CIP != ' ' && *CIP != '\t' &&
108 *CIP != ',' && *CIP != '\n'))
109 ++CIP;
110 CPULen = CIP - CPUStart;
111 }
112 }
113 }
114 }
115 }
116
117 if (CPUStart == 0)
118 while (CIP < CPUInfoEnd && *CIP != '\n')
119 ++CIP;
120 }
121
122 if (CPUStart == 0)
123 return generic;
124
125 return StringSwitch<const char *>(StringRef(CPUStart, CPULen))
126 .Case("604e", "604e")
127 .Case("604", "604")
128 .Case("7400", "7400")
129 .Case("7410", "7400")
130 .Case("7447", "7400")
131 .Case("7455", "7450")
132 .Case("G4", "g4")
133 .Case("POWER4", "970")
134 .Case("PPC970FX", "970")
135 .Case("PPC970MP", "970")
136 .Case("G5", "g5")
137 .Case("POWER5", "g5")
138 .Case("A2", "a2")
139 .Case("POWER6", "pwr6")
140 .Case("POWER7", "pwr7")
141 .Case("POWER8", "pwr8")
142 .Case("POWER8E", "pwr8")
143 .Case("POWER8NVL", "pwr8")
144 .Case("POWER9", "pwr9")
145 .Default(generic);
146}
147
148StringRef sys::detail::getHostCPUNameForARM(
149 const StringRef &ProcCpuinfoContent) {
150 // The cpuid register on arm is not accessible from user space. On Linux,
151 // it is exposed through the /proc/cpuinfo file.
152
153 // Read 32 lines from /proc/cpuinfo, which should contain the CPU part line
154 // in all cases.
155 SmallVector<StringRef, 32> Lines;
156 ProcCpuinfoContent.split(Lines, "\n");
157
158 // Look for the CPU implementer line.
159 StringRef Implementer;
160 StringRef Hardware;
161 for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
162 if (Lines[I].startswith("CPU implementer"))
163 Implementer = Lines[I].substr(15).ltrim("\t :");
164 if (Lines[I].startswith("Hardware"))
165 Hardware = Lines[I].substr(8).ltrim("\t :");
166 }
167
168 if (Implementer == "0x41") { // ARM Ltd.
169 // MSM8992/8994 may give cpu part for the core that the kernel is running on,
170 // which is undeterministic and wrong. Always return cortex-a53 for these SoC.
171 if (Hardware.endswith("MSM8994") || Hardware.endswith("MSM8996"))
172 return "cortex-a53";
173
174
175 // Look for the CPU part line.
176 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
177 if (Lines[I].startswith("CPU part"))
178 // The CPU part is a 3 digit hexadecimal number with a 0x prefix. The
179 // values correspond to the "Part number" in the CP15/c0 register. The
180 // contents are specified in the various processor manuals.
181 return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
182 .Case("0x926", "arm926ej-s")
183 .Case("0xb02", "mpcore")
184 .Case("0xb36", "arm1136j-s")
185 .Case("0xb56", "arm1156t2-s")
186 .Case("0xb76", "arm1176jz-s")
187 .Case("0xc08", "cortex-a8")
188 .Case("0xc09", "cortex-a9")
189 .Case("0xc0f", "cortex-a15")
190 .Case("0xc20", "cortex-m0")
191 .Case("0xc23", "cortex-m3")
192 .Case("0xc24", "cortex-m4")
193 .Case("0xd04", "cortex-a35")
194 .Case("0xd03", "cortex-a53")
195 .Case("0xd07", "cortex-a57")
196 .Case("0xd08", "cortex-a72")
197 .Case("0xd09", "cortex-a73")
198 .Default("generic");
199 }
200
201 if (Implementer == "0x51") // Qualcomm Technologies, Inc.
202 // Look for the CPU part line.
203 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
204 if (Lines[I].startswith("CPU part"))
205 // The CPU part is a 3 digit hexadecimal number with a 0x prefix. The
206 // values correspond to the "Part number" in the CP15/c0 register. The
207 // contents are specified in the various processor manuals.
208 return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
209 .Case("0x06f", "krait") // APQ8064
210 .Case("0x201", "kryo")
211 .Case("0x205", "kryo")
Eli Friedmanbde9fc72017-09-13 21:48:00 +0000212 .Case("0x211", "kryo")
213 .Case("0x800", "cortex-a73")
214 .Case("0x801", "cortex-a73")
Balaram Makama1e7ecc72017-09-22 17:46:36 +0000215 .Case("0xc00", "falkor")
Chad Rosier71070852017-09-25 14:05:00 +0000216 .Case("0xc01", "saphira")
Simon Pilgrima271c542017-05-03 15:42:29 +0000217 .Default("generic");
218
Evandro Menezes5d7a9e62017-12-08 21:09:59 +0000219 if (Implementer == "0x53") { // Samsung Electronics Co., Ltd.
220 // The Exynos chips have a convoluted ID scheme that doesn't seem to follow
221 // any predictive pattern across variants and parts.
222 unsigned Variant = 0, Part = 0;
223
224 // Look for the CPU variant line, whose value is a 1 digit hexadecimal
225 // number, corresponding to the Variant bits in the CP15/C0 register.
226 for (auto I : Lines)
227 if (I.consume_front("CPU variant"))
228 I.ltrim("\t :").getAsInteger(0, Variant);
229
230 // Look for the CPU part line, whose value is a 3 digit hexadecimal
231 // number, corresponding to the PartNum bits in the CP15/C0 register.
232 for (auto I : Lines)
233 if (I.consume_front("CPU part"))
234 I.ltrim("\t :").getAsInteger(0, Part);
235
236 unsigned Exynos = (Variant << 12) | Part;
237 switch (Exynos) {
238 default:
239 // Default by falling through to Exynos M1.
240 LLVM_FALLTHROUGH;
241
242 case 0x1001:
243 return "exynos-m1";
244
245 case 0x4001:
246 return "exynos-m2";
247 }
248 }
249
Simon Pilgrima271c542017-05-03 15:42:29 +0000250 return "generic";
251}
252
253StringRef sys::detail::getHostCPUNameForS390x(
254 const StringRef &ProcCpuinfoContent) {
255 // STIDP is a privileged operation, so use /proc/cpuinfo instead.
256
257 // The "processor 0:" line comes after a fair amount of other information,
258 // including a cache breakdown, but this should be plenty.
259 SmallVector<StringRef, 32> Lines;
260 ProcCpuinfoContent.split(Lines, "\n");
261
262 // Look for the CPU features.
263 SmallVector<StringRef, 32> CPUFeatures;
264 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
265 if (Lines[I].startswith("features")) {
266 size_t Pos = Lines[I].find(":");
267 if (Pos != StringRef::npos) {
268 Lines[I].drop_front(Pos + 1).split(CPUFeatures, ' ');
269 break;
270 }
271 }
272
273 // We need to check for the presence of vector support independently of
274 // the machine type, since we may only use the vector register set when
275 // supported by the kernel (and hypervisor).
276 bool HaveVectorSupport = false;
277 for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
278 if (CPUFeatures[I] == "vx")
279 HaveVectorSupport = true;
280 }
281
282 // Now check the processor machine type.
283 for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
284 if (Lines[I].startswith("processor ")) {
285 size_t Pos = Lines[I].find("machine = ");
286 if (Pos != StringRef::npos) {
287 Pos += sizeof("machine = ") - 1;
288 unsigned int Id;
289 if (!Lines[I].drop_front(Pos).getAsInteger(10, Id)) {
Ulrich Weigand2b3482f2017-07-17 17:41:11 +0000290 if (Id >= 3906 && HaveVectorSupport)
291 return "z14";
Simon Pilgrima271c542017-05-03 15:42:29 +0000292 if (Id >= 2964 && HaveVectorSupport)
293 return "z13";
294 if (Id >= 2827)
295 return "zEC12";
296 if (Id >= 2817)
297 return "z196";
298 }
299 }
300 break;
301 }
302 }
303
304 return "generic";
305}
306
Yonghong Songdc1dbf62017-08-23 04:25:57 +0000307StringRef sys::detail::getHostCPUNameForBPF() {
308#if !defined(__linux__) || !defined(__x86_64__)
309 return "generic";
310#else
311 uint8_t insns[40] __attribute__ ((aligned (8))) =
312 /* BPF_MOV64_IMM(BPF_REG_0, 0) */
313 { 0xb7, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
314 /* BPF_MOV64_IMM(BPF_REG_2, 1) */
315 0xb7, 0x2, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
316 /* BPF_JMP_REG(BPF_JLT, BPF_REG_0, BPF_REG_2, 1) */
317 0xad, 0x20, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0,
318 /* BPF_MOV64_IMM(BPF_REG_0, 1) */
319 0xb7, 0x0, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
320 /* BPF_EXIT_INSN() */
321 0x95, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0 };
322
323 struct bpf_prog_load_attr {
324 uint32_t prog_type;
325 uint32_t insn_cnt;
326 uint64_t insns;
327 uint64_t license;
328 uint32_t log_level;
329 uint32_t log_size;
330 uint64_t log_buf;
331 uint32_t kern_version;
332 uint32_t prog_flags;
333 } attr = {};
334 attr.prog_type = 1; /* BPF_PROG_TYPE_SOCKET_FILTER */
335 attr.insn_cnt = 5;
336 attr.insns = (uint64_t)insns;
337 attr.license = (uint64_t)"DUMMY";
338
339 int fd = syscall(321 /* __NR_bpf */, 5 /* BPF_PROG_LOAD */, &attr, sizeof(attr));
Yonghong Songc6d25712017-08-23 16:24:31 +0000340 if (fd >= 0) {
341 close(fd);
342 return "v2";
343 }
344 return "v1";
Yonghong Songdc1dbf62017-08-23 04:25:57 +0000345#endif
346}
347
Simon Pilgrima271c542017-05-03 15:42:29 +0000348#if defined(__i386__) || defined(_M_IX86) || \
349 defined(__x86_64__) || defined(_M_X64)
350
351enum VendorSignatures {
352 SIG_INTEL = 0x756e6547 /* Genu */,
353 SIG_AMD = 0x68747541 /* Auth */
354};
355
Simon Pilgrima271c542017-05-03 15:42:29 +0000356// The check below for i386 was copied from clang's cpuid.h (__get_cpuid_max).
357// Check motivated by bug reports for OpenSSL crashing on CPUs without CPUID
358// support. Consequently, for i386, the presence of CPUID is checked first
359// via the corresponding eflags bit.
360// Removal of cpuid.h header motivated by PR30384
361// Header cpuid.h and method __get_cpuid_max are not used in llvm, clang, openmp
362// or test-suite, but are used in external projects e.g. libstdcxx
363static bool isCpuIdSupported() {
364#if defined(__GNUC__) || defined(__clang__)
365#if defined(__i386__)
366 int __cpuid_supported;
367 __asm__(" pushfl\n"
368 " popl %%eax\n"
369 " movl %%eax,%%ecx\n"
370 " xorl $0x00200000,%%eax\n"
371 " pushl %%eax\n"
372 " popfl\n"
373 " pushfl\n"
374 " popl %%eax\n"
375 " movl $0,%0\n"
376 " cmpl %%eax,%%ecx\n"
377 " je 1f\n"
378 " movl $1,%0\n"
379 "1:"
380 : "=r"(__cpuid_supported)
381 :
382 : "eax", "ecx");
383 if (!__cpuid_supported)
384 return false;
385#endif
386 return true;
387#endif
388 return true;
389}
390
391/// getX86CpuIDAndInfo - Execute the specified cpuid and return the 4 values in
392/// the specified arguments. If we can't run cpuid on the host, return true.
393static bool getX86CpuIDAndInfo(unsigned value, unsigned *rEAX, unsigned *rEBX,
394 unsigned *rECX, unsigned *rEDX) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000395#if defined(__GNUC__) || defined(__clang__)
396#if defined(__x86_64__)
397 // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
398 // FIXME: should we save this for Clang?
399 __asm__("movq\t%%rbx, %%rsi\n\t"
400 "cpuid\n\t"
401 "xchgq\t%%rbx, %%rsi\n\t"
402 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
403 : "a"(value));
Craig Topper1efd10a2017-07-10 06:04:11 +0000404 return false;
Simon Pilgrima271c542017-05-03 15:42:29 +0000405#elif defined(__i386__)
406 __asm__("movl\t%%ebx, %%esi\n\t"
407 "cpuid\n\t"
408 "xchgl\t%%ebx, %%esi\n\t"
409 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
410 : "a"(value));
Craig Topper1efd10a2017-07-10 06:04:11 +0000411 return false;
Simon Pilgrima271c542017-05-03 15:42:29 +0000412#else
Craig Topper1efd10a2017-07-10 06:04:11 +0000413 return true;
Simon Pilgrima271c542017-05-03 15:42:29 +0000414#endif
415#elif defined(_MSC_VER)
416 // The MSVC intrinsic is portable across x86 and x64.
417 int registers[4];
418 __cpuid(registers, value);
419 *rEAX = registers[0];
420 *rEBX = registers[1];
421 *rECX = registers[2];
422 *rEDX = registers[3];
Simon Pilgrima271c542017-05-03 15:42:29 +0000423 return false;
424#else
425 return true;
426#endif
427}
428
429/// getX86CpuIDAndInfoEx - Execute the specified cpuid with subleaf and return
430/// the 4 values in the specified arguments. If we can't run cpuid on the host,
431/// return true.
432static bool getX86CpuIDAndInfoEx(unsigned value, unsigned subleaf,
433 unsigned *rEAX, unsigned *rEBX, unsigned *rECX,
434 unsigned *rEDX) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000435#if defined(__GNUC__) || defined(__clang__)
Craig Topper828cf302017-07-17 05:16:16 +0000436#if defined(__x86_64__)
Craig Topperada983a2017-07-10 06:09:22 +0000437 // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
Simon Pilgrima271c542017-05-03 15:42:29 +0000438 // FIXME: should we save this for Clang?
439 __asm__("movq\t%%rbx, %%rsi\n\t"
440 "cpuid\n\t"
441 "xchgq\t%%rbx, %%rsi\n\t"
442 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
443 : "a"(value), "c"(subleaf));
Craig Topper1efd10a2017-07-10 06:04:11 +0000444 return false;
Craig Topper828cf302017-07-17 05:16:16 +0000445#elif defined(__i386__)
446 __asm__("movl\t%%ebx, %%esi\n\t"
447 "cpuid\n\t"
448 "xchgl\t%%ebx, %%esi\n\t"
449 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
450 : "a"(value), "c"(subleaf));
451 return false;
452#else
453 return true;
454#endif
Simon Pilgrima271c542017-05-03 15:42:29 +0000455#elif defined(_MSC_VER)
456 int registers[4];
457 __cpuidex(registers, value, subleaf);
458 *rEAX = registers[0];
459 *rEBX = registers[1];
460 *rECX = registers[2];
461 *rEDX = registers[3];
Craig Topper1efd10a2017-07-10 06:04:11 +0000462 return false;
463#else
464 return true;
Simon Pilgrima271c542017-05-03 15:42:29 +0000465#endif
Simon Pilgrima271c542017-05-03 15:42:29 +0000466}
467
Craig Topperf3af64e2017-07-12 06:49:57 +0000468// Read control register 0 (XCR0). Used to detect features such as AVX.
Simon Pilgrima271c542017-05-03 15:42:29 +0000469static bool getX86XCR0(unsigned *rEAX, unsigned *rEDX) {
470#if defined(__GNUC__) || defined(__clang__)
471 // Check xgetbv; this uses a .byte sequence instead of the instruction
472 // directly because older assemblers do not include support for xgetbv and
473 // there is no easy way to conditionally compile based on the assembler used.
474 __asm__(".byte 0x0f, 0x01, 0xd0" : "=a"(*rEAX), "=d"(*rEDX) : "c"(0));
475 return false;
476#elif defined(_MSC_FULL_VER) && defined(_XCR_XFEATURE_ENABLED_MASK)
477 unsigned long long Result = _xgetbv(_XCR_XFEATURE_ENABLED_MASK);
478 *rEAX = Result;
479 *rEDX = Result >> 32;
480 return false;
481#else
482 return true;
483#endif
484}
485
486static void detectX86FamilyModel(unsigned EAX, unsigned *Family,
487 unsigned *Model) {
488 *Family = (EAX >> 8) & 0xf; // Bits 8 - 11
489 *Model = (EAX >> 4) & 0xf; // Bits 4 - 7
490 if (*Family == 6 || *Family == 0xf) {
491 if (*Family == 0xf)
492 // Examine extended family ID if family ID is F.
493 *Family += (EAX >> 20) & 0xff; // Bits 20 - 27
494 // Examine extended model ID if family ID is 6 or F.
495 *Model += ((EAX >> 16) & 0xf) << 4; // Bits 16 - 19
496 }
497}
498
499static void
Craig Topperc6bbe4b2017-07-08 05:16:14 +0000500getIntelProcessorTypeAndSubtype(unsigned Family, unsigned Model,
501 unsigned Brand_id, unsigned Features,
Craig Topper3a5d0822017-07-12 06:49:58 +0000502 unsigned Features2, unsigned *Type,
503 unsigned *Subtype) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000504 if (Brand_id != 0)
505 return;
506 switch (Family) {
507 case 3:
Craig Topperc77d00e2017-11-10 17:10:57 +0000508 *Type = X86::INTEL_i386;
Simon Pilgrima271c542017-05-03 15:42:29 +0000509 break;
510 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000511 *Type = X86::INTEL_i486;
Simon Pilgrima271c542017-05-03 15:42:29 +0000512 break;
513 case 5:
Craig Topper47c87392017-11-21 23:36:42 +0000514 if (Features & (1 << X86::FEATURE_MMX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000515 *Type = X86::INTEL_PENTIUM_MMX;
Simon Pilgrima271c542017-05-03 15:42:29 +0000516 break;
517 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000518 *Type = X86::INTEL_PENTIUM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000519 break;
520 case 6:
521 switch (Model) {
522 case 0x01: // Pentium Pro processor
Craig Topperc77d00e2017-11-10 17:10:57 +0000523 *Type = X86::INTEL_PENTIUM_PRO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000524 break;
525 case 0x03: // Intel Pentium II OverDrive processor, Pentium II processor,
526 // model 03
527 case 0x05: // Pentium II processor, model 05, Pentium II Xeon processor,
528 // model 05, and Intel Celeron processor, model 05
529 case 0x06: // Celeron processor, model 06
Craig Topperc77d00e2017-11-10 17:10:57 +0000530 *Type = X86::INTEL_PENTIUM_II;
Simon Pilgrima271c542017-05-03 15:42:29 +0000531 break;
532 case 0x07: // Pentium III processor, model 07, and Pentium III Xeon
533 // processor, model 07
534 case 0x08: // Pentium III processor, model 08, Pentium III Xeon processor,
535 // model 08, and Celeron processor, model 08
536 case 0x0a: // Pentium III Xeon processor, model 0Ah
537 case 0x0b: // Pentium III processor, model 0Bh
Craig Topperc77d00e2017-11-10 17:10:57 +0000538 *Type = X86::INTEL_PENTIUM_III;
Simon Pilgrima271c542017-05-03 15:42:29 +0000539 break;
540 case 0x09: // Intel Pentium M processor, Intel Celeron M processor model 09.
541 case 0x0d: // Intel Pentium M processor, Intel Celeron M processor, model
542 // 0Dh. All processors are manufactured using the 90 nm process.
543 case 0x15: // Intel EP80579 Integrated Processor and Intel EP80579
544 // Integrated Processor with Intel QuickAssist Technology
Craig Topperc77d00e2017-11-10 17:10:57 +0000545 *Type = X86::INTEL_PENTIUM_M;
Simon Pilgrima271c542017-05-03 15:42:29 +0000546 break;
547 case 0x0e: // Intel Core Duo processor, Intel Core Solo processor, model
548 // 0Eh. All processors are manufactured using the 65 nm process.
Craig Topperc77d00e2017-11-10 17:10:57 +0000549 *Type = X86::INTEL_CORE_DUO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000550 break; // yonah
551 case 0x0f: // Intel Core 2 Duo processor, Intel Core 2 Duo mobile
552 // processor, Intel Core 2 Quad processor, Intel Core 2 Quad
553 // mobile processor, Intel Core 2 Extreme processor, Intel
554 // Pentium Dual-Core processor, Intel Xeon processor, model
555 // 0Fh. All processors are manufactured using the 65 nm process.
556 case 0x16: // Intel Celeron processor model 16h. All processors are
557 // manufactured using the 65 nm process
Craig Topperc77d00e2017-11-10 17:10:57 +0000558 *Type = X86::INTEL_CORE2; // "core2"
559 *Subtype = X86::INTEL_CORE2_65;
Simon Pilgrima271c542017-05-03 15:42:29 +0000560 break;
561 case 0x17: // Intel Core 2 Extreme processor, Intel Xeon processor, model
562 // 17h. All processors are manufactured using the 45 nm process.
563 //
564 // 45nm: Penryn , Wolfdale, Yorkfield (XE)
565 case 0x1d: // Intel Xeon processor MP. All processors are manufactured using
566 // the 45 nm process.
Craig Topperc77d00e2017-11-10 17:10:57 +0000567 *Type = X86::INTEL_CORE2; // "penryn"
568 *Subtype = X86::INTEL_CORE2_45;
Simon Pilgrima271c542017-05-03 15:42:29 +0000569 break;
570 case 0x1a: // Intel Core i7 processor and Intel Xeon processor. All
571 // processors are manufactured using the 45 nm process.
572 case 0x1e: // Intel(R) Core(TM) i7 CPU 870 @ 2.93GHz.
573 // As found in a Summer 2010 model iMac.
574 case 0x1f:
575 case 0x2e: // Nehalem EX
Craig Topperc77d00e2017-11-10 17:10:57 +0000576 *Type = X86::INTEL_COREI7; // "nehalem"
577 *Subtype = X86::INTEL_COREI7_NEHALEM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000578 break;
579 case 0x25: // Intel Core i7, laptop version.
580 case 0x2c: // Intel Core i7 processor and Intel Xeon processor. All
581 // processors are manufactured using the 32 nm process.
582 case 0x2f: // Westmere EX
Craig Topperc77d00e2017-11-10 17:10:57 +0000583 *Type = X86::INTEL_COREI7; // "westmere"
584 *Subtype = X86::INTEL_COREI7_WESTMERE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000585 break;
586 case 0x2a: // Intel Core i7 processor. All processors are manufactured
587 // using the 32 nm process.
588 case 0x2d:
Craig Topperc77d00e2017-11-10 17:10:57 +0000589 *Type = X86::INTEL_COREI7; //"sandybridge"
590 *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000591 break;
592 case 0x3a:
593 case 0x3e: // Ivy Bridge EP
Craig Topperc77d00e2017-11-10 17:10:57 +0000594 *Type = X86::INTEL_COREI7; // "ivybridge"
595 *Subtype = X86::INTEL_COREI7_IVYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000596 break;
597
598 // Haswell:
599 case 0x3c:
600 case 0x3f:
601 case 0x45:
602 case 0x46:
Craig Topperc77d00e2017-11-10 17:10:57 +0000603 *Type = X86::INTEL_COREI7; // "haswell"
604 *Subtype = X86::INTEL_COREI7_HASWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000605 break;
606
607 // Broadwell:
608 case 0x3d:
609 case 0x47:
610 case 0x4f:
611 case 0x56:
Craig Topperc77d00e2017-11-10 17:10:57 +0000612 *Type = X86::INTEL_COREI7; // "broadwell"
613 *Subtype = X86::INTEL_COREI7_BROADWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000614 break;
615
616 // Skylake:
617 case 0x4e: // Skylake mobile
618 case 0x5e: // Skylake desktop
619 case 0x8e: // Kaby Lake mobile
620 case 0x9e: // Kaby Lake desktop
Craig Topperc77d00e2017-11-10 17:10:57 +0000621 *Type = X86::INTEL_COREI7; // "skylake"
622 *Subtype = X86::INTEL_COREI7_SKYLAKE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000623 break;
624
625 // Skylake Xeon:
626 case 0x55:
Craig Topperc77d00e2017-11-10 17:10:57 +0000627 *Type = X86::INTEL_COREI7;
628 *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512; // "skylake-avx512"
Simon Pilgrima271c542017-05-03 15:42:29 +0000629 break;
630
Craig Topper07491862017-11-15 06:02:42 +0000631 // Cannonlake:
632 case 0x66:
633 *Type = X86::INTEL_COREI7;
634 *Subtype = X86::INTEL_COREI7_CANNONLAKE; // "cannonlake"
635 break;
636
Simon Pilgrima271c542017-05-03 15:42:29 +0000637 case 0x1c: // Most 45 nm Intel Atom processors
638 case 0x26: // 45 nm Atom Lincroft
639 case 0x27: // 32 nm Atom Medfield
640 case 0x35: // 32 nm Atom Midview
641 case 0x36: // 32 nm Atom Midview
Craig Topperc77d00e2017-11-10 17:10:57 +0000642 *Type = X86::INTEL_BONNELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000643 break; // "bonnell"
644
645 // Atom Silvermont codes from the Intel software optimization guide.
646 case 0x37:
647 case 0x4a:
648 case 0x4d:
649 case 0x5a:
650 case 0x5d:
651 case 0x4c: // really airmont
Craig Topperc77d00e2017-11-10 17:10:57 +0000652 *Type = X86::INTEL_SILVERMONT;
Simon Pilgrima271c542017-05-03 15:42:29 +0000653 break; // "silvermont"
Michael Zuckerman4bcb9c32017-06-29 10:00:33 +0000654 // Goldmont:
Craig Topper0dadfe32017-11-15 06:02:43 +0000655 case 0x5c: // Apollo Lake
656 case 0x5f: // Denverton
657 case 0x7a: // Gemini Lake
Craig Topperc77d00e2017-11-10 17:10:57 +0000658 *Type = X86::INTEL_GOLDMONT;
Michael Zuckerman4bcb9c32017-06-29 10:00:33 +0000659 break; // "goldmont"
Simon Pilgrima271c542017-05-03 15:42:29 +0000660 case 0x57:
Craig Topperc77d00e2017-11-10 17:10:57 +0000661 *Type = X86::INTEL_KNL; // knl
Simon Pilgrima271c542017-05-03 15:42:29 +0000662 break;
Craig Topper5d692912017-10-13 18:10:17 +0000663 case 0x85:
Craig Topperc77d00e2017-11-10 17:10:57 +0000664 *Type = X86::INTEL_KNM; // knm
Craig Topper5d692912017-10-13 18:10:17 +0000665 break;
Simon Pilgrima271c542017-05-03 15:42:29 +0000666
667 default: // Unknown family 6 CPU, try to guess.
Craig Topper47c87392017-11-21 23:36:42 +0000668 if (Features & (1 << X86::FEATURE_AVX512VBMI)) {
Craig Topper07491862017-11-15 06:02:42 +0000669 *Type = X86::INTEL_COREI7;
670 *Subtype = X86::INTEL_COREI7_CANNONLAKE;
Craig Topper4eda7562017-07-27 03:26:52 +0000671 break;
672 }
Craig Topper07491862017-11-15 06:02:42 +0000673
Craig Topper47c87392017-11-21 23:36:42 +0000674 if (Features & (1 << X86::FEATURE_AVX512VL)) {
Craig Topper07491862017-11-15 06:02:42 +0000675 *Type = X86::INTEL_COREI7;
676 *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512;
677 break;
678 }
679
Craig Topper47c87392017-11-21 23:36:42 +0000680 if (Features & (1 << X86::FEATURE_AVX512ER)) {
Craig Topper07491862017-11-15 06:02:42 +0000681 *Type = X86::INTEL_KNL; // knl
682 break;
683 }
684
Craig Topper47c87392017-11-21 23:36:42 +0000685 if (Features2 & (1 << (X86::FEATURE_CLFLUSHOPT - 32))) {
686 if (Features2 & (1 << (X86::FEATURE_SHA - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000687 *Type = X86::INTEL_GOLDMONT;
Craig Topper4eda7562017-07-27 03:26:52 +0000688 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000689 *Type = X86::INTEL_COREI7;
690 *Subtype = X86::INTEL_COREI7_SKYLAKE;
Craig Topper4eda7562017-07-27 03:26:52 +0000691 }
Simon Pilgrima271c542017-05-03 15:42:29 +0000692 break;
693 }
Craig Topper47c87392017-11-21 23:36:42 +0000694 if (Features2 & (1 << (X86::FEATURE_ADX - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000695 *Type = X86::INTEL_COREI7;
696 *Subtype = X86::INTEL_COREI7_BROADWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000697 break;
698 }
Craig Topper47c87392017-11-21 23:36:42 +0000699 if (Features & (1 << X86::FEATURE_AVX2)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000700 *Type = X86::INTEL_COREI7;
701 *Subtype = X86::INTEL_COREI7_HASWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000702 break;
703 }
Craig Topper47c87392017-11-21 23:36:42 +0000704 if (Features & (1 << X86::FEATURE_AVX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000705 *Type = X86::INTEL_COREI7;
706 *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000707 break;
708 }
Craig Topper47c87392017-11-21 23:36:42 +0000709 if (Features & (1 << X86::FEATURE_SSE4_2)) {
710 if (Features2 & (1 << (X86::FEATURE_MOVBE - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000711 *Type = X86::INTEL_SILVERMONT;
Simon Pilgrima271c542017-05-03 15:42:29 +0000712 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000713 *Type = X86::INTEL_COREI7;
714 *Subtype = X86::INTEL_COREI7_NEHALEM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000715 }
716 break;
717 }
Craig Topper47c87392017-11-21 23:36:42 +0000718 if (Features & (1 << X86::FEATURE_SSE4_1)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000719 *Type = X86::INTEL_CORE2; // "penryn"
720 *Subtype = X86::INTEL_CORE2_45;
Simon Pilgrima271c542017-05-03 15:42:29 +0000721 break;
722 }
Craig Topper47c87392017-11-21 23:36:42 +0000723 if (Features & (1 << X86::FEATURE_SSSE3)) {
724 if (Features2 & (1 << (X86::FEATURE_MOVBE - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000725 *Type = X86::INTEL_BONNELL; // "bonnell"
Simon Pilgrima271c542017-05-03 15:42:29 +0000726 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000727 *Type = X86::INTEL_CORE2; // "core2"
728 *Subtype = X86::INTEL_CORE2_65;
Simon Pilgrima271c542017-05-03 15:42:29 +0000729 }
730 break;
731 }
Craig Topper47c87392017-11-21 23:36:42 +0000732 if (Features2 & (1 << (X86::FEATURE_EM64T - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000733 *Type = X86::INTEL_CORE2; // "core2"
734 *Subtype = X86::INTEL_CORE2_65;
Craig Toppera233e162017-11-02 19:13:32 +0000735 break;
736 }
Craig Topper47c87392017-11-21 23:36:42 +0000737 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000738 *Type = X86::INTEL_CORE_DUO;
Craig Toppera233e162017-11-02 19:13:32 +0000739 break;
Simon Pilgrima271c542017-05-03 15:42:29 +0000740 }
Craig Topper47c87392017-11-21 23:36:42 +0000741 if (Features & (1 << X86::FEATURE_SSE2)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000742 *Type = X86::INTEL_PENTIUM_M;
Simon Pilgrima271c542017-05-03 15:42:29 +0000743 break;
744 }
Craig Topper47c87392017-11-21 23:36:42 +0000745 if (Features & (1 << X86::FEATURE_SSE)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000746 *Type = X86::INTEL_PENTIUM_III;
Simon Pilgrima271c542017-05-03 15:42:29 +0000747 break;
748 }
Craig Topper47c87392017-11-21 23:36:42 +0000749 if (Features & (1 << X86::FEATURE_MMX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000750 *Type = X86::INTEL_PENTIUM_II;
Simon Pilgrima271c542017-05-03 15:42:29 +0000751 break;
752 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000753 *Type = X86::INTEL_PENTIUM_PRO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000754 break;
755 }
756 break;
757 case 15: {
Craig Topper47c87392017-11-21 23:36:42 +0000758 if (Features2 & (1 << (X86::FEATURE_EM64T - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000759 *Type = X86::INTEL_NOCONA;
Simon Pilgrima271c542017-05-03 15:42:29 +0000760 break;
761 }
Craig Topper47c87392017-11-21 23:36:42 +0000762 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000763 *Type = X86::INTEL_PRESCOTT;
Craig Topper14949152017-11-02 19:13:34 +0000764 break;
765 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000766 *Type = X86::INTEL_PENTIUM_IV;
Simon Pilgrima271c542017-05-03 15:42:29 +0000767 break;
768 }
769 default:
770 break; /*"generic"*/
771 }
772}
773
Craig Topper2ace1532017-07-08 06:44:34 +0000774static void getAMDProcessorTypeAndSubtype(unsigned Family, unsigned Model,
775 unsigned Features, unsigned *Type,
Simon Pilgrima271c542017-05-03 15:42:29 +0000776 unsigned *Subtype) {
777 // FIXME: this poorly matches the generated SubtargetFeatureKV table. There
778 // appears to be no way to generate the wide variety of AMD-specific targets
779 // from the information returned from CPUID.
780 switch (Family) {
781 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000782 *Type = X86::AMD_i486;
Simon Pilgrima271c542017-05-03 15:42:29 +0000783 break;
784 case 5:
Craig Topperc77d00e2017-11-10 17:10:57 +0000785 *Type = X86::AMDPENTIUM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000786 switch (Model) {
787 case 6:
788 case 7:
Craig Topperc77d00e2017-11-10 17:10:57 +0000789 *Subtype = X86::AMDPENTIUM_K6;
Simon Pilgrima271c542017-05-03 15:42:29 +0000790 break; // "k6"
791 case 8:
Craig Topperc77d00e2017-11-10 17:10:57 +0000792 *Subtype = X86::AMDPENTIUM_K62;
Simon Pilgrima271c542017-05-03 15:42:29 +0000793 break; // "k6-2"
794 case 9:
795 case 13:
Craig Topperc77d00e2017-11-10 17:10:57 +0000796 *Subtype = X86::AMDPENTIUM_K63;
Simon Pilgrima271c542017-05-03 15:42:29 +0000797 break; // "k6-3"
798 case 10:
Craig Topperc77d00e2017-11-10 17:10:57 +0000799 *Subtype = X86::AMDPENTIUM_GEODE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000800 break; // "geode"
801 }
802 break;
803 case 6:
Craig Topper47c87392017-11-21 23:36:42 +0000804 if (Features & (1 << X86::FEATURE_SSE)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000805 *Type = X86::AMD_ATHLON_XP;
Simon Pilgrima271c542017-05-03 15:42:29 +0000806 break; // "athlon-xp"
807 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000808 *Type = X86::AMD_ATHLON;
Craig Topperf3de5eb2017-07-13 06:34:10 +0000809 break; // "athlon"
Simon Pilgrima271c542017-05-03 15:42:29 +0000810 case 15:
Craig Topper47c87392017-11-21 23:36:42 +0000811 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000812 *Type = X86::AMD_K8SSE3;
Simon Pilgrima271c542017-05-03 15:42:29 +0000813 break; // "k8-sse3"
814 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000815 *Type = X86::AMD_K8;
Craig Topperf3de5eb2017-07-13 06:34:10 +0000816 break; // "k8"
Simon Pilgrima271c542017-05-03 15:42:29 +0000817 case 16:
Craig Topperc77d00e2017-11-10 17:10:57 +0000818 *Type = X86::AMDFAM10H; // "amdfam10"
Simon Pilgrima271c542017-05-03 15:42:29 +0000819 switch (Model) {
820 case 2:
Craig Topperc77d00e2017-11-10 17:10:57 +0000821 *Subtype = X86::AMDFAM10H_BARCELONA;
Simon Pilgrima271c542017-05-03 15:42:29 +0000822 break;
823 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000824 *Subtype = X86::AMDFAM10H_SHANGHAI;
Simon Pilgrima271c542017-05-03 15:42:29 +0000825 break;
826 case 8:
Craig Topperc77d00e2017-11-10 17:10:57 +0000827 *Subtype = X86::AMDFAM10H_ISTANBUL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000828 break;
829 }
830 break;
831 case 20:
Craig Topperc77d00e2017-11-10 17:10:57 +0000832 *Type = X86::AMD_BTVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000833 break; // "btver1";
834 case 21:
Craig Topperc77d00e2017-11-10 17:10:57 +0000835 *Type = X86::AMDFAM15H;
Craig Topper1f9d3c02017-07-08 06:44:35 +0000836 if (Model >= 0x60 && Model <= 0x7f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000837 *Subtype = X86::AMDFAM15H_BDVER4;
Craig Topper3db11702017-07-12 06:49:56 +0000838 break; // "bdver4"; 60h-7Fh: Excavator
Simon Pilgrima271c542017-05-03 15:42:29 +0000839 }
840 if (Model >= 0x30 && Model <= 0x3f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000841 *Subtype = X86::AMDFAM15H_BDVER3;
Simon Pilgrima271c542017-05-03 15:42:29 +0000842 break; // "bdver3"; 30h-3Fh: Steamroller
843 }
844 if (Model >= 0x10 && Model <= 0x1f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000845 *Subtype = X86::AMDFAM15H_BDVER2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000846 break; // "bdver2"; 10h-1Fh: Piledriver
847 }
848 if (Model <= 0x0f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000849 *Subtype = X86::AMDFAM15H_BDVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000850 break; // "bdver1"; 00h-0Fh: Bulldozer
851 }
852 break;
853 case 22:
Craig Topperc77d00e2017-11-10 17:10:57 +0000854 *Type = X86::AMD_BTVER2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000855 break; // "btver2"
856 case 23:
Craig Topperc77d00e2017-11-10 17:10:57 +0000857 *Type = X86::AMDFAM17H;
858 *Subtype = X86::AMDFAM17H_ZNVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000859 break;
860 default:
861 break; // "generic"
862 }
863}
864
Craig Topper3a5d0822017-07-12 06:49:58 +0000865static void getAvailableFeatures(unsigned ECX, unsigned EDX, unsigned MaxLeaf,
866 unsigned *FeaturesOut,
867 unsigned *Features2Out) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000868 unsigned Features = 0;
Craig Topper3a5d0822017-07-12 06:49:58 +0000869 unsigned Features2 = 0;
Craig Topperc6bbe4b2017-07-08 05:16:14 +0000870 unsigned EAX, EBX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000871
872 if ((EDX >> 15) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000873 Features |= 1 << X86::FEATURE_CMOV;
Craig Topper3a5d0822017-07-12 06:49:58 +0000874 if ((EDX >> 23) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000875 Features |= 1 << X86::FEATURE_MMX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000876 if ((EDX >> 25) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000877 Features |= 1 << X86::FEATURE_SSE;
Craig Topper3a5d0822017-07-12 06:49:58 +0000878 if ((EDX >> 26) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000879 Features |= 1 << X86::FEATURE_SSE2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000880
881 if ((ECX >> 0) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000882 Features |= 1 << X86::FEATURE_SSE3;
Craig Topper3a5d0822017-07-12 06:49:58 +0000883 if ((ECX >> 1) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000884 Features |= 1 << X86::FEATURE_PCLMUL;
Craig Topper3a5d0822017-07-12 06:49:58 +0000885 if ((ECX >> 9) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000886 Features |= 1 << X86::FEATURE_SSSE3;
Craig Topper3a5d0822017-07-12 06:49:58 +0000887 if ((ECX >> 12) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000888 Features |= 1 << X86::FEATURE_FMA;
Craig Topper3a5d0822017-07-12 06:49:58 +0000889 if ((ECX >> 19) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000890 Features |= 1 << X86::FEATURE_SSE4_1;
Craig Topper3a5d0822017-07-12 06:49:58 +0000891 if ((ECX >> 20) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000892 Features |= 1 << X86::FEATURE_SSE4_2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000893 if ((ECX >> 23) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000894 Features |= 1 << X86::FEATURE_POPCNT;
Craig Topper3a5d0822017-07-12 06:49:58 +0000895 if ((ECX >> 25) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000896 Features |= 1 << X86::FEATURE_AES;
Craig Topper3a5d0822017-07-12 06:49:58 +0000897
898 if ((ECX >> 22) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000899 Features2 |= 1 << (X86::FEATURE_MOVBE - 32);
Simon Pilgrima271c542017-05-03 15:42:29 +0000900
901 // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
902 // indicates that the AVX registers will be saved and restored on context
903 // switch, then we have full AVX support.
904 const unsigned AVXBits = (1 << 27) | (1 << 28);
905 bool HasAVX = ((ECX & AVXBits) == AVXBits) && !getX86XCR0(&EAX, &EDX) &&
906 ((EAX & 0x6) == 0x6);
907 bool HasAVX512Save = HasAVX && ((EAX & 0xe0) == 0xe0);
Craig Topper3a5d0822017-07-12 06:49:58 +0000908
909 if (HasAVX)
Craig Topper47c87392017-11-21 23:36:42 +0000910 Features |= 1 << X86::FEATURE_AVX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000911
Simon Pilgrima271c542017-05-03 15:42:29 +0000912 bool HasLeaf7 =
913 MaxLeaf >= 0x7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
Craig Topper3a5d0822017-07-12 06:49:58 +0000914
915 if (HasLeaf7 && ((EBX >> 3) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000916 Features |= 1 << X86::FEATURE_BMI;
Craig Topper3a5d0822017-07-12 06:49:58 +0000917 if (HasLeaf7 && ((EBX >> 5) & 1) && HasAVX)
Craig Topper47c87392017-11-21 23:36:42 +0000918 Features |= 1 << X86::FEATURE_AVX2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000919 if (HasLeaf7 && ((EBX >> 9) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000920 Features |= 1 << X86::FEATURE_BMI2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000921 if (HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000922 Features |= 1 << X86::FEATURE_AVX512F;
Craig Topper3a5d0822017-07-12 06:49:58 +0000923 if (HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000924 Features |= 1 << X86::FEATURE_AVX512DQ;
Craig Topper3a5d0822017-07-12 06:49:58 +0000925 if (HasLeaf7 && ((EBX >> 19) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000926 Features2 |= 1 << (X86::FEATURE_ADX - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000927 if (HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000928 Features |= 1 << X86::FEATURE_AVX512IFMA;
Craig Topper4eda7562017-07-27 03:26:52 +0000929 if (HasLeaf7 && ((EBX >> 23) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000930 Features2 |= 1 << (X86::FEATURE_CLFLUSHOPT - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000931 if (HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000932 Features |= 1 << X86::FEATURE_AVX512PF;
Craig Topper3a5d0822017-07-12 06:49:58 +0000933 if (HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000934 Features |= 1 << X86::FEATURE_AVX512ER;
Craig Topper3a5d0822017-07-12 06:49:58 +0000935 if (HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000936 Features |= 1 << X86::FEATURE_AVX512CD;
Craig Topper4eda7562017-07-27 03:26:52 +0000937 if (HasLeaf7 && ((EBX >> 29) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000938 Features2 |= 1 << (X86::FEATURE_SHA - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000939 if (HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000940 Features |= 1 << X86::FEATURE_AVX512BW;
Craig Topper3a5d0822017-07-12 06:49:58 +0000941 if (HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000942 Features |= 1 << X86::FEATURE_AVX512VL;
Craig Topper3a5d0822017-07-12 06:49:58 +0000943
944 if (HasLeaf7 && ((ECX >> 1) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000945 Features |= 1 << X86::FEATURE_AVX512VBMI;
Craig Topper3a5d0822017-07-12 06:49:58 +0000946 if (HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000947 Features |= 1 << X86::FEATURE_AVX512VPOPCNTDQ;
Craig Topper3a5d0822017-07-12 06:49:58 +0000948
949 if (HasLeaf7 && ((EDX >> 2) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000950 Features |= 1 << X86::FEATURE_AVX5124VNNIW;
Craig Topper3a5d0822017-07-12 06:49:58 +0000951 if (HasLeaf7 && ((EDX >> 3) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000952 Features |= 1 << X86::FEATURE_AVX5124FMAPS;
Simon Pilgrima271c542017-05-03 15:42:29 +0000953
Craig Topperbb8c7992017-07-08 05:16:13 +0000954 unsigned MaxExtLevel;
955 getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
956
957 bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
958 !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
Craig Topper3a5d0822017-07-12 06:49:58 +0000959 if (HasExtLeaf1 && ((ECX >> 6) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000960 Features |= 1 << X86::FEATURE_SSE4_A;
Craig Topper3a5d0822017-07-12 06:49:58 +0000961 if (HasExtLeaf1 && ((ECX >> 11) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000962 Features |= 1 << X86::FEATURE_XOP;
Craig Topper3a5d0822017-07-12 06:49:58 +0000963 if (HasExtLeaf1 && ((ECX >> 16) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000964 Features |= 1 << X86::FEATURE_FMA4;
Craig Topperbb8c7992017-07-08 05:16:13 +0000965
Craig Topper3a5d0822017-07-12 06:49:58 +0000966 if (HasExtLeaf1 && ((EDX >> 29) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000967 Features2 |= 1 << (X86::FEATURE_EM64T - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000968
969 *FeaturesOut = Features;
970 *Features2Out = Features2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000971}
972
973StringRef sys::getHostCPUName() {
974 unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
975 unsigned MaxLeaf, Vendor;
976
977#if defined(__GNUC__) || defined(__clang__)
978 //FIXME: include cpuid.h from clang or copy __get_cpuid_max here
979 // and simplify it to not invoke __cpuid (like cpu_model.c in
980 // compiler-rt/lib/builtins/cpu_model.c?
981 // Opting for the second option.
982 if(!isCpuIdSupported())
983 return "generic";
984#endif
Craig Topperbb8c7992017-07-08 05:16:13 +0000985 if (getX86CpuIDAndInfo(0, &MaxLeaf, &Vendor, &ECX, &EDX) || MaxLeaf < 1)
Simon Pilgrima271c542017-05-03 15:42:29 +0000986 return "generic";
Craig Topperbb8c7992017-07-08 05:16:13 +0000987 getX86CpuIDAndInfo(0x1, &EAX, &EBX, &ECX, &EDX);
Simon Pilgrima271c542017-05-03 15:42:29 +0000988
989 unsigned Brand_id = EBX & 0xff;
990 unsigned Family = 0, Model = 0;
Craig Topper3a5d0822017-07-12 06:49:58 +0000991 unsigned Features = 0, Features2 = 0;
Simon Pilgrima271c542017-05-03 15:42:29 +0000992 detectX86FamilyModel(EAX, &Family, &Model);
Craig Topper3a5d0822017-07-12 06:49:58 +0000993 getAvailableFeatures(ECX, EDX, MaxLeaf, &Features, &Features2);
Simon Pilgrima271c542017-05-03 15:42:29 +0000994
Craig Topper741e7e62017-11-03 18:02:44 +0000995 unsigned Type = 0;
996 unsigned Subtype = 0;
Simon Pilgrima271c542017-05-03 15:42:29 +0000997
998 if (Vendor == SIG_INTEL) {
Craig Topper3a5d0822017-07-12 06:49:58 +0000999 getIntelProcessorTypeAndSubtype(Family, Model, Brand_id, Features,
1000 Features2, &Type, &Subtype);
Simon Pilgrima271c542017-05-03 15:42:29 +00001001 } else if (Vendor == SIG_AMD) {
1002 getAMDProcessorTypeAndSubtype(Family, Model, Features, &Type, &Subtype);
Simon Pilgrima271c542017-05-03 15:42:29 +00001003 }
Craig Topperc77d00e2017-11-10 17:10:57 +00001004
1005 // Check subtypes first since those are more specific.
1006#define X86_CPU_SUBTYPE(ARCHNAME, ENUM) \
1007 if (Subtype == X86::ENUM) \
1008 return ARCHNAME;
1009#include "llvm/Support/X86TargetParser.def"
1010
1011 // Now check types.
1012#define X86_CPU_SUBTYPE(ARCHNAME, ENUM) \
1013 if (Type == X86::ENUM) \
1014 return ARCHNAME;
1015#include "llvm/Support/X86TargetParser.def"
1016
Simon Pilgrima271c542017-05-03 15:42:29 +00001017 return "generic";
1018}
1019
1020#elif defined(__APPLE__) && (defined(__ppc__) || defined(__powerpc__))
1021StringRef sys::getHostCPUName() {
1022 host_basic_info_data_t hostInfo;
1023 mach_msg_type_number_t infoCount;
1024
1025 infoCount = HOST_BASIC_INFO_COUNT;
1026 host_info(mach_host_self(), HOST_BASIC_INFO, (host_info_t)&hostInfo,
1027 &infoCount);
1028
1029 if (hostInfo.cpu_type != CPU_TYPE_POWERPC)
1030 return "generic";
1031
1032 switch (hostInfo.cpu_subtype) {
1033 case CPU_SUBTYPE_POWERPC_601:
1034 return "601";
1035 case CPU_SUBTYPE_POWERPC_602:
1036 return "602";
1037 case CPU_SUBTYPE_POWERPC_603:
1038 return "603";
1039 case CPU_SUBTYPE_POWERPC_603e:
1040 return "603e";
1041 case CPU_SUBTYPE_POWERPC_603ev:
1042 return "603ev";
1043 case CPU_SUBTYPE_POWERPC_604:
1044 return "604";
1045 case CPU_SUBTYPE_POWERPC_604e:
1046 return "604e";
1047 case CPU_SUBTYPE_POWERPC_620:
1048 return "620";
1049 case CPU_SUBTYPE_POWERPC_750:
1050 return "750";
1051 case CPU_SUBTYPE_POWERPC_7400:
1052 return "7400";
1053 case CPU_SUBTYPE_POWERPC_7450:
1054 return "7450";
1055 case CPU_SUBTYPE_POWERPC_970:
1056 return "970";
1057 default:;
1058 }
1059
1060 return "generic";
1061}
1062#elif defined(__linux__) && (defined(__ppc__) || defined(__powerpc__))
1063StringRef sys::getHostCPUName() {
1064 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1065 const StringRef& Content = P ? P->getBuffer() : "";
1066 return detail::getHostCPUNameForPowerPC(Content);
1067}
1068#elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1069StringRef sys::getHostCPUName() {
1070 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1071 const StringRef& Content = P ? P->getBuffer() : "";
1072 return detail::getHostCPUNameForARM(Content);
1073}
1074#elif defined(__linux__) && defined(__s390x__)
1075StringRef sys::getHostCPUName() {
1076 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1077 const StringRef& Content = P ? P->getBuffer() : "";
1078 return detail::getHostCPUNameForS390x(Content);
1079}
1080#else
1081StringRef sys::getHostCPUName() { return "generic"; }
1082#endif
1083
1084#if defined(__linux__) && defined(__x86_64__)
1085// On Linux, the number of physical cores can be computed from /proc/cpuinfo,
1086// using the number of unique physical/core id pairs. The following
1087// implementation reads the /proc/cpuinfo format on an x86_64 system.
1088static int computeHostNumPhysicalCores() {
1089 // Read /proc/cpuinfo as a stream (until EOF reached). It cannot be
1090 // mmapped because it appears to have 0 size.
1091 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Text =
1092 llvm::MemoryBuffer::getFileAsStream("/proc/cpuinfo");
1093 if (std::error_code EC = Text.getError()) {
1094 llvm::errs() << "Can't read "
1095 << "/proc/cpuinfo: " << EC.message() << "\n";
1096 return -1;
1097 }
1098 SmallVector<StringRef, 8> strs;
1099 (*Text)->getBuffer().split(strs, "\n", /*MaxSplit=*/-1,
1100 /*KeepEmpty=*/false);
1101 int CurPhysicalId = -1;
1102 int CurCoreId = -1;
1103 SmallSet<std::pair<int, int>, 32> UniqueItems;
1104 for (auto &Line : strs) {
1105 Line = Line.trim();
1106 if (!Line.startswith("physical id") && !Line.startswith("core id"))
1107 continue;
1108 std::pair<StringRef, StringRef> Data = Line.split(':');
1109 auto Name = Data.first.trim();
1110 auto Val = Data.second.trim();
1111 if (Name == "physical id") {
1112 assert(CurPhysicalId == -1 &&
1113 "Expected a core id before seeing another physical id");
1114 Val.getAsInteger(10, CurPhysicalId);
1115 }
1116 if (Name == "core id") {
1117 assert(CurCoreId == -1 &&
1118 "Expected a physical id before seeing another core id");
1119 Val.getAsInteger(10, CurCoreId);
1120 }
1121 if (CurPhysicalId != -1 && CurCoreId != -1) {
1122 UniqueItems.insert(std::make_pair(CurPhysicalId, CurCoreId));
1123 CurPhysicalId = -1;
1124 CurCoreId = -1;
1125 }
1126 }
1127 return UniqueItems.size();
1128}
1129#elif defined(__APPLE__) && defined(__x86_64__)
1130#include <sys/param.h>
1131#include <sys/sysctl.h>
1132
1133// Gets the number of *physical cores* on the machine.
1134static int computeHostNumPhysicalCores() {
1135 uint32_t count;
1136 size_t len = sizeof(count);
1137 sysctlbyname("hw.physicalcpu", &count, &len, NULL, 0);
1138 if (count < 1) {
1139 int nm[2];
1140 nm[0] = CTL_HW;
1141 nm[1] = HW_AVAILCPU;
1142 sysctl(nm, 2, &count, &len, NULL, 0);
1143 if (count < 1)
1144 return -1;
1145 }
1146 return count;
1147}
1148#else
1149// On other systems, return -1 to indicate unknown.
1150static int computeHostNumPhysicalCores() { return -1; }
1151#endif
1152
1153int sys::getHostNumPhysicalCores() {
1154 static int NumCores = computeHostNumPhysicalCores();
1155 return NumCores;
1156}
1157
1158#if defined(__i386__) || defined(_M_IX86) || \
1159 defined(__x86_64__) || defined(_M_X64)
1160bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1161 unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
1162 unsigned MaxLevel;
1163 union {
1164 unsigned u[3];
1165 char c[12];
1166 } text;
1167
1168 if (getX86CpuIDAndInfo(0, &MaxLevel, text.u + 0, text.u + 2, text.u + 1) ||
1169 MaxLevel < 1)
1170 return false;
1171
1172 getX86CpuIDAndInfo(1, &EAX, &EBX, &ECX, &EDX);
1173
Craig Topper1af7e442017-11-19 23:30:22 +00001174 Features["cmov"] = (EDX >> 15) & 1;
1175 Features["mmx"] = (EDX >> 23) & 1;
1176 Features["sse"] = (EDX >> 25) & 1;
1177 Features["sse2"] = (EDX >> 26) & 1;
1178
1179 Features["sse3"] = (ECX >> 0) & 1;
1180 Features["pclmul"] = (ECX >> 1) & 1;
1181 Features["ssse3"] = (ECX >> 9) & 1;
1182 Features["cx16"] = (ECX >> 13) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001183 Features["sse4.1"] = (ECX >> 19) & 1;
1184 Features["sse4.2"] = (ECX >> 20) & 1;
Craig Topper1af7e442017-11-19 23:30:22 +00001185 Features["movbe"] = (ECX >> 22) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001186 Features["popcnt"] = (ECX >> 23) & 1;
Craig Topper1af7e442017-11-19 23:30:22 +00001187 Features["aes"] = (ECX >> 25) & 1;
1188 Features["rdrnd"] = (ECX >> 30) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001189
1190 // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
1191 // indicates that the AVX registers will be saved and restored on context
1192 // switch, then we have full AVX support.
1193 bool HasAVXSave = ((ECX >> 27) & 1) && ((ECX >> 28) & 1) &&
1194 !getX86XCR0(&EAX, &EDX) && ((EAX & 0x6) == 0x6);
Simon Pilgrima271c542017-05-03 15:42:29 +00001195 // AVX512 requires additional context to be saved by the OS.
1196 bool HasAVX512Save = HasAVXSave && ((EAX & 0xe0) == 0xe0);
1197
Craig Topper1af7e442017-11-19 23:30:22 +00001198 Features["avx"] = HasAVXSave;
1199 Features["fma"] = ((ECX >> 12) & 1) && HasAVXSave;
1200 // Only enable XSAVE if OS has enabled support for saving YMM state.
1201 Features["xsave"] = ((ECX >> 26) & 1) && HasAVXSave;
1202 Features["f16c"] = ((ECX >> 29) & 1) && HasAVXSave;
1203
Simon Pilgrima271c542017-05-03 15:42:29 +00001204 unsigned MaxExtLevel;
1205 getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
1206
1207 bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
1208 !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
Craig Topper1af7e442017-11-19 23:30:22 +00001209 Features["lzcnt"] = HasExtLeaf1 && ((ECX >> 5) & 1);
1210 Features["sse4a"] = HasExtLeaf1 && ((ECX >> 6) & 1);
1211 Features["prfchw"] = HasExtLeaf1 && ((ECX >> 8) & 1);
1212 Features["xop"] = HasExtLeaf1 && ((ECX >> 11) & 1) && HasAVXSave;
1213 Features["lwp"] = HasExtLeaf1 && ((ECX >> 15) & 1);
1214 Features["fma4"] = HasExtLeaf1 && ((ECX >> 16) & 1) && HasAVXSave;
1215 Features["tbm"] = HasExtLeaf1 && ((ECX >> 21) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001216 Features["mwaitx"] = HasExtLeaf1 && ((ECX >> 29) & 1);
1217
1218 bool HasExtLeaf8 = MaxExtLevel >= 0x80000008 &&
Craig Topperdcd69792017-11-19 23:49:19 +00001219 !getX86CpuIDAndInfo(0x80000008, &EAX, &EBX, &ECX, &EDX);
Simon Pilgrima271c542017-05-03 15:42:29 +00001220 Features["clzero"] = HasExtLeaf8 && ((EBX >> 0) & 1);
1221
1222 bool HasLeaf7 =
1223 MaxLevel >= 7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
1224
Craig Topper1af7e442017-11-19 23:30:22 +00001225 Features["fsgsbase"] = HasLeaf7 && ((EBX >> 0) & 1);
1226 Features["sgx"] = HasLeaf7 && ((EBX >> 2) & 1);
1227 Features["bmi"] = HasLeaf7 && ((EBX >> 3) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001228 // AVX2 is only supported if we have the OS save support from AVX.
Craig Topper1af7e442017-11-19 23:30:22 +00001229 Features["avx2"] = HasLeaf7 && ((EBX >> 5) & 1) && HasAVXSave;
1230 Features["bmi2"] = HasLeaf7 && ((EBX >> 8) & 1);
1231 Features["rtm"] = HasLeaf7 && ((EBX >> 11) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001232 // AVX512 is only supported if the OS supports the context save for it.
Craig Topper1af7e442017-11-19 23:30:22 +00001233 Features["avx512f"] = HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save;
1234 Features["avx512dq"] = HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save;
1235 Features["rdseed"] = HasLeaf7 && ((EBX >> 18) & 1);
1236 Features["adx"] = HasLeaf7 && ((EBX >> 19) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001237 Features["avx512ifma"] = HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save;
Craig Topper1af7e442017-11-19 23:30:22 +00001238 Features["clflushopt"] = HasLeaf7 && ((EBX >> 23) & 1);
1239 Features["clwb"] = HasLeaf7 && ((EBX >> 24) & 1);
1240 Features["avx512pf"] = HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save;
1241 Features["avx512er"] = HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save;
1242 Features["avx512cd"] = HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save;
1243 Features["sha"] = HasLeaf7 && ((EBX >> 29) & 1);
1244 Features["avx512bw"] = HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save;
1245 Features["avx512vl"] = HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save;
Simon Pilgrima271c542017-05-03 15:42:29 +00001246
Craig Topper1af7e442017-11-19 23:30:22 +00001247 Features["prefetchwt1"] = HasLeaf7 && ((ECX >> 0) & 1);
1248 Features["avx512vbmi"] = HasLeaf7 && ((ECX >> 1) & 1) && HasAVX512Save;
Craig Topper9b03f672017-11-21 18:50:41 +00001249 Features["pku"] = HasLeaf7 && ((ECX >> 4) & 1);
Coby Tayree71e37cc2017-11-21 09:48:44 +00001250 Features["avx512vbmi2"] = HasLeaf7 && ((ECX >> 6) & 1) && HasAVX512Save;
Oren Ben Simhonfa582b02017-11-26 13:02:45 +00001251 Features["shstk"] = HasLeaf7 && ((ECX >> 7) & 1);
Coby Tayreed8b17be2017-11-26 09:36:41 +00001252 Features["gfni"] = HasLeaf7 && ((ECX >> 8) & 1);
Craig Topper9b03f672017-11-21 18:50:41 +00001253 Features["vaes"] = HasLeaf7 && ((ECX >> 9) & 1) && HasAVXSave;
1254 Features["vpclmulqdq"] = HasLeaf7 && ((ECX >> 10) & 1) && HasAVXSave;
1255 Features["avx512vnni"] = HasLeaf7 && ((ECX >> 11) & 1) && HasAVX512Save;
1256 Features["avx512bitalg"] = HasLeaf7 && ((ECX >> 12) & 1) && HasAVX512Save;
Yonghong Songdc1dbf62017-08-23 04:25:57 +00001257 Features["avx512vpopcntdq"] = HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save;
Craig Topper84b26b92018-01-18 23:52:31 +00001258 Features["rdpid"] = HasLeaf7 && ((ECX >> 22) & 1);
1259
1260 Features["ibt"] = HasLeaf7 && ((EDX >> 20) & 1);
Oren Ben Simhonfa582b02017-11-26 13:02:45 +00001261
Simon Pilgrima271c542017-05-03 15:42:29 +00001262 bool HasLeafD = MaxLevel >= 0xd &&
1263 !getX86CpuIDAndInfoEx(0xd, 0x1, &EAX, &EBX, &ECX, &EDX);
1264
1265 // Only enable XSAVE if OS has enabled support for saving YMM state.
Craig Topper1af7e442017-11-19 23:30:22 +00001266 Features["xsaveopt"] = HasLeafD && ((EAX >> 0) & 1) && HasAVXSave;
1267 Features["xsavec"] = HasLeafD && ((EAX >> 1) & 1) && HasAVXSave;
1268 Features["xsaves"] = HasLeafD && ((EAX >> 3) & 1) && HasAVXSave;
Simon Pilgrima271c542017-05-03 15:42:29 +00001269
1270 return true;
1271}
1272#elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1273bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1274 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1275 if (!P)
1276 return false;
1277
1278 SmallVector<StringRef, 32> Lines;
1279 P->getBuffer().split(Lines, "\n");
1280
1281 SmallVector<StringRef, 32> CPUFeatures;
1282
1283 // Look for the CPU features.
1284 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
1285 if (Lines[I].startswith("Features")) {
1286 Lines[I].split(CPUFeatures, ' ');
1287 break;
1288 }
1289
1290#if defined(__aarch64__)
1291 // Keep track of which crypto features we have seen
1292 enum { CAP_AES = 0x1, CAP_PMULL = 0x2, CAP_SHA1 = 0x4, CAP_SHA2 = 0x8 };
1293 uint32_t crypto = 0;
1294#endif
1295
1296 for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
1297 StringRef LLVMFeatureStr = StringSwitch<StringRef>(CPUFeatures[I])
1298#if defined(__aarch64__)
1299 .Case("asimd", "neon")
1300 .Case("fp", "fp-armv8")
1301 .Case("crc32", "crc")
1302#else
1303 .Case("half", "fp16")
1304 .Case("neon", "neon")
1305 .Case("vfpv3", "vfp3")
1306 .Case("vfpv3d16", "d16")
1307 .Case("vfpv4", "vfp4")
1308 .Case("idiva", "hwdiv-arm")
1309 .Case("idivt", "hwdiv")
1310#endif
1311 .Default("");
1312
1313#if defined(__aarch64__)
1314 // We need to check crypto separately since we need all of the crypto
1315 // extensions to enable the subtarget feature
1316 if (CPUFeatures[I] == "aes")
1317 crypto |= CAP_AES;
1318 else if (CPUFeatures[I] == "pmull")
1319 crypto |= CAP_PMULL;
1320 else if (CPUFeatures[I] == "sha1")
1321 crypto |= CAP_SHA1;
1322 else if (CPUFeatures[I] == "sha2")
1323 crypto |= CAP_SHA2;
1324#endif
1325
1326 if (LLVMFeatureStr != "")
1327 Features[LLVMFeatureStr] = true;
1328 }
1329
1330#if defined(__aarch64__)
1331 // If we have all crypto bits we can add the feature
1332 if (crypto == (CAP_AES | CAP_PMULL | CAP_SHA1 | CAP_SHA2))
1333 Features["crypto"] = true;
1334#endif
1335
1336 return true;
1337}
1338#else
1339bool sys::getHostCPUFeatures(StringMap<bool> &Features) { return false; }
1340#endif
1341
1342std::string sys::getProcessTriple() {
Alex Lorenz3803df32017-07-07 09:53:47 +00001343 std::string TargetTripleString = updateTripleOSVersion(LLVM_HOST_TRIPLE);
1344 Triple PT(Triple::normalize(TargetTripleString));
Simon Pilgrima271c542017-05-03 15:42:29 +00001345
1346 if (sizeof(void *) == 8 && PT.isArch32Bit())
1347 PT = PT.get64BitArchVariant();
1348 if (sizeof(void *) == 4 && PT.isArch64Bit())
1349 PT = PT.get32BitArchVariant();
1350
1351 return PT.str();
1352}