blob: 656e257760806802409f5c703160090324c36eb4 [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
Craig Topper8665f592018-03-07 17:53:16 +000068StringRef sys::detail::getHostCPUNameForPowerPC(StringRef ProcCpuinfoContent) {
Simon Pilgrima271c542017-05-03 15:42:29 +000069 // Access to the Processor Version Register (PVR) on PowerPC is privileged,
70 // and so we must use an operating-system interface to determine the current
71 // processor type. On Linux, this is exposed through the /proc/cpuinfo file.
72 const char *generic = "generic";
73
74 // The cpu line is second (after the 'processor: 0' line), so if this
75 // buffer is too small then something has changed (or is wrong).
76 StringRef::const_iterator CPUInfoStart = ProcCpuinfoContent.begin();
77 StringRef::const_iterator CPUInfoEnd = ProcCpuinfoContent.end();
78
79 StringRef::const_iterator CIP = CPUInfoStart;
80
81 StringRef::const_iterator CPUStart = 0;
82 size_t CPULen = 0;
83
84 // We need to find the first line which starts with cpu, spaces, and a colon.
85 // After the colon, there may be some additional spaces and then the cpu type.
86 while (CIP < CPUInfoEnd && CPUStart == 0) {
87 if (CIP < CPUInfoEnd && *CIP == '\n')
88 ++CIP;
89
90 if (CIP < CPUInfoEnd && *CIP == 'c') {
91 ++CIP;
92 if (CIP < CPUInfoEnd && *CIP == 'p') {
93 ++CIP;
94 if (CIP < CPUInfoEnd && *CIP == 'u') {
95 ++CIP;
96 while (CIP < CPUInfoEnd && (*CIP == ' ' || *CIP == '\t'))
97 ++CIP;
98
99 if (CIP < CPUInfoEnd && *CIP == ':') {
100 ++CIP;
101 while (CIP < CPUInfoEnd && (*CIP == ' ' || *CIP == '\t'))
102 ++CIP;
103
104 if (CIP < CPUInfoEnd) {
105 CPUStart = CIP;
106 while (CIP < CPUInfoEnd && (*CIP != ' ' && *CIP != '\t' &&
107 *CIP != ',' && *CIP != '\n'))
108 ++CIP;
109 CPULen = CIP - CPUStart;
110 }
111 }
112 }
113 }
114 }
115
116 if (CPUStart == 0)
117 while (CIP < CPUInfoEnd && *CIP != '\n')
118 ++CIP;
119 }
120
121 if (CPUStart == 0)
122 return generic;
123
124 return StringSwitch<const char *>(StringRef(CPUStart, CPULen))
125 .Case("604e", "604e")
126 .Case("604", "604")
127 .Case("7400", "7400")
128 .Case("7410", "7400")
129 .Case("7447", "7400")
130 .Case("7455", "7450")
131 .Case("G4", "g4")
132 .Case("POWER4", "970")
133 .Case("PPC970FX", "970")
134 .Case("PPC970MP", "970")
135 .Case("G5", "g5")
136 .Case("POWER5", "g5")
137 .Case("A2", "a2")
138 .Case("POWER6", "pwr6")
139 .Case("POWER7", "pwr7")
140 .Case("POWER8", "pwr8")
141 .Case("POWER8E", "pwr8")
142 .Case("POWER8NVL", "pwr8")
143 .Case("POWER9", "pwr9")
144 .Default(generic);
145}
146
Craig Topper8665f592018-03-07 17:53:16 +0000147StringRef sys::detail::getHostCPUNameForARM(StringRef ProcCpuinfoContent) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000148 // The cpuid register on arm is not accessible from user space. On Linux,
149 // it is exposed through the /proc/cpuinfo file.
150
151 // Read 32 lines from /proc/cpuinfo, which should contain the CPU part line
152 // in all cases.
153 SmallVector<StringRef, 32> Lines;
154 ProcCpuinfoContent.split(Lines, "\n");
155
156 // Look for the CPU implementer line.
157 StringRef Implementer;
158 StringRef Hardware;
159 for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
160 if (Lines[I].startswith("CPU implementer"))
161 Implementer = Lines[I].substr(15).ltrim("\t :");
162 if (Lines[I].startswith("Hardware"))
163 Hardware = Lines[I].substr(8).ltrim("\t :");
164 }
165
166 if (Implementer == "0x41") { // ARM Ltd.
167 // MSM8992/8994 may give cpu part for the core that the kernel is running on,
168 // which is undeterministic and wrong. Always return cortex-a53 for these SoC.
169 if (Hardware.endswith("MSM8994") || Hardware.endswith("MSM8996"))
170 return "cortex-a53";
171
172
173 // Look for the CPU part line.
174 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
175 if (Lines[I].startswith("CPU part"))
176 // The CPU part is a 3 digit hexadecimal number with a 0x prefix. The
177 // values correspond to the "Part number" in the CP15/c0 register. The
178 // contents are specified in the various processor manuals.
179 return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
180 .Case("0x926", "arm926ej-s")
181 .Case("0xb02", "mpcore")
182 .Case("0xb36", "arm1136j-s")
183 .Case("0xb56", "arm1156t2-s")
184 .Case("0xb76", "arm1176jz-s")
185 .Case("0xc08", "cortex-a8")
186 .Case("0xc09", "cortex-a9")
187 .Case("0xc0f", "cortex-a15")
188 .Case("0xc20", "cortex-m0")
189 .Case("0xc23", "cortex-m3")
190 .Case("0xc24", "cortex-m4")
191 .Case("0xd04", "cortex-a35")
192 .Case("0xd03", "cortex-a53")
193 .Case("0xd07", "cortex-a57")
194 .Case("0xd08", "cortex-a72")
195 .Case("0xd09", "cortex-a73")
196 .Default("generic");
197 }
198
199 if (Implementer == "0x51") // Qualcomm Technologies, Inc.
200 // Look for the CPU part line.
201 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
202 if (Lines[I].startswith("CPU part"))
203 // The CPU part is a 3 digit hexadecimal number with a 0x prefix. The
204 // values correspond to the "Part number" in the CP15/c0 register. The
205 // contents are specified in the various processor manuals.
206 return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
207 .Case("0x06f", "krait") // APQ8064
208 .Case("0x201", "kryo")
209 .Case("0x205", "kryo")
Eli Friedmanbde9fc72017-09-13 21:48:00 +0000210 .Case("0x211", "kryo")
211 .Case("0x800", "cortex-a73")
212 .Case("0x801", "cortex-a73")
Balaram Makama1e7ecc72017-09-22 17:46:36 +0000213 .Case("0xc00", "falkor")
Chad Rosier71070852017-09-25 14:05:00 +0000214 .Case("0xc01", "saphira")
Simon Pilgrima271c542017-05-03 15:42:29 +0000215 .Default("generic");
216
Evandro Menezes5d7a9e62017-12-08 21:09:59 +0000217 if (Implementer == "0x53") { // Samsung Electronics Co., Ltd.
218 // The Exynos chips have a convoluted ID scheme that doesn't seem to follow
219 // any predictive pattern across variants and parts.
220 unsigned Variant = 0, Part = 0;
221
222 // Look for the CPU variant line, whose value is a 1 digit hexadecimal
223 // number, corresponding to the Variant bits in the CP15/C0 register.
224 for (auto I : Lines)
225 if (I.consume_front("CPU variant"))
226 I.ltrim("\t :").getAsInteger(0, Variant);
227
228 // Look for the CPU part line, whose value is a 3 digit hexadecimal
229 // number, corresponding to the PartNum bits in the CP15/C0 register.
230 for (auto I : Lines)
231 if (I.consume_front("CPU part"))
232 I.ltrim("\t :").getAsInteger(0, Part);
233
234 unsigned Exynos = (Variant << 12) | Part;
235 switch (Exynos) {
236 default:
237 // Default by falling through to Exynos M1.
238 LLVM_FALLTHROUGH;
239
240 case 0x1001:
241 return "exynos-m1";
242
243 case 0x4001:
244 return "exynos-m2";
245 }
246 }
247
Simon Pilgrima271c542017-05-03 15:42:29 +0000248 return "generic";
249}
250
Craig Topper8665f592018-03-07 17:53:16 +0000251StringRef sys::detail::getHostCPUNameForS390x(StringRef ProcCpuinfoContent) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000252 // STIDP is a privileged operation, so use /proc/cpuinfo instead.
253
254 // The "processor 0:" line comes after a fair amount of other information,
255 // including a cache breakdown, but this should be plenty.
256 SmallVector<StringRef, 32> Lines;
257 ProcCpuinfoContent.split(Lines, "\n");
258
259 // Look for the CPU features.
260 SmallVector<StringRef, 32> CPUFeatures;
261 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
262 if (Lines[I].startswith("features")) {
263 size_t Pos = Lines[I].find(":");
264 if (Pos != StringRef::npos) {
265 Lines[I].drop_front(Pos + 1).split(CPUFeatures, ' ');
266 break;
267 }
268 }
269
270 // We need to check for the presence of vector support independently of
271 // the machine type, since we may only use the vector register set when
272 // supported by the kernel (and hypervisor).
273 bool HaveVectorSupport = false;
274 for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
275 if (CPUFeatures[I] == "vx")
276 HaveVectorSupport = true;
277 }
278
279 // Now check the processor machine type.
280 for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
281 if (Lines[I].startswith("processor ")) {
282 size_t Pos = Lines[I].find("machine = ");
283 if (Pos != StringRef::npos) {
284 Pos += sizeof("machine = ") - 1;
285 unsigned int Id;
286 if (!Lines[I].drop_front(Pos).getAsInteger(10, Id)) {
Ulrich Weigand2b3482f2017-07-17 17:41:11 +0000287 if (Id >= 3906 && HaveVectorSupport)
288 return "z14";
Simon Pilgrima271c542017-05-03 15:42:29 +0000289 if (Id >= 2964 && HaveVectorSupport)
290 return "z13";
291 if (Id >= 2827)
292 return "zEC12";
293 if (Id >= 2817)
294 return "z196";
295 }
296 }
297 break;
298 }
299 }
300
301 return "generic";
302}
303
Yonghong Songdc1dbf62017-08-23 04:25:57 +0000304StringRef sys::detail::getHostCPUNameForBPF() {
305#if !defined(__linux__) || !defined(__x86_64__)
306 return "generic";
307#else
308 uint8_t insns[40] __attribute__ ((aligned (8))) =
309 /* BPF_MOV64_IMM(BPF_REG_0, 0) */
310 { 0xb7, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
311 /* BPF_MOV64_IMM(BPF_REG_2, 1) */
312 0xb7, 0x2, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
313 /* BPF_JMP_REG(BPF_JLT, BPF_REG_0, BPF_REG_2, 1) */
314 0xad, 0x20, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0,
315 /* BPF_MOV64_IMM(BPF_REG_0, 1) */
316 0xb7, 0x0, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
317 /* BPF_EXIT_INSN() */
318 0x95, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0 };
319
320 struct bpf_prog_load_attr {
321 uint32_t prog_type;
322 uint32_t insn_cnt;
323 uint64_t insns;
324 uint64_t license;
325 uint32_t log_level;
326 uint32_t log_size;
327 uint64_t log_buf;
328 uint32_t kern_version;
329 uint32_t prog_flags;
330 } attr = {};
331 attr.prog_type = 1; /* BPF_PROG_TYPE_SOCKET_FILTER */
332 attr.insn_cnt = 5;
333 attr.insns = (uint64_t)insns;
334 attr.license = (uint64_t)"DUMMY";
335
336 int fd = syscall(321 /* __NR_bpf */, 5 /* BPF_PROG_LOAD */, &attr, sizeof(attr));
Yonghong Songc6d25712017-08-23 16:24:31 +0000337 if (fd >= 0) {
338 close(fd);
339 return "v2";
340 }
341 return "v1";
Yonghong Songdc1dbf62017-08-23 04:25:57 +0000342#endif
343}
344
Simon Pilgrima271c542017-05-03 15:42:29 +0000345#if defined(__i386__) || defined(_M_IX86) || \
346 defined(__x86_64__) || defined(_M_X64)
347
348enum VendorSignatures {
349 SIG_INTEL = 0x756e6547 /* Genu */,
350 SIG_AMD = 0x68747541 /* Auth */
351};
352
Simon Pilgrima271c542017-05-03 15:42:29 +0000353// The check below for i386 was copied from clang's cpuid.h (__get_cpuid_max).
354// Check motivated by bug reports for OpenSSL crashing on CPUs without CPUID
355// support. Consequently, for i386, the presence of CPUID is checked first
356// via the corresponding eflags bit.
357// Removal of cpuid.h header motivated by PR30384
358// Header cpuid.h and method __get_cpuid_max are not used in llvm, clang, openmp
359// or test-suite, but are used in external projects e.g. libstdcxx
360static bool isCpuIdSupported() {
361#if defined(__GNUC__) || defined(__clang__)
362#if defined(__i386__)
363 int __cpuid_supported;
364 __asm__(" pushfl\n"
365 " popl %%eax\n"
366 " movl %%eax,%%ecx\n"
367 " xorl $0x00200000,%%eax\n"
368 " pushl %%eax\n"
369 " popfl\n"
370 " pushfl\n"
371 " popl %%eax\n"
372 " movl $0,%0\n"
373 " cmpl %%eax,%%ecx\n"
374 " je 1f\n"
375 " movl $1,%0\n"
376 "1:"
377 : "=r"(__cpuid_supported)
378 :
379 : "eax", "ecx");
380 if (!__cpuid_supported)
381 return false;
382#endif
383 return true;
384#endif
385 return true;
386}
387
388/// getX86CpuIDAndInfo - Execute the specified cpuid and return the 4 values in
389/// the specified arguments. If we can't run cpuid on the host, return true.
390static bool getX86CpuIDAndInfo(unsigned value, unsigned *rEAX, unsigned *rEBX,
391 unsigned *rECX, unsigned *rEDX) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000392#if defined(__GNUC__) || defined(__clang__)
393#if defined(__x86_64__)
394 // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
395 // FIXME: should we save this for Clang?
396 __asm__("movq\t%%rbx, %%rsi\n\t"
397 "cpuid\n\t"
398 "xchgq\t%%rbx, %%rsi\n\t"
399 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
400 : "a"(value));
Craig Topper1efd10a2017-07-10 06:04:11 +0000401 return false;
Simon Pilgrima271c542017-05-03 15:42:29 +0000402#elif defined(__i386__)
403 __asm__("movl\t%%ebx, %%esi\n\t"
404 "cpuid\n\t"
405 "xchgl\t%%ebx, %%esi\n\t"
406 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
407 : "a"(value));
Craig Topper1efd10a2017-07-10 06:04:11 +0000408 return false;
Simon Pilgrima271c542017-05-03 15:42:29 +0000409#else
Craig Topper1efd10a2017-07-10 06:04:11 +0000410 return true;
Simon Pilgrima271c542017-05-03 15:42:29 +0000411#endif
412#elif defined(_MSC_VER)
413 // The MSVC intrinsic is portable across x86 and x64.
414 int registers[4];
415 __cpuid(registers, value);
416 *rEAX = registers[0];
417 *rEBX = registers[1];
418 *rECX = registers[2];
419 *rEDX = registers[3];
Simon Pilgrima271c542017-05-03 15:42:29 +0000420 return false;
421#else
422 return true;
423#endif
424}
425
426/// getX86CpuIDAndInfoEx - Execute the specified cpuid with subleaf and return
427/// the 4 values in the specified arguments. If we can't run cpuid on the host,
428/// return true.
429static bool getX86CpuIDAndInfoEx(unsigned value, unsigned subleaf,
430 unsigned *rEAX, unsigned *rEBX, unsigned *rECX,
431 unsigned *rEDX) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000432#if defined(__GNUC__) || defined(__clang__)
Craig Topper828cf302017-07-17 05:16:16 +0000433#if defined(__x86_64__)
Craig Topperada983a2017-07-10 06:09:22 +0000434 // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
Simon Pilgrima271c542017-05-03 15:42:29 +0000435 // FIXME: should we save this for Clang?
436 __asm__("movq\t%%rbx, %%rsi\n\t"
437 "cpuid\n\t"
438 "xchgq\t%%rbx, %%rsi\n\t"
439 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
440 : "a"(value), "c"(subleaf));
Craig Topper1efd10a2017-07-10 06:04:11 +0000441 return false;
Craig Topper828cf302017-07-17 05:16:16 +0000442#elif defined(__i386__)
443 __asm__("movl\t%%ebx, %%esi\n\t"
444 "cpuid\n\t"
445 "xchgl\t%%ebx, %%esi\n\t"
446 : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
447 : "a"(value), "c"(subleaf));
448 return false;
449#else
450 return true;
451#endif
Simon Pilgrima271c542017-05-03 15:42:29 +0000452#elif defined(_MSC_VER)
453 int registers[4];
454 __cpuidex(registers, value, subleaf);
455 *rEAX = registers[0];
456 *rEBX = registers[1];
457 *rECX = registers[2];
458 *rEDX = registers[3];
Craig Topper1efd10a2017-07-10 06:04:11 +0000459 return false;
460#else
461 return true;
Simon Pilgrima271c542017-05-03 15:42:29 +0000462#endif
Simon Pilgrima271c542017-05-03 15:42:29 +0000463}
464
Craig Topperf3af64e2017-07-12 06:49:57 +0000465// Read control register 0 (XCR0). Used to detect features such as AVX.
Simon Pilgrima271c542017-05-03 15:42:29 +0000466static bool getX86XCR0(unsigned *rEAX, unsigned *rEDX) {
467#if defined(__GNUC__) || defined(__clang__)
468 // Check xgetbv; this uses a .byte sequence instead of the instruction
469 // directly because older assemblers do not include support for xgetbv and
470 // there is no easy way to conditionally compile based on the assembler used.
471 __asm__(".byte 0x0f, 0x01, 0xd0" : "=a"(*rEAX), "=d"(*rEDX) : "c"(0));
472 return false;
473#elif defined(_MSC_FULL_VER) && defined(_XCR_XFEATURE_ENABLED_MASK)
474 unsigned long long Result = _xgetbv(_XCR_XFEATURE_ENABLED_MASK);
475 *rEAX = Result;
476 *rEDX = Result >> 32;
477 return false;
478#else
479 return true;
480#endif
481}
482
483static void detectX86FamilyModel(unsigned EAX, unsigned *Family,
484 unsigned *Model) {
485 *Family = (EAX >> 8) & 0xf; // Bits 8 - 11
486 *Model = (EAX >> 4) & 0xf; // Bits 4 - 7
487 if (*Family == 6 || *Family == 0xf) {
488 if (*Family == 0xf)
489 // Examine extended family ID if family ID is F.
490 *Family += (EAX >> 20) & 0xff; // Bits 20 - 27
491 // Examine extended model ID if family ID is 6 or F.
492 *Model += ((EAX >> 16) & 0xf) << 4; // Bits 16 - 19
493 }
494}
495
496static void
Craig Topperc6bbe4b2017-07-08 05:16:14 +0000497getIntelProcessorTypeAndSubtype(unsigned Family, unsigned Model,
498 unsigned Brand_id, unsigned Features,
Craig Topper3a5d0822017-07-12 06:49:58 +0000499 unsigned Features2, unsigned *Type,
500 unsigned *Subtype) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000501 if (Brand_id != 0)
502 return;
503 switch (Family) {
504 case 3:
Craig Topperc77d00e2017-11-10 17:10:57 +0000505 *Type = X86::INTEL_i386;
Simon Pilgrima271c542017-05-03 15:42:29 +0000506 break;
507 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000508 *Type = X86::INTEL_i486;
Simon Pilgrima271c542017-05-03 15:42:29 +0000509 break;
510 case 5:
Craig Topper47c87392017-11-21 23:36:42 +0000511 if (Features & (1 << X86::FEATURE_MMX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000512 *Type = X86::INTEL_PENTIUM_MMX;
Simon Pilgrima271c542017-05-03 15:42:29 +0000513 break;
514 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000515 *Type = X86::INTEL_PENTIUM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000516 break;
517 case 6:
518 switch (Model) {
519 case 0x01: // Pentium Pro processor
Craig Topperc77d00e2017-11-10 17:10:57 +0000520 *Type = X86::INTEL_PENTIUM_PRO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000521 break;
522 case 0x03: // Intel Pentium II OverDrive processor, Pentium II processor,
523 // model 03
524 case 0x05: // Pentium II processor, model 05, Pentium II Xeon processor,
525 // model 05, and Intel Celeron processor, model 05
526 case 0x06: // Celeron processor, model 06
Craig Topperc77d00e2017-11-10 17:10:57 +0000527 *Type = X86::INTEL_PENTIUM_II;
Simon Pilgrima271c542017-05-03 15:42:29 +0000528 break;
529 case 0x07: // Pentium III processor, model 07, and Pentium III Xeon
530 // processor, model 07
531 case 0x08: // Pentium III processor, model 08, Pentium III Xeon processor,
532 // model 08, and Celeron processor, model 08
533 case 0x0a: // Pentium III Xeon processor, model 0Ah
534 case 0x0b: // Pentium III processor, model 0Bh
Craig Topperc77d00e2017-11-10 17:10:57 +0000535 *Type = X86::INTEL_PENTIUM_III;
Simon Pilgrima271c542017-05-03 15:42:29 +0000536 break;
537 case 0x09: // Intel Pentium M processor, Intel Celeron M processor model 09.
538 case 0x0d: // Intel Pentium M processor, Intel Celeron M processor, model
539 // 0Dh. All processors are manufactured using the 90 nm process.
540 case 0x15: // Intel EP80579 Integrated Processor and Intel EP80579
541 // Integrated Processor with Intel QuickAssist Technology
Craig Topperc77d00e2017-11-10 17:10:57 +0000542 *Type = X86::INTEL_PENTIUM_M;
Simon Pilgrima271c542017-05-03 15:42:29 +0000543 break;
544 case 0x0e: // Intel Core Duo processor, Intel Core Solo processor, model
545 // 0Eh. All processors are manufactured using the 65 nm process.
Craig Topperc77d00e2017-11-10 17:10:57 +0000546 *Type = X86::INTEL_CORE_DUO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000547 break; // yonah
548 case 0x0f: // Intel Core 2 Duo processor, Intel Core 2 Duo mobile
549 // processor, Intel Core 2 Quad processor, Intel Core 2 Quad
550 // mobile processor, Intel Core 2 Extreme processor, Intel
551 // Pentium Dual-Core processor, Intel Xeon processor, model
552 // 0Fh. All processors are manufactured using the 65 nm process.
553 case 0x16: // Intel Celeron processor model 16h. All processors are
554 // manufactured using the 65 nm process
Craig Topperc77d00e2017-11-10 17:10:57 +0000555 *Type = X86::INTEL_CORE2; // "core2"
556 *Subtype = X86::INTEL_CORE2_65;
Simon Pilgrima271c542017-05-03 15:42:29 +0000557 break;
558 case 0x17: // Intel Core 2 Extreme processor, Intel Xeon processor, model
559 // 17h. All processors are manufactured using the 45 nm process.
560 //
561 // 45nm: Penryn , Wolfdale, Yorkfield (XE)
562 case 0x1d: // Intel Xeon processor MP. All processors are manufactured using
563 // the 45 nm process.
Craig Topperc77d00e2017-11-10 17:10:57 +0000564 *Type = X86::INTEL_CORE2; // "penryn"
565 *Subtype = X86::INTEL_CORE2_45;
Simon Pilgrima271c542017-05-03 15:42:29 +0000566 break;
567 case 0x1a: // Intel Core i7 processor and Intel Xeon processor. All
568 // processors are manufactured using the 45 nm process.
569 case 0x1e: // Intel(R) Core(TM) i7 CPU 870 @ 2.93GHz.
570 // As found in a Summer 2010 model iMac.
571 case 0x1f:
572 case 0x2e: // Nehalem EX
Craig Topperc77d00e2017-11-10 17:10:57 +0000573 *Type = X86::INTEL_COREI7; // "nehalem"
574 *Subtype = X86::INTEL_COREI7_NEHALEM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000575 break;
576 case 0x25: // Intel Core i7, laptop version.
577 case 0x2c: // Intel Core i7 processor and Intel Xeon processor. All
578 // processors are manufactured using the 32 nm process.
579 case 0x2f: // Westmere EX
Craig Topperc77d00e2017-11-10 17:10:57 +0000580 *Type = X86::INTEL_COREI7; // "westmere"
581 *Subtype = X86::INTEL_COREI7_WESTMERE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000582 break;
583 case 0x2a: // Intel Core i7 processor. All processors are manufactured
584 // using the 32 nm process.
585 case 0x2d:
Craig Topperc77d00e2017-11-10 17:10:57 +0000586 *Type = X86::INTEL_COREI7; //"sandybridge"
587 *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000588 break;
589 case 0x3a:
590 case 0x3e: // Ivy Bridge EP
Craig Topperc77d00e2017-11-10 17:10:57 +0000591 *Type = X86::INTEL_COREI7; // "ivybridge"
592 *Subtype = X86::INTEL_COREI7_IVYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000593 break;
594
595 // Haswell:
596 case 0x3c:
597 case 0x3f:
598 case 0x45:
599 case 0x46:
Craig Topperc77d00e2017-11-10 17:10:57 +0000600 *Type = X86::INTEL_COREI7; // "haswell"
601 *Subtype = X86::INTEL_COREI7_HASWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000602 break;
603
604 // Broadwell:
605 case 0x3d:
606 case 0x47:
607 case 0x4f:
608 case 0x56:
Craig Topperc77d00e2017-11-10 17:10:57 +0000609 *Type = X86::INTEL_COREI7; // "broadwell"
610 *Subtype = X86::INTEL_COREI7_BROADWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000611 break;
612
613 // Skylake:
614 case 0x4e: // Skylake mobile
615 case 0x5e: // Skylake desktop
616 case 0x8e: // Kaby Lake mobile
617 case 0x9e: // Kaby Lake desktop
Craig Topperc77d00e2017-11-10 17:10:57 +0000618 *Type = X86::INTEL_COREI7; // "skylake"
619 *Subtype = X86::INTEL_COREI7_SKYLAKE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000620 break;
621
622 // Skylake Xeon:
623 case 0x55:
Craig Topperc77d00e2017-11-10 17:10:57 +0000624 *Type = X86::INTEL_COREI7;
625 *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512; // "skylake-avx512"
Simon Pilgrima271c542017-05-03 15:42:29 +0000626 break;
627
Craig Topper07491862017-11-15 06:02:42 +0000628 // Cannonlake:
629 case 0x66:
630 *Type = X86::INTEL_COREI7;
631 *Subtype = X86::INTEL_COREI7_CANNONLAKE; // "cannonlake"
632 break;
633
Simon Pilgrima271c542017-05-03 15:42:29 +0000634 case 0x1c: // Most 45 nm Intel Atom processors
635 case 0x26: // 45 nm Atom Lincroft
636 case 0x27: // 32 nm Atom Medfield
637 case 0x35: // 32 nm Atom Midview
638 case 0x36: // 32 nm Atom Midview
Craig Topperc77d00e2017-11-10 17:10:57 +0000639 *Type = X86::INTEL_BONNELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000640 break; // "bonnell"
641
642 // Atom Silvermont codes from the Intel software optimization guide.
643 case 0x37:
644 case 0x4a:
645 case 0x4d:
646 case 0x5a:
647 case 0x5d:
648 case 0x4c: // really airmont
Craig Topperc77d00e2017-11-10 17:10:57 +0000649 *Type = X86::INTEL_SILVERMONT;
Simon Pilgrima271c542017-05-03 15:42:29 +0000650 break; // "silvermont"
Michael Zuckerman4bcb9c32017-06-29 10:00:33 +0000651 // Goldmont:
Craig Topper0dadfe32017-11-15 06:02:43 +0000652 case 0x5c: // Apollo Lake
653 case 0x5f: // Denverton
654 case 0x7a: // Gemini Lake
Craig Topperc77d00e2017-11-10 17:10:57 +0000655 *Type = X86::INTEL_GOLDMONT;
Michael Zuckerman4bcb9c32017-06-29 10:00:33 +0000656 break; // "goldmont"
Simon Pilgrima271c542017-05-03 15:42:29 +0000657 case 0x57:
Craig Topperc77d00e2017-11-10 17:10:57 +0000658 *Type = X86::INTEL_KNL; // knl
Simon Pilgrima271c542017-05-03 15:42:29 +0000659 break;
Craig Topper5d692912017-10-13 18:10:17 +0000660 case 0x85:
Craig Topperc77d00e2017-11-10 17:10:57 +0000661 *Type = X86::INTEL_KNM; // knm
Craig Topper5d692912017-10-13 18:10:17 +0000662 break;
Simon Pilgrima271c542017-05-03 15:42:29 +0000663
664 default: // Unknown family 6 CPU, try to guess.
Craig Topper47c87392017-11-21 23:36:42 +0000665 if (Features & (1 << X86::FEATURE_AVX512VBMI)) {
Craig Topper07491862017-11-15 06:02:42 +0000666 *Type = X86::INTEL_COREI7;
667 *Subtype = X86::INTEL_COREI7_CANNONLAKE;
Craig Topper4eda7562017-07-27 03:26:52 +0000668 break;
669 }
Craig Topper07491862017-11-15 06:02:42 +0000670
Craig Topper47c87392017-11-21 23:36:42 +0000671 if (Features & (1 << X86::FEATURE_AVX512VL)) {
Craig Topper07491862017-11-15 06:02:42 +0000672 *Type = X86::INTEL_COREI7;
673 *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512;
674 break;
675 }
676
Craig Topper47c87392017-11-21 23:36:42 +0000677 if (Features & (1 << X86::FEATURE_AVX512ER)) {
Craig Topper07491862017-11-15 06:02:42 +0000678 *Type = X86::INTEL_KNL; // knl
679 break;
680 }
681
Craig Topper47c87392017-11-21 23:36:42 +0000682 if (Features2 & (1 << (X86::FEATURE_CLFLUSHOPT - 32))) {
683 if (Features2 & (1 << (X86::FEATURE_SHA - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000684 *Type = X86::INTEL_GOLDMONT;
Craig Topper4eda7562017-07-27 03:26:52 +0000685 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000686 *Type = X86::INTEL_COREI7;
687 *Subtype = X86::INTEL_COREI7_SKYLAKE;
Craig Topper4eda7562017-07-27 03:26:52 +0000688 }
Simon Pilgrima271c542017-05-03 15:42:29 +0000689 break;
690 }
Craig Topper47c87392017-11-21 23:36:42 +0000691 if (Features2 & (1 << (X86::FEATURE_ADX - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000692 *Type = X86::INTEL_COREI7;
693 *Subtype = X86::INTEL_COREI7_BROADWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000694 break;
695 }
Craig Topper47c87392017-11-21 23:36:42 +0000696 if (Features & (1 << X86::FEATURE_AVX2)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000697 *Type = X86::INTEL_COREI7;
698 *Subtype = X86::INTEL_COREI7_HASWELL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000699 break;
700 }
Craig Topper47c87392017-11-21 23:36:42 +0000701 if (Features & (1 << X86::FEATURE_AVX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000702 *Type = X86::INTEL_COREI7;
703 *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000704 break;
705 }
Craig Topper47c87392017-11-21 23:36:42 +0000706 if (Features & (1 << X86::FEATURE_SSE4_2)) {
707 if (Features2 & (1 << (X86::FEATURE_MOVBE - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000708 *Type = X86::INTEL_SILVERMONT;
Simon Pilgrima271c542017-05-03 15:42:29 +0000709 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000710 *Type = X86::INTEL_COREI7;
711 *Subtype = X86::INTEL_COREI7_NEHALEM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000712 }
713 break;
714 }
Craig Topper47c87392017-11-21 23:36:42 +0000715 if (Features & (1 << X86::FEATURE_SSE4_1)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000716 *Type = X86::INTEL_CORE2; // "penryn"
717 *Subtype = X86::INTEL_CORE2_45;
Simon Pilgrima271c542017-05-03 15:42:29 +0000718 break;
719 }
Craig Topper47c87392017-11-21 23:36:42 +0000720 if (Features & (1 << X86::FEATURE_SSSE3)) {
721 if (Features2 & (1 << (X86::FEATURE_MOVBE - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000722 *Type = X86::INTEL_BONNELL; // "bonnell"
Simon Pilgrima271c542017-05-03 15:42:29 +0000723 } else {
Craig Topperc77d00e2017-11-10 17:10:57 +0000724 *Type = X86::INTEL_CORE2; // "core2"
725 *Subtype = X86::INTEL_CORE2_65;
Simon Pilgrima271c542017-05-03 15:42:29 +0000726 }
727 break;
728 }
Craig Topper47c87392017-11-21 23:36:42 +0000729 if (Features2 & (1 << (X86::FEATURE_EM64T - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000730 *Type = X86::INTEL_CORE2; // "core2"
731 *Subtype = X86::INTEL_CORE2_65;
Craig Toppera233e162017-11-02 19:13:32 +0000732 break;
733 }
Craig Topper47c87392017-11-21 23:36:42 +0000734 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000735 *Type = X86::INTEL_CORE_DUO;
Craig Toppera233e162017-11-02 19:13:32 +0000736 break;
Simon Pilgrima271c542017-05-03 15:42:29 +0000737 }
Craig Topper47c87392017-11-21 23:36:42 +0000738 if (Features & (1 << X86::FEATURE_SSE2)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000739 *Type = X86::INTEL_PENTIUM_M;
Simon Pilgrima271c542017-05-03 15:42:29 +0000740 break;
741 }
Craig Topper47c87392017-11-21 23:36:42 +0000742 if (Features & (1 << X86::FEATURE_SSE)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000743 *Type = X86::INTEL_PENTIUM_III;
Simon Pilgrima271c542017-05-03 15:42:29 +0000744 break;
745 }
Craig Topper47c87392017-11-21 23:36:42 +0000746 if (Features & (1 << X86::FEATURE_MMX)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000747 *Type = X86::INTEL_PENTIUM_II;
Simon Pilgrima271c542017-05-03 15:42:29 +0000748 break;
749 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000750 *Type = X86::INTEL_PENTIUM_PRO;
Simon Pilgrima271c542017-05-03 15:42:29 +0000751 break;
752 }
753 break;
754 case 15: {
Craig Topper47c87392017-11-21 23:36:42 +0000755 if (Features2 & (1 << (X86::FEATURE_EM64T - 32))) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000756 *Type = X86::INTEL_NOCONA;
Simon Pilgrima271c542017-05-03 15:42:29 +0000757 break;
758 }
Craig Topper47c87392017-11-21 23:36:42 +0000759 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000760 *Type = X86::INTEL_PRESCOTT;
Craig Topper14949152017-11-02 19:13:34 +0000761 break;
762 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000763 *Type = X86::INTEL_PENTIUM_IV;
Simon Pilgrima271c542017-05-03 15:42:29 +0000764 break;
765 }
766 default:
767 break; /*"generic"*/
768 }
769}
770
Craig Topper2ace1532017-07-08 06:44:34 +0000771static void getAMDProcessorTypeAndSubtype(unsigned Family, unsigned Model,
772 unsigned Features, unsigned *Type,
Simon Pilgrima271c542017-05-03 15:42:29 +0000773 unsigned *Subtype) {
774 // FIXME: this poorly matches the generated SubtargetFeatureKV table. There
775 // appears to be no way to generate the wide variety of AMD-specific targets
776 // from the information returned from CPUID.
777 switch (Family) {
778 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000779 *Type = X86::AMD_i486;
Simon Pilgrima271c542017-05-03 15:42:29 +0000780 break;
781 case 5:
Craig Topperc77d00e2017-11-10 17:10:57 +0000782 *Type = X86::AMDPENTIUM;
Simon Pilgrima271c542017-05-03 15:42:29 +0000783 switch (Model) {
784 case 6:
785 case 7:
Craig Topperc77d00e2017-11-10 17:10:57 +0000786 *Subtype = X86::AMDPENTIUM_K6;
Simon Pilgrima271c542017-05-03 15:42:29 +0000787 break; // "k6"
788 case 8:
Craig Topperc77d00e2017-11-10 17:10:57 +0000789 *Subtype = X86::AMDPENTIUM_K62;
Simon Pilgrima271c542017-05-03 15:42:29 +0000790 break; // "k6-2"
791 case 9:
792 case 13:
Craig Topperc77d00e2017-11-10 17:10:57 +0000793 *Subtype = X86::AMDPENTIUM_K63;
Simon Pilgrima271c542017-05-03 15:42:29 +0000794 break; // "k6-3"
795 case 10:
Craig Topperc77d00e2017-11-10 17:10:57 +0000796 *Subtype = X86::AMDPENTIUM_GEODE;
Simon Pilgrima271c542017-05-03 15:42:29 +0000797 break; // "geode"
798 }
799 break;
800 case 6:
Craig Topper47c87392017-11-21 23:36:42 +0000801 if (Features & (1 << X86::FEATURE_SSE)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000802 *Type = X86::AMD_ATHLON_XP;
Simon Pilgrima271c542017-05-03 15:42:29 +0000803 break; // "athlon-xp"
804 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000805 *Type = X86::AMD_ATHLON;
Craig Topperf3de5eb2017-07-13 06:34:10 +0000806 break; // "athlon"
Simon Pilgrima271c542017-05-03 15:42:29 +0000807 case 15:
Craig Topper47c87392017-11-21 23:36:42 +0000808 if (Features & (1 << X86::FEATURE_SSE3)) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000809 *Type = X86::AMD_K8SSE3;
Simon Pilgrima271c542017-05-03 15:42:29 +0000810 break; // "k8-sse3"
811 }
Craig Topperc77d00e2017-11-10 17:10:57 +0000812 *Type = X86::AMD_K8;
Craig Topperf3de5eb2017-07-13 06:34:10 +0000813 break; // "k8"
Simon Pilgrima271c542017-05-03 15:42:29 +0000814 case 16:
Craig Topperc77d00e2017-11-10 17:10:57 +0000815 *Type = X86::AMDFAM10H; // "amdfam10"
Simon Pilgrima271c542017-05-03 15:42:29 +0000816 switch (Model) {
817 case 2:
Craig Topperc77d00e2017-11-10 17:10:57 +0000818 *Subtype = X86::AMDFAM10H_BARCELONA;
Simon Pilgrima271c542017-05-03 15:42:29 +0000819 break;
820 case 4:
Craig Topperc77d00e2017-11-10 17:10:57 +0000821 *Subtype = X86::AMDFAM10H_SHANGHAI;
Simon Pilgrima271c542017-05-03 15:42:29 +0000822 break;
823 case 8:
Craig Topperc77d00e2017-11-10 17:10:57 +0000824 *Subtype = X86::AMDFAM10H_ISTANBUL;
Simon Pilgrima271c542017-05-03 15:42:29 +0000825 break;
826 }
827 break;
828 case 20:
Craig Topperc77d00e2017-11-10 17:10:57 +0000829 *Type = X86::AMD_BTVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000830 break; // "btver1";
831 case 21:
Craig Topperc77d00e2017-11-10 17:10:57 +0000832 *Type = X86::AMDFAM15H;
Craig Topper1f9d3c02017-07-08 06:44:35 +0000833 if (Model >= 0x60 && Model <= 0x7f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000834 *Subtype = X86::AMDFAM15H_BDVER4;
Craig Topper3db11702017-07-12 06:49:56 +0000835 break; // "bdver4"; 60h-7Fh: Excavator
Simon Pilgrima271c542017-05-03 15:42:29 +0000836 }
837 if (Model >= 0x30 && Model <= 0x3f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000838 *Subtype = X86::AMDFAM15H_BDVER3;
Simon Pilgrima271c542017-05-03 15:42:29 +0000839 break; // "bdver3"; 30h-3Fh: Steamroller
840 }
841 if (Model >= 0x10 && Model <= 0x1f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000842 *Subtype = X86::AMDFAM15H_BDVER2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000843 break; // "bdver2"; 10h-1Fh: Piledriver
844 }
845 if (Model <= 0x0f) {
Craig Topperc77d00e2017-11-10 17:10:57 +0000846 *Subtype = X86::AMDFAM15H_BDVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000847 break; // "bdver1"; 00h-0Fh: Bulldozer
848 }
849 break;
850 case 22:
Craig Topperc77d00e2017-11-10 17:10:57 +0000851 *Type = X86::AMD_BTVER2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000852 break; // "btver2"
853 case 23:
Craig Topperc77d00e2017-11-10 17:10:57 +0000854 *Type = X86::AMDFAM17H;
855 *Subtype = X86::AMDFAM17H_ZNVER1;
Simon Pilgrima271c542017-05-03 15:42:29 +0000856 break;
857 default:
858 break; // "generic"
859 }
860}
861
Craig Topper3a5d0822017-07-12 06:49:58 +0000862static void getAvailableFeatures(unsigned ECX, unsigned EDX, unsigned MaxLeaf,
863 unsigned *FeaturesOut,
864 unsigned *Features2Out) {
Simon Pilgrima271c542017-05-03 15:42:29 +0000865 unsigned Features = 0;
Craig Topper3a5d0822017-07-12 06:49:58 +0000866 unsigned Features2 = 0;
Craig Topperc6bbe4b2017-07-08 05:16:14 +0000867 unsigned EAX, EBX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000868
869 if ((EDX >> 15) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000870 Features |= 1 << X86::FEATURE_CMOV;
Craig Topper3a5d0822017-07-12 06:49:58 +0000871 if ((EDX >> 23) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000872 Features |= 1 << X86::FEATURE_MMX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000873 if ((EDX >> 25) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000874 Features |= 1 << X86::FEATURE_SSE;
Craig Topper3a5d0822017-07-12 06:49:58 +0000875 if ((EDX >> 26) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000876 Features |= 1 << X86::FEATURE_SSE2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000877
878 if ((ECX >> 0) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000879 Features |= 1 << X86::FEATURE_SSE3;
Craig Topper3a5d0822017-07-12 06:49:58 +0000880 if ((ECX >> 1) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000881 Features |= 1 << X86::FEATURE_PCLMUL;
Craig Topper3a5d0822017-07-12 06:49:58 +0000882 if ((ECX >> 9) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000883 Features |= 1 << X86::FEATURE_SSSE3;
Craig Topper3a5d0822017-07-12 06:49:58 +0000884 if ((ECX >> 12) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000885 Features |= 1 << X86::FEATURE_FMA;
Craig Topper3a5d0822017-07-12 06:49:58 +0000886 if ((ECX >> 19) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000887 Features |= 1 << X86::FEATURE_SSE4_1;
Craig Topper3a5d0822017-07-12 06:49:58 +0000888 if ((ECX >> 20) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000889 Features |= 1 << X86::FEATURE_SSE4_2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000890 if ((ECX >> 23) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000891 Features |= 1 << X86::FEATURE_POPCNT;
Craig Topper3a5d0822017-07-12 06:49:58 +0000892 if ((ECX >> 25) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000893 Features |= 1 << X86::FEATURE_AES;
Craig Topper3a5d0822017-07-12 06:49:58 +0000894
895 if ((ECX >> 22) & 1)
Craig Topper47c87392017-11-21 23:36:42 +0000896 Features2 |= 1 << (X86::FEATURE_MOVBE - 32);
Simon Pilgrima271c542017-05-03 15:42:29 +0000897
898 // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
899 // indicates that the AVX registers will be saved and restored on context
900 // switch, then we have full AVX support.
901 const unsigned AVXBits = (1 << 27) | (1 << 28);
902 bool HasAVX = ((ECX & AVXBits) == AVXBits) && !getX86XCR0(&EAX, &EDX) &&
903 ((EAX & 0x6) == 0x6);
904 bool HasAVX512Save = HasAVX && ((EAX & 0xe0) == 0xe0);
Craig Topper3a5d0822017-07-12 06:49:58 +0000905
906 if (HasAVX)
Craig Topper47c87392017-11-21 23:36:42 +0000907 Features |= 1 << X86::FEATURE_AVX;
Craig Topper3a5d0822017-07-12 06:49:58 +0000908
Simon Pilgrima271c542017-05-03 15:42:29 +0000909 bool HasLeaf7 =
910 MaxLeaf >= 0x7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
Craig Topper3a5d0822017-07-12 06:49:58 +0000911
912 if (HasLeaf7 && ((EBX >> 3) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000913 Features |= 1 << X86::FEATURE_BMI;
Craig Topper3a5d0822017-07-12 06:49:58 +0000914 if (HasLeaf7 && ((EBX >> 5) & 1) && HasAVX)
Craig Topper47c87392017-11-21 23:36:42 +0000915 Features |= 1 << X86::FEATURE_AVX2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000916 if (HasLeaf7 && ((EBX >> 9) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000917 Features |= 1 << X86::FEATURE_BMI2;
Craig Topper3a5d0822017-07-12 06:49:58 +0000918 if (HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000919 Features |= 1 << X86::FEATURE_AVX512F;
Craig Topper3a5d0822017-07-12 06:49:58 +0000920 if (HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000921 Features |= 1 << X86::FEATURE_AVX512DQ;
Craig Topper3a5d0822017-07-12 06:49:58 +0000922 if (HasLeaf7 && ((EBX >> 19) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000923 Features2 |= 1 << (X86::FEATURE_ADX - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000924 if (HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000925 Features |= 1 << X86::FEATURE_AVX512IFMA;
Craig Topper4eda7562017-07-27 03:26:52 +0000926 if (HasLeaf7 && ((EBX >> 23) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000927 Features2 |= 1 << (X86::FEATURE_CLFLUSHOPT - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000928 if (HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000929 Features |= 1 << X86::FEATURE_AVX512PF;
Craig Topper3a5d0822017-07-12 06:49:58 +0000930 if (HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000931 Features |= 1 << X86::FEATURE_AVX512ER;
Craig Topper3a5d0822017-07-12 06:49:58 +0000932 if (HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000933 Features |= 1 << X86::FEATURE_AVX512CD;
Craig Topper4eda7562017-07-27 03:26:52 +0000934 if (HasLeaf7 && ((EBX >> 29) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000935 Features2 |= 1 << (X86::FEATURE_SHA - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000936 if (HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000937 Features |= 1 << X86::FEATURE_AVX512BW;
Craig Topper3a5d0822017-07-12 06:49:58 +0000938 if (HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000939 Features |= 1 << X86::FEATURE_AVX512VL;
Craig Topper3a5d0822017-07-12 06:49:58 +0000940
941 if (HasLeaf7 && ((ECX >> 1) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000942 Features |= 1 << X86::FEATURE_AVX512VBMI;
Craig Topper3a5d0822017-07-12 06:49:58 +0000943 if (HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000944 Features |= 1 << X86::FEATURE_AVX512VPOPCNTDQ;
Craig Topper3a5d0822017-07-12 06:49:58 +0000945
946 if (HasLeaf7 && ((EDX >> 2) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000947 Features |= 1 << X86::FEATURE_AVX5124VNNIW;
Craig Topper3a5d0822017-07-12 06:49:58 +0000948 if (HasLeaf7 && ((EDX >> 3) & 1) && HasAVX512Save)
Craig Topper47c87392017-11-21 23:36:42 +0000949 Features |= 1 << X86::FEATURE_AVX5124FMAPS;
Simon Pilgrima271c542017-05-03 15:42:29 +0000950
Craig Topperbb8c7992017-07-08 05:16:13 +0000951 unsigned MaxExtLevel;
952 getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
953
954 bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
955 !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
Craig Topper3a5d0822017-07-12 06:49:58 +0000956 if (HasExtLeaf1 && ((ECX >> 6) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000957 Features |= 1 << X86::FEATURE_SSE4_A;
Craig Topper3a5d0822017-07-12 06:49:58 +0000958 if (HasExtLeaf1 && ((ECX >> 11) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000959 Features |= 1 << X86::FEATURE_XOP;
Craig Topper3a5d0822017-07-12 06:49:58 +0000960 if (HasExtLeaf1 && ((ECX >> 16) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000961 Features |= 1 << X86::FEATURE_FMA4;
Craig Topperbb8c7992017-07-08 05:16:13 +0000962
Craig Topper3a5d0822017-07-12 06:49:58 +0000963 if (HasExtLeaf1 && ((EDX >> 29) & 1))
Craig Topper47c87392017-11-21 23:36:42 +0000964 Features2 |= 1 << (X86::FEATURE_EM64T - 32);
Craig Topper3a5d0822017-07-12 06:49:58 +0000965
966 *FeaturesOut = Features;
967 *Features2Out = Features2;
Simon Pilgrima271c542017-05-03 15:42:29 +0000968}
969
970StringRef sys::getHostCPUName() {
971 unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
972 unsigned MaxLeaf, Vendor;
973
974#if defined(__GNUC__) || defined(__clang__)
975 //FIXME: include cpuid.h from clang or copy __get_cpuid_max here
976 // and simplify it to not invoke __cpuid (like cpu_model.c in
977 // compiler-rt/lib/builtins/cpu_model.c?
978 // Opting for the second option.
979 if(!isCpuIdSupported())
980 return "generic";
981#endif
Craig Topperbb8c7992017-07-08 05:16:13 +0000982 if (getX86CpuIDAndInfo(0, &MaxLeaf, &Vendor, &ECX, &EDX) || MaxLeaf < 1)
Simon Pilgrima271c542017-05-03 15:42:29 +0000983 return "generic";
Craig Topperbb8c7992017-07-08 05:16:13 +0000984 getX86CpuIDAndInfo(0x1, &EAX, &EBX, &ECX, &EDX);
Simon Pilgrima271c542017-05-03 15:42:29 +0000985
986 unsigned Brand_id = EBX & 0xff;
987 unsigned Family = 0, Model = 0;
Craig Topper3a5d0822017-07-12 06:49:58 +0000988 unsigned Features = 0, Features2 = 0;
Simon Pilgrima271c542017-05-03 15:42:29 +0000989 detectX86FamilyModel(EAX, &Family, &Model);
Craig Topper3a5d0822017-07-12 06:49:58 +0000990 getAvailableFeatures(ECX, EDX, MaxLeaf, &Features, &Features2);
Simon Pilgrima271c542017-05-03 15:42:29 +0000991
Craig Topper741e7e62017-11-03 18:02:44 +0000992 unsigned Type = 0;
993 unsigned Subtype = 0;
Simon Pilgrima271c542017-05-03 15:42:29 +0000994
995 if (Vendor == SIG_INTEL) {
Craig Topper3a5d0822017-07-12 06:49:58 +0000996 getIntelProcessorTypeAndSubtype(Family, Model, Brand_id, Features,
997 Features2, &Type, &Subtype);
Simon Pilgrima271c542017-05-03 15:42:29 +0000998 } else if (Vendor == SIG_AMD) {
999 getAMDProcessorTypeAndSubtype(Family, Model, Features, &Type, &Subtype);
Simon Pilgrima271c542017-05-03 15:42:29 +00001000 }
Craig Topperc77d00e2017-11-10 17:10:57 +00001001
1002 // Check subtypes first since those are more specific.
1003#define X86_CPU_SUBTYPE(ARCHNAME, ENUM) \
1004 if (Subtype == X86::ENUM) \
1005 return ARCHNAME;
1006#include "llvm/Support/X86TargetParser.def"
1007
1008 // Now check types.
Craig Topper55ad3292018-03-06 22:45:31 +00001009#define X86_CPU_TYPE(ARCHNAME, ENUM) \
Craig Topperc77d00e2017-11-10 17:10:57 +00001010 if (Type == X86::ENUM) \
1011 return ARCHNAME;
1012#include "llvm/Support/X86TargetParser.def"
1013
Simon Pilgrima271c542017-05-03 15:42:29 +00001014 return "generic";
1015}
1016
1017#elif defined(__APPLE__) && (defined(__ppc__) || defined(__powerpc__))
1018StringRef sys::getHostCPUName() {
1019 host_basic_info_data_t hostInfo;
1020 mach_msg_type_number_t infoCount;
1021
1022 infoCount = HOST_BASIC_INFO_COUNT;
1023 host_info(mach_host_self(), HOST_BASIC_INFO, (host_info_t)&hostInfo,
1024 &infoCount);
1025
1026 if (hostInfo.cpu_type != CPU_TYPE_POWERPC)
1027 return "generic";
1028
1029 switch (hostInfo.cpu_subtype) {
1030 case CPU_SUBTYPE_POWERPC_601:
1031 return "601";
1032 case CPU_SUBTYPE_POWERPC_602:
1033 return "602";
1034 case CPU_SUBTYPE_POWERPC_603:
1035 return "603";
1036 case CPU_SUBTYPE_POWERPC_603e:
1037 return "603e";
1038 case CPU_SUBTYPE_POWERPC_603ev:
1039 return "603ev";
1040 case CPU_SUBTYPE_POWERPC_604:
1041 return "604";
1042 case CPU_SUBTYPE_POWERPC_604e:
1043 return "604e";
1044 case CPU_SUBTYPE_POWERPC_620:
1045 return "620";
1046 case CPU_SUBTYPE_POWERPC_750:
1047 return "750";
1048 case CPU_SUBTYPE_POWERPC_7400:
1049 return "7400";
1050 case CPU_SUBTYPE_POWERPC_7450:
1051 return "7450";
1052 case CPU_SUBTYPE_POWERPC_970:
1053 return "970";
1054 default:;
1055 }
1056
1057 return "generic";
1058}
1059#elif defined(__linux__) && (defined(__ppc__) || defined(__powerpc__))
1060StringRef sys::getHostCPUName() {
1061 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
Craig Topper8665f592018-03-07 17:53:16 +00001062 StringRef Content = P ? P->getBuffer() : "";
Simon Pilgrima271c542017-05-03 15:42:29 +00001063 return detail::getHostCPUNameForPowerPC(Content);
1064}
1065#elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1066StringRef sys::getHostCPUName() {
1067 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
Craig Topper8665f592018-03-07 17:53:16 +00001068 StringRef Content = P ? P->getBuffer() : "";
Simon Pilgrima271c542017-05-03 15:42:29 +00001069 return detail::getHostCPUNameForARM(Content);
1070}
1071#elif defined(__linux__) && defined(__s390x__)
1072StringRef sys::getHostCPUName() {
1073 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
Craig Topper8665f592018-03-07 17:53:16 +00001074 StringRef Content = P ? P->getBuffer() : "";
Simon Pilgrima271c542017-05-03 15:42:29 +00001075 return detail::getHostCPUNameForS390x(Content);
1076}
1077#else
1078StringRef sys::getHostCPUName() { return "generic"; }
1079#endif
1080
1081#if defined(__linux__) && defined(__x86_64__)
1082// On Linux, the number of physical cores can be computed from /proc/cpuinfo,
1083// using the number of unique physical/core id pairs. The following
1084// implementation reads the /proc/cpuinfo format on an x86_64 system.
1085static int computeHostNumPhysicalCores() {
1086 // Read /proc/cpuinfo as a stream (until EOF reached). It cannot be
1087 // mmapped because it appears to have 0 size.
1088 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Text =
1089 llvm::MemoryBuffer::getFileAsStream("/proc/cpuinfo");
1090 if (std::error_code EC = Text.getError()) {
1091 llvm::errs() << "Can't read "
1092 << "/proc/cpuinfo: " << EC.message() << "\n";
1093 return -1;
1094 }
1095 SmallVector<StringRef, 8> strs;
1096 (*Text)->getBuffer().split(strs, "\n", /*MaxSplit=*/-1,
1097 /*KeepEmpty=*/false);
1098 int CurPhysicalId = -1;
1099 int CurCoreId = -1;
1100 SmallSet<std::pair<int, int>, 32> UniqueItems;
1101 for (auto &Line : strs) {
1102 Line = Line.trim();
1103 if (!Line.startswith("physical id") && !Line.startswith("core id"))
1104 continue;
1105 std::pair<StringRef, StringRef> Data = Line.split(':');
1106 auto Name = Data.first.trim();
1107 auto Val = Data.second.trim();
1108 if (Name == "physical id") {
1109 assert(CurPhysicalId == -1 &&
1110 "Expected a core id before seeing another physical id");
1111 Val.getAsInteger(10, CurPhysicalId);
1112 }
1113 if (Name == "core id") {
1114 assert(CurCoreId == -1 &&
1115 "Expected a physical id before seeing another core id");
1116 Val.getAsInteger(10, CurCoreId);
1117 }
1118 if (CurPhysicalId != -1 && CurCoreId != -1) {
1119 UniqueItems.insert(std::make_pair(CurPhysicalId, CurCoreId));
1120 CurPhysicalId = -1;
1121 CurCoreId = -1;
1122 }
1123 }
1124 return UniqueItems.size();
1125}
1126#elif defined(__APPLE__) && defined(__x86_64__)
1127#include <sys/param.h>
1128#include <sys/sysctl.h>
1129
1130// Gets the number of *physical cores* on the machine.
1131static int computeHostNumPhysicalCores() {
1132 uint32_t count;
1133 size_t len = sizeof(count);
1134 sysctlbyname("hw.physicalcpu", &count, &len, NULL, 0);
1135 if (count < 1) {
1136 int nm[2];
1137 nm[0] = CTL_HW;
1138 nm[1] = HW_AVAILCPU;
1139 sysctl(nm, 2, &count, &len, NULL, 0);
1140 if (count < 1)
1141 return -1;
1142 }
1143 return count;
1144}
1145#else
1146// On other systems, return -1 to indicate unknown.
1147static int computeHostNumPhysicalCores() { return -1; }
1148#endif
1149
1150int sys::getHostNumPhysicalCores() {
1151 static int NumCores = computeHostNumPhysicalCores();
1152 return NumCores;
1153}
1154
1155#if defined(__i386__) || defined(_M_IX86) || \
1156 defined(__x86_64__) || defined(_M_X64)
1157bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1158 unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
1159 unsigned MaxLevel;
1160 union {
1161 unsigned u[3];
1162 char c[12];
1163 } text;
1164
1165 if (getX86CpuIDAndInfo(0, &MaxLevel, text.u + 0, text.u + 2, text.u + 1) ||
1166 MaxLevel < 1)
1167 return false;
1168
1169 getX86CpuIDAndInfo(1, &EAX, &EBX, &ECX, &EDX);
1170
Craig Topper1af7e442017-11-19 23:30:22 +00001171 Features["cmov"] = (EDX >> 15) & 1;
1172 Features["mmx"] = (EDX >> 23) & 1;
1173 Features["sse"] = (EDX >> 25) & 1;
1174 Features["sse2"] = (EDX >> 26) & 1;
1175
1176 Features["sse3"] = (ECX >> 0) & 1;
1177 Features["pclmul"] = (ECX >> 1) & 1;
1178 Features["ssse3"] = (ECX >> 9) & 1;
1179 Features["cx16"] = (ECX >> 13) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001180 Features["sse4.1"] = (ECX >> 19) & 1;
1181 Features["sse4.2"] = (ECX >> 20) & 1;
Craig Topper1af7e442017-11-19 23:30:22 +00001182 Features["movbe"] = (ECX >> 22) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001183 Features["popcnt"] = (ECX >> 23) & 1;
Craig Topper1af7e442017-11-19 23:30:22 +00001184 Features["aes"] = (ECX >> 25) & 1;
1185 Features["rdrnd"] = (ECX >> 30) & 1;
Simon Pilgrima271c542017-05-03 15:42:29 +00001186
1187 // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
1188 // indicates that the AVX registers will be saved and restored on context
1189 // switch, then we have full AVX support.
1190 bool HasAVXSave = ((ECX >> 27) & 1) && ((ECX >> 28) & 1) &&
1191 !getX86XCR0(&EAX, &EDX) && ((EAX & 0x6) == 0x6);
Simon Pilgrima271c542017-05-03 15:42:29 +00001192 // AVX512 requires additional context to be saved by the OS.
1193 bool HasAVX512Save = HasAVXSave && ((EAX & 0xe0) == 0xe0);
1194
Craig Topper1af7e442017-11-19 23:30:22 +00001195 Features["avx"] = HasAVXSave;
1196 Features["fma"] = ((ECX >> 12) & 1) && HasAVXSave;
1197 // Only enable XSAVE if OS has enabled support for saving YMM state.
1198 Features["xsave"] = ((ECX >> 26) & 1) && HasAVXSave;
1199 Features["f16c"] = ((ECX >> 29) & 1) && HasAVXSave;
1200
Simon Pilgrima271c542017-05-03 15:42:29 +00001201 unsigned MaxExtLevel;
1202 getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
1203
1204 bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
1205 !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
Craig Topper8d02be32018-02-17 16:52:49 +00001206 Features["sahf"] = HasExtLeaf1 && ((ECX >> 0) & 1);
Craig Topper1af7e442017-11-19 23:30:22 +00001207 Features["lzcnt"] = HasExtLeaf1 && ((ECX >> 5) & 1);
1208 Features["sse4a"] = HasExtLeaf1 && ((ECX >> 6) & 1);
1209 Features["prfchw"] = HasExtLeaf1 && ((ECX >> 8) & 1);
1210 Features["xop"] = HasExtLeaf1 && ((ECX >> 11) & 1) && HasAVXSave;
1211 Features["lwp"] = HasExtLeaf1 && ((ECX >> 15) & 1);
1212 Features["fma4"] = HasExtLeaf1 && ((ECX >> 16) & 1) && HasAVXSave;
1213 Features["tbm"] = HasExtLeaf1 && ((ECX >> 21) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001214 Features["mwaitx"] = HasExtLeaf1 && ((ECX >> 29) & 1);
1215
1216 bool HasExtLeaf8 = MaxExtLevel >= 0x80000008 &&
Craig Topperdcd69792017-11-19 23:49:19 +00001217 !getX86CpuIDAndInfo(0x80000008, &EAX, &EBX, &ECX, &EDX);
Simon Pilgrima271c542017-05-03 15:42:29 +00001218 Features["clzero"] = HasExtLeaf8 && ((EBX >> 0) & 1);
1219
1220 bool HasLeaf7 =
1221 MaxLevel >= 7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
1222
Craig Topper1af7e442017-11-19 23:30:22 +00001223 Features["fsgsbase"] = HasLeaf7 && ((EBX >> 0) & 1);
1224 Features["sgx"] = HasLeaf7 && ((EBX >> 2) & 1);
1225 Features["bmi"] = HasLeaf7 && ((EBX >> 3) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001226 // AVX2 is only supported if we have the OS save support from AVX.
Craig Topper1af7e442017-11-19 23:30:22 +00001227 Features["avx2"] = HasLeaf7 && ((EBX >> 5) & 1) && HasAVXSave;
1228 Features["bmi2"] = HasLeaf7 && ((EBX >> 8) & 1);
1229 Features["rtm"] = HasLeaf7 && ((EBX >> 11) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001230 // AVX512 is only supported if the OS supports the context save for it.
Craig Topper1af7e442017-11-19 23:30:22 +00001231 Features["avx512f"] = HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save;
1232 Features["avx512dq"] = HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save;
1233 Features["rdseed"] = HasLeaf7 && ((EBX >> 18) & 1);
1234 Features["adx"] = HasLeaf7 && ((EBX >> 19) & 1);
Simon Pilgrima271c542017-05-03 15:42:29 +00001235 Features["avx512ifma"] = HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save;
Craig Topper1af7e442017-11-19 23:30:22 +00001236 Features["clflushopt"] = HasLeaf7 && ((EBX >> 23) & 1);
1237 Features["clwb"] = HasLeaf7 && ((EBX >> 24) & 1);
1238 Features["avx512pf"] = HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save;
1239 Features["avx512er"] = HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save;
1240 Features["avx512cd"] = HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save;
1241 Features["sha"] = HasLeaf7 && ((EBX >> 29) & 1);
1242 Features["avx512bw"] = HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save;
1243 Features["avx512vl"] = HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save;
Simon Pilgrima271c542017-05-03 15:42:29 +00001244
Craig Topper1af7e442017-11-19 23:30:22 +00001245 Features["prefetchwt1"] = HasLeaf7 && ((ECX >> 0) & 1);
1246 Features["avx512vbmi"] = HasLeaf7 && ((ECX >> 1) & 1) && HasAVX512Save;
Craig Topper9b03f672017-11-21 18:50:41 +00001247 Features["pku"] = HasLeaf7 && ((ECX >> 4) & 1);
Coby Tayree71e37cc2017-11-21 09:48:44 +00001248 Features["avx512vbmi2"] = HasLeaf7 && ((ECX >> 6) & 1) && HasAVX512Save;
Oren Ben Simhonfa582b02017-11-26 13:02:45 +00001249 Features["shstk"] = HasLeaf7 && ((ECX >> 7) & 1);
Coby Tayreed8b17be2017-11-26 09:36:41 +00001250 Features["gfni"] = HasLeaf7 && ((ECX >> 8) & 1);
Craig Topper9b03f672017-11-21 18:50:41 +00001251 Features["vaes"] = HasLeaf7 && ((ECX >> 9) & 1) && HasAVXSave;
1252 Features["vpclmulqdq"] = HasLeaf7 && ((ECX >> 10) & 1) && HasAVXSave;
1253 Features["avx512vnni"] = HasLeaf7 && ((ECX >> 11) & 1) && HasAVX512Save;
1254 Features["avx512bitalg"] = HasLeaf7 && ((ECX >> 12) & 1) && HasAVX512Save;
Yonghong Songdc1dbf62017-08-23 04:25:57 +00001255 Features["avx512vpopcntdq"] = HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save;
Craig Topper84b26b92018-01-18 23:52:31 +00001256 Features["rdpid"] = HasLeaf7 && ((ECX >> 22) & 1);
1257
1258 Features["ibt"] = HasLeaf7 && ((EDX >> 20) & 1);
Oren Ben Simhonfa582b02017-11-26 13:02:45 +00001259
Simon Pilgrima271c542017-05-03 15:42:29 +00001260 bool HasLeafD = MaxLevel >= 0xd &&
1261 !getX86CpuIDAndInfoEx(0xd, 0x1, &EAX, &EBX, &ECX, &EDX);
1262
1263 // Only enable XSAVE if OS has enabled support for saving YMM state.
Craig Topper1af7e442017-11-19 23:30:22 +00001264 Features["xsaveopt"] = HasLeafD && ((EAX >> 0) & 1) && HasAVXSave;
1265 Features["xsavec"] = HasLeafD && ((EAX >> 1) & 1) && HasAVXSave;
1266 Features["xsaves"] = HasLeafD && ((EAX >> 3) & 1) && HasAVXSave;
Simon Pilgrima271c542017-05-03 15:42:29 +00001267
1268 return true;
1269}
1270#elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1271bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1272 std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1273 if (!P)
1274 return false;
1275
1276 SmallVector<StringRef, 32> Lines;
1277 P->getBuffer().split(Lines, "\n");
1278
1279 SmallVector<StringRef, 32> CPUFeatures;
1280
1281 // Look for the CPU features.
1282 for (unsigned I = 0, E = Lines.size(); I != E; ++I)
1283 if (Lines[I].startswith("Features")) {
1284 Lines[I].split(CPUFeatures, ' ');
1285 break;
1286 }
1287
1288#if defined(__aarch64__)
1289 // Keep track of which crypto features we have seen
1290 enum { CAP_AES = 0x1, CAP_PMULL = 0x2, CAP_SHA1 = 0x4, CAP_SHA2 = 0x8 };
1291 uint32_t crypto = 0;
1292#endif
1293
1294 for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
1295 StringRef LLVMFeatureStr = StringSwitch<StringRef>(CPUFeatures[I])
1296#if defined(__aarch64__)
1297 .Case("asimd", "neon")
1298 .Case("fp", "fp-armv8")
1299 .Case("crc32", "crc")
1300#else
1301 .Case("half", "fp16")
1302 .Case("neon", "neon")
1303 .Case("vfpv3", "vfp3")
1304 .Case("vfpv3d16", "d16")
1305 .Case("vfpv4", "vfp4")
1306 .Case("idiva", "hwdiv-arm")
1307 .Case("idivt", "hwdiv")
1308#endif
1309 .Default("");
1310
1311#if defined(__aarch64__)
1312 // We need to check crypto separately since we need all of the crypto
1313 // extensions to enable the subtarget feature
1314 if (CPUFeatures[I] == "aes")
1315 crypto |= CAP_AES;
1316 else if (CPUFeatures[I] == "pmull")
1317 crypto |= CAP_PMULL;
1318 else if (CPUFeatures[I] == "sha1")
1319 crypto |= CAP_SHA1;
1320 else if (CPUFeatures[I] == "sha2")
1321 crypto |= CAP_SHA2;
1322#endif
1323
1324 if (LLVMFeatureStr != "")
1325 Features[LLVMFeatureStr] = true;
1326 }
1327
1328#if defined(__aarch64__)
1329 // If we have all crypto bits we can add the feature
1330 if (crypto == (CAP_AES | CAP_PMULL | CAP_SHA1 | CAP_SHA2))
1331 Features["crypto"] = true;
1332#endif
1333
1334 return true;
1335}
1336#else
1337bool sys::getHostCPUFeatures(StringMap<bool> &Features) { return false; }
1338#endif
1339
1340std::string sys::getProcessTriple() {
Alex Lorenz3803df32017-07-07 09:53:47 +00001341 std::string TargetTripleString = updateTripleOSVersion(LLVM_HOST_TRIPLE);
1342 Triple PT(Triple::normalize(TargetTripleString));
Simon Pilgrima271c542017-05-03 15:42:29 +00001343
1344 if (sizeof(void *) == 8 && PT.isArch32Bit())
1345 PT = PT.get64BitArchVariant();
1346 if (sizeof(void *) == 4 && PT.isArch64Bit())
1347 PT = PT.get32BitArchVariant();
1348
1349 return PT.str();
1350}