1 //===-- 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"
15 #include "llvm/Support/TargetParser.h"
16 #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/llvm-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 _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 
54 using namespace llvm;
55 
56 static 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 
68 StringRef sys::detail::getHostCPUNameForPowerPC(StringRef ProcCpuinfoContent) {
69   // 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 
147 StringRef sys::detail::getHostCPUNameForARM(StringRef ProcCpuinfoContent) {
148   // 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 == "0x42" || Implementer == "0x43") { // Broadcom | Cavium.
200     for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
201       if (Lines[I].startswith("CPU part")) {
202         return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
203           .Case("0x516", "thunderx2t99")
204           .Case("0x0516", "thunderx2t99")
205           .Case("0xaf", "thunderx2t99")
206           .Case("0x0af", "thunderx2t99")
207           .Case("0xa1", "thunderxt88")
208           .Case("0x0a1", "thunderxt88")
209           .Default("generic");
210       }
211     }
212   }
213 
214   if (Implementer == "0x51") // Qualcomm Technologies, Inc.
215     // Look for the CPU part line.
216     for (unsigned I = 0, E = Lines.size(); I != E; ++I)
217       if (Lines[I].startswith("CPU part"))
218         // The CPU part is a 3 digit hexadecimal number with a 0x prefix. The
219         // values correspond to the "Part number" in the CP15/c0 register. The
220         // contents are specified in the various processor manuals.
221         return StringSwitch<const char *>(Lines[I].substr(8).ltrim("\t :"))
222             .Case("0x06f", "krait") // APQ8064
223             .Case("0x201", "kryo")
224             .Case("0x205", "kryo")
225             .Case("0x211", "kryo")
226             .Case("0x800", "cortex-a73")
227             .Case("0x801", "cortex-a73")
228             .Case("0xc00", "falkor")
229             .Case("0xc01", "saphira")
230             .Default("generic");
231 
232   if (Implementer == "0x53") { // Samsung Electronics Co., Ltd.
233     // The Exynos chips have a convoluted ID scheme that doesn't seem to follow
234     // any predictive pattern across variants and parts.
235     unsigned Variant = 0, Part = 0;
236 
237     // Look for the CPU variant line, whose value is a 1 digit hexadecimal
238     // number, corresponding to the Variant bits in the CP15/C0 register.
239     for (auto I : Lines)
240       if (I.consume_front("CPU variant"))
241         I.ltrim("\t :").getAsInteger(0, Variant);
242 
243     // Look for the CPU part line, whose value is a 3 digit hexadecimal
244     // number, corresponding to the PartNum bits in the CP15/C0 register.
245     for (auto I : Lines)
246       if (I.consume_front("CPU part"))
247         I.ltrim("\t :").getAsInteger(0, Part);
248 
249     unsigned Exynos = (Variant << 12) | Part;
250     switch (Exynos) {
251     default:
252       // Default by falling through to Exynos M1.
253       LLVM_FALLTHROUGH;
254 
255     case 0x1001:
256       return "exynos-m1";
257 
258     case 0x4001:
259       return "exynos-m2";
260     }
261   }
262 
263   return "generic";
264 }
265 
266 StringRef sys::detail::getHostCPUNameForS390x(StringRef ProcCpuinfoContent) {
267   // STIDP is a privileged operation, so use /proc/cpuinfo instead.
268 
269   // The "processor 0:" line comes after a fair amount of other information,
270   // including a cache breakdown, but this should be plenty.
271   SmallVector<StringRef, 32> Lines;
272   ProcCpuinfoContent.split(Lines, "\n");
273 
274   // Look for the CPU features.
275   SmallVector<StringRef, 32> CPUFeatures;
276   for (unsigned I = 0, E = Lines.size(); I != E; ++I)
277     if (Lines[I].startswith("features")) {
278       size_t Pos = Lines[I].find(":");
279       if (Pos != StringRef::npos) {
280         Lines[I].drop_front(Pos + 1).split(CPUFeatures, ' ');
281         break;
282       }
283     }
284 
285   // We need to check for the presence of vector support independently of
286   // the machine type, since we may only use the vector register set when
287   // supported by the kernel (and hypervisor).
288   bool HaveVectorSupport = false;
289   for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
290     if (CPUFeatures[I] == "vx")
291       HaveVectorSupport = true;
292   }
293 
294   // Now check the processor machine type.
295   for (unsigned I = 0, E = Lines.size(); I != E; ++I) {
296     if (Lines[I].startswith("processor ")) {
297       size_t Pos = Lines[I].find("machine = ");
298       if (Pos != StringRef::npos) {
299         Pos += sizeof("machine = ") - 1;
300         unsigned int Id;
301         if (!Lines[I].drop_front(Pos).getAsInteger(10, Id)) {
302           if (Id >= 3906 && HaveVectorSupport)
303             return "z14";
304           if (Id >= 2964 && HaveVectorSupport)
305             return "z13";
306           if (Id >= 2827)
307             return "zEC12";
308           if (Id >= 2817)
309             return "z196";
310         }
311       }
312       break;
313     }
314   }
315 
316   return "generic";
317 }
318 
319 StringRef sys::detail::getHostCPUNameForBPF() {
320 #if !defined(__linux__) || !defined(__x86_64__)
321   return "generic";
322 #else
323   uint8_t insns[40] __attribute__ ((aligned (8))) =
324       /* BPF_MOV64_IMM(BPF_REG_0, 0) */
325     { 0xb7, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
326       /* BPF_MOV64_IMM(BPF_REG_2, 1) */
327       0xb7, 0x2, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
328       /* BPF_JMP_REG(BPF_JLT, BPF_REG_0, BPF_REG_2, 1) */
329       0xad, 0x20, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0,
330       /* BPF_MOV64_IMM(BPF_REG_0, 1) */
331       0xb7, 0x0, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0,
332       /* BPF_EXIT_INSN() */
333       0x95, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0 };
334 
335   struct bpf_prog_load_attr {
336     uint32_t prog_type;
337     uint32_t insn_cnt;
338     uint64_t insns;
339     uint64_t license;
340     uint32_t log_level;
341     uint32_t log_size;
342     uint64_t log_buf;
343     uint32_t kern_version;
344     uint32_t prog_flags;
345   } attr = {};
346   attr.prog_type = 1; /* BPF_PROG_TYPE_SOCKET_FILTER */
347   attr.insn_cnt = 5;
348   attr.insns = (uint64_t)insns;
349   attr.license = (uint64_t)"DUMMY";
350 
351   int fd = syscall(321 /* __NR_bpf */, 5 /* BPF_PROG_LOAD */, &attr, sizeof(attr));
352   if (fd >= 0) {
353     close(fd);
354     return "v2";
355   }
356   return "v1";
357 #endif
358 }
359 
360 #if defined(__i386__) || defined(_M_IX86) || \
361     defined(__x86_64__) || defined(_M_X64)
362 
363 enum VendorSignatures {
364   SIG_INTEL = 0x756e6547 /* Genu */,
365   SIG_AMD = 0x68747541 /* Auth */
366 };
367 
368 // The check below for i386 was copied from clang's cpuid.h (__get_cpuid_max).
369 // Check motivated by bug reports for OpenSSL crashing on CPUs without CPUID
370 // support. Consequently, for i386, the presence of CPUID is checked first
371 // via the corresponding eflags bit.
372 // Removal of cpuid.h header motivated by PR30384
373 // Header cpuid.h and method __get_cpuid_max are not used in llvm, clang, openmp
374 // or test-suite, but are used in external projects e.g. libstdcxx
375 static bool isCpuIdSupported() {
376 #if defined(__GNUC__) || defined(__clang__)
377 #if defined(__i386__)
378   int __cpuid_supported;
379   __asm__("  pushfl\n"
380           "  popl   %%eax\n"
381           "  movl   %%eax,%%ecx\n"
382           "  xorl   $0x00200000,%%eax\n"
383           "  pushl  %%eax\n"
384           "  popfl\n"
385           "  pushfl\n"
386           "  popl   %%eax\n"
387           "  movl   $0,%0\n"
388           "  cmpl   %%eax,%%ecx\n"
389           "  je     1f\n"
390           "  movl   $1,%0\n"
391           "1:"
392           : "=r"(__cpuid_supported)
393           :
394           : "eax", "ecx");
395   if (!__cpuid_supported)
396     return false;
397 #endif
398   return true;
399 #endif
400   return true;
401 }
402 
403 /// getX86CpuIDAndInfo - Execute the specified cpuid and return the 4 values in
404 /// the specified arguments.  If we can't run cpuid on the host, return true.
405 static bool getX86CpuIDAndInfo(unsigned value, unsigned *rEAX, unsigned *rEBX,
406                                unsigned *rECX, unsigned *rEDX) {
407 #if defined(__GNUC__) || defined(__clang__)
408 #if defined(__x86_64__)
409   // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
410   // FIXME: should we save this for Clang?
411   __asm__("movq\t%%rbx, %%rsi\n\t"
412           "cpuid\n\t"
413           "xchgq\t%%rbx, %%rsi\n\t"
414           : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
415           : "a"(value));
416   return false;
417 #elif defined(__i386__)
418   __asm__("movl\t%%ebx, %%esi\n\t"
419           "cpuid\n\t"
420           "xchgl\t%%ebx, %%esi\n\t"
421           : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
422           : "a"(value));
423   return false;
424 #else
425   return true;
426 #endif
427 #elif defined(_MSC_VER)
428   // The MSVC intrinsic is portable across x86 and x64.
429   int registers[4];
430   __cpuid(registers, value);
431   *rEAX = registers[0];
432   *rEBX = registers[1];
433   *rECX = registers[2];
434   *rEDX = registers[3];
435   return false;
436 #else
437   return true;
438 #endif
439 }
440 
441 /// getX86CpuIDAndInfoEx - Execute the specified cpuid with subleaf and return
442 /// the 4 values in the specified arguments.  If we can't run cpuid on the host,
443 /// return true.
444 static bool getX86CpuIDAndInfoEx(unsigned value, unsigned subleaf,
445                                  unsigned *rEAX, unsigned *rEBX, unsigned *rECX,
446                                  unsigned *rEDX) {
447 #if defined(__GNUC__) || defined(__clang__)
448 #if defined(__x86_64__)
449   // gcc doesn't know cpuid would clobber ebx/rbx. Preserve it manually.
450   // FIXME: should we save this for Clang?
451   __asm__("movq\t%%rbx, %%rsi\n\t"
452           "cpuid\n\t"
453           "xchgq\t%%rbx, %%rsi\n\t"
454           : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
455           : "a"(value), "c"(subleaf));
456   return false;
457 #elif defined(__i386__)
458   __asm__("movl\t%%ebx, %%esi\n\t"
459           "cpuid\n\t"
460           "xchgl\t%%ebx, %%esi\n\t"
461           : "=a"(*rEAX), "=S"(*rEBX), "=c"(*rECX), "=d"(*rEDX)
462           : "a"(value), "c"(subleaf));
463   return false;
464 #else
465   return true;
466 #endif
467 #elif defined(_MSC_VER)
468   int registers[4];
469   __cpuidex(registers, value, subleaf);
470   *rEAX = registers[0];
471   *rEBX = registers[1];
472   *rECX = registers[2];
473   *rEDX = registers[3];
474   return false;
475 #else
476   return true;
477 #endif
478 }
479 
480 // Read control register 0 (XCR0). Used to detect features such as AVX.
481 static bool getX86XCR0(unsigned *rEAX, unsigned *rEDX) {
482 #if defined(__GNUC__) || defined(__clang__)
483   // Check xgetbv; this uses a .byte sequence instead of the instruction
484   // directly because older assemblers do not include support for xgetbv and
485   // there is no easy way to conditionally compile based on the assembler used.
486   __asm__(".byte 0x0f, 0x01, 0xd0" : "=a"(*rEAX), "=d"(*rEDX) : "c"(0));
487   return false;
488 #elif defined(_MSC_FULL_VER) && defined(_XCR_XFEATURE_ENABLED_MASK)
489   unsigned long long Result = _xgetbv(_XCR_XFEATURE_ENABLED_MASK);
490   *rEAX = Result;
491   *rEDX = Result >> 32;
492   return false;
493 #else
494   return true;
495 #endif
496 }
497 
498 static void detectX86FamilyModel(unsigned EAX, unsigned *Family,
499                                  unsigned *Model) {
500   *Family = (EAX >> 8) & 0xf; // Bits 8 - 11
501   *Model = (EAX >> 4) & 0xf;  // Bits 4 - 7
502   if (*Family == 6 || *Family == 0xf) {
503     if (*Family == 0xf)
504       // Examine extended family ID if family ID is F.
505       *Family += (EAX >> 20) & 0xff; // Bits 20 - 27
506     // Examine extended model ID if family ID is 6 or F.
507     *Model += ((EAX >> 16) & 0xf) << 4; // Bits 16 - 19
508   }
509 }
510 
511 static void
512 getIntelProcessorTypeAndSubtype(unsigned Family, unsigned Model,
513                                 unsigned Brand_id, unsigned Features,
514                                 unsigned Features2, unsigned Features3,
515                                 unsigned *Type, unsigned *Subtype) {
516   if (Brand_id != 0)
517     return;
518   switch (Family) {
519   case 3:
520     *Type = X86::INTEL_i386;
521     break;
522   case 4:
523     *Type = X86::INTEL_i486;
524     break;
525   case 5:
526     if (Features & (1 << X86::FEATURE_MMX)) {
527       *Type = X86::INTEL_PENTIUM_MMX;
528       break;
529     }
530     *Type = X86::INTEL_PENTIUM;
531     break;
532   case 6:
533     switch (Model) {
534     case 0x01: // Pentium Pro processor
535       *Type = X86::INTEL_PENTIUM_PRO;
536       break;
537     case 0x03: // Intel Pentium II OverDrive processor, Pentium II processor,
538                // model 03
539     case 0x05: // Pentium II processor, model 05, Pentium II Xeon processor,
540                // model 05, and Intel Celeron processor, model 05
541     case 0x06: // Celeron processor, model 06
542       *Type = X86::INTEL_PENTIUM_II;
543       break;
544     case 0x07: // Pentium III processor, model 07, and Pentium III Xeon
545                // processor, model 07
546     case 0x08: // Pentium III processor, model 08, Pentium III Xeon processor,
547                // model 08, and Celeron processor, model 08
548     case 0x0a: // Pentium III Xeon processor, model 0Ah
549     case 0x0b: // Pentium III processor, model 0Bh
550       *Type = X86::INTEL_PENTIUM_III;
551       break;
552     case 0x09: // Intel Pentium M processor, Intel Celeron M processor model 09.
553     case 0x0d: // Intel Pentium M processor, Intel Celeron M processor, model
554                // 0Dh. All processors are manufactured using the 90 nm process.
555     case 0x15: // Intel EP80579 Integrated Processor and Intel EP80579
556                // Integrated Processor with Intel QuickAssist Technology
557       *Type = X86::INTEL_PENTIUM_M;
558       break;
559     case 0x0e: // Intel Core Duo processor, Intel Core Solo processor, model
560                // 0Eh. All processors are manufactured using the 65 nm process.
561       *Type = X86::INTEL_CORE_DUO;
562       break;   // yonah
563     case 0x0f: // Intel Core 2 Duo processor, Intel Core 2 Duo mobile
564                // processor, Intel Core 2 Quad processor, Intel Core 2 Quad
565                // mobile processor, Intel Core 2 Extreme processor, Intel
566                // Pentium Dual-Core processor, Intel Xeon processor, model
567                // 0Fh. All processors are manufactured using the 65 nm process.
568     case 0x16: // Intel Celeron processor model 16h. All processors are
569                // manufactured using the 65 nm process
570       *Type = X86::INTEL_CORE2; // "core2"
571       *Subtype = X86::INTEL_CORE2_65;
572       break;
573     case 0x17: // Intel Core 2 Extreme processor, Intel Xeon processor, model
574                // 17h. All processors are manufactured using the 45 nm process.
575                //
576                // 45nm: Penryn , Wolfdale, Yorkfield (XE)
577     case 0x1d: // Intel Xeon processor MP. All processors are manufactured using
578                // the 45 nm process.
579       *Type = X86::INTEL_CORE2; // "penryn"
580       *Subtype = X86::INTEL_CORE2_45;
581       break;
582     case 0x1a: // Intel Core i7 processor and Intel Xeon processor. All
583                // processors are manufactured using the 45 nm process.
584     case 0x1e: // Intel(R) Core(TM) i7 CPU         870  @ 2.93GHz.
585                // As found in a Summer 2010 model iMac.
586     case 0x1f:
587     case 0x2e:             // Nehalem EX
588       *Type = X86::INTEL_COREI7; // "nehalem"
589       *Subtype = X86::INTEL_COREI7_NEHALEM;
590       break;
591     case 0x25: // Intel Core i7, laptop version.
592     case 0x2c: // Intel Core i7 processor and Intel Xeon processor. All
593                // processors are manufactured using the 32 nm process.
594     case 0x2f: // Westmere EX
595       *Type = X86::INTEL_COREI7; // "westmere"
596       *Subtype = X86::INTEL_COREI7_WESTMERE;
597       break;
598     case 0x2a: // Intel Core i7 processor. All processors are manufactured
599                // using the 32 nm process.
600     case 0x2d:
601       *Type = X86::INTEL_COREI7; //"sandybridge"
602       *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
603       break;
604     case 0x3a:
605     case 0x3e:             // Ivy Bridge EP
606       *Type = X86::INTEL_COREI7; // "ivybridge"
607       *Subtype = X86::INTEL_COREI7_IVYBRIDGE;
608       break;
609 
610     // Haswell:
611     case 0x3c:
612     case 0x3f:
613     case 0x45:
614     case 0x46:
615       *Type = X86::INTEL_COREI7; // "haswell"
616       *Subtype = X86::INTEL_COREI7_HASWELL;
617       break;
618 
619     // Broadwell:
620     case 0x3d:
621     case 0x47:
622     case 0x4f:
623     case 0x56:
624       *Type = X86::INTEL_COREI7; // "broadwell"
625       *Subtype = X86::INTEL_COREI7_BROADWELL;
626       break;
627 
628     // Skylake:
629     case 0x4e: // Skylake mobile
630     case 0x5e: // Skylake desktop
631     case 0x8e: // Kaby Lake mobile
632     case 0x9e: // Kaby Lake desktop
633       *Type = X86::INTEL_COREI7; // "skylake"
634       *Subtype = X86::INTEL_COREI7_SKYLAKE;
635       break;
636 
637     // Skylake Xeon:
638     case 0x55:
639       *Type = X86::INTEL_COREI7;
640       *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512; // "skylake-avx512"
641       break;
642 
643     // Cannonlake:
644     case 0x66:
645       *Type = X86::INTEL_COREI7;
646       *Subtype = X86::INTEL_COREI7_CANNONLAKE; // "cannonlake"
647       break;
648 
649     case 0x1c: // Most 45 nm Intel Atom processors
650     case 0x26: // 45 nm Atom Lincroft
651     case 0x27: // 32 nm Atom Medfield
652     case 0x35: // 32 nm Atom Midview
653     case 0x36: // 32 nm Atom Midview
654       *Type = X86::INTEL_BONNELL;
655       break; // "bonnell"
656 
657     // Atom Silvermont codes from the Intel software optimization guide.
658     case 0x37:
659     case 0x4a:
660     case 0x4d:
661     case 0x5a:
662     case 0x5d:
663     case 0x4c: // really airmont
664       *Type = X86::INTEL_SILVERMONT;
665       break; // "silvermont"
666     // Goldmont:
667     case 0x5c: // Apollo Lake
668     case 0x5f: // Denverton
669       *Type = X86::INTEL_GOLDMONT;
670       break; // "goldmont"
671     case 0x7a:
672       *Type = X86::INTEL_GOLDMONT_PLUS;
673       break;
674     case 0x57:
675       *Type = X86::INTEL_KNL; // knl
676       break;
677     case 0x85:
678       *Type = X86::INTEL_KNM; // knm
679       break;
680 
681     default: // Unknown family 6 CPU, try to guess.
682       if (Features & (1 << X86::FEATURE_AVX512VBMI)) {
683         *Type = X86::INTEL_COREI7;
684         *Subtype = X86::INTEL_COREI7_CANNONLAKE;
685         break;
686       }
687 
688       if (Features & (1 << X86::FEATURE_AVX512VL)) {
689         *Type = X86::INTEL_COREI7;
690         *Subtype = X86::INTEL_COREI7_SKYLAKE_AVX512;
691         break;
692       }
693 
694       if (Features & (1 << X86::FEATURE_AVX512ER)) {
695         *Type = X86::INTEL_KNL; // knl
696         break;
697       }
698 
699       if (Features3 & (1 << (X86::FEATURE_CLFLUSHOPT - 64))) {
700         if (Features3 & (1 << (X86::FEATURE_SHA - 64))) {
701           *Type = X86::INTEL_GOLDMONT;
702         } else {
703           *Type = X86::INTEL_COREI7;
704           *Subtype = X86::INTEL_COREI7_SKYLAKE;
705         }
706         break;
707       }
708       if (Features3 & (1 << (X86::FEATURE_ADX - 64))) {
709         *Type = X86::INTEL_COREI7;
710         *Subtype = X86::INTEL_COREI7_BROADWELL;
711         break;
712       }
713       if (Features & (1 << X86::FEATURE_AVX2)) {
714         *Type = X86::INTEL_COREI7;
715         *Subtype = X86::INTEL_COREI7_HASWELL;
716         break;
717       }
718       if (Features & (1 << X86::FEATURE_AVX)) {
719         *Type = X86::INTEL_COREI7;
720         *Subtype = X86::INTEL_COREI7_SANDYBRIDGE;
721         break;
722       }
723       if (Features & (1 << X86::FEATURE_SSE4_2)) {
724         if (Features3 & (1 << (X86::FEATURE_MOVBE - 64))) {
725           *Type = X86::INTEL_SILVERMONT;
726         } else {
727           *Type = X86::INTEL_COREI7;
728           *Subtype = X86::INTEL_COREI7_NEHALEM;
729         }
730         break;
731       }
732       if (Features & (1 << X86::FEATURE_SSE4_1)) {
733         *Type = X86::INTEL_CORE2; // "penryn"
734         *Subtype = X86::INTEL_CORE2_45;
735         break;
736       }
737       if (Features & (1 << X86::FEATURE_SSSE3)) {
738         if (Features3 & (1 << (X86::FEATURE_MOVBE - 64))) {
739           *Type = X86::INTEL_BONNELL; // "bonnell"
740         } else {
741           *Type = X86::INTEL_CORE2; // "core2"
742           *Subtype = X86::INTEL_CORE2_65;
743         }
744         break;
745       }
746       if (Features3 & (1 << (X86::FEATURE_EM64T - 64))) {
747         *Type = X86::INTEL_CORE2; // "core2"
748         *Subtype = X86::INTEL_CORE2_65;
749         break;
750       }
751       if (Features & (1 << X86::FEATURE_SSE3)) {
752         *Type = X86::INTEL_CORE_DUO;
753         break;
754       }
755       if (Features & (1 << X86::FEATURE_SSE2)) {
756         *Type = X86::INTEL_PENTIUM_M;
757         break;
758       }
759       if (Features & (1 << X86::FEATURE_SSE)) {
760         *Type = X86::INTEL_PENTIUM_III;
761         break;
762       }
763       if (Features & (1 << X86::FEATURE_MMX)) {
764         *Type = X86::INTEL_PENTIUM_II;
765         break;
766       }
767       *Type = X86::INTEL_PENTIUM_PRO;
768       break;
769     }
770     break;
771   case 15: {
772     if (Features3 & (1 << (X86::FEATURE_EM64T - 64))) {
773       *Type = X86::INTEL_NOCONA;
774       break;
775     }
776     if (Features & (1 << X86::FEATURE_SSE3)) {
777       *Type = X86::INTEL_PRESCOTT;
778       break;
779     }
780     *Type = X86::INTEL_PENTIUM_IV;
781     break;
782   }
783   default:
784     break; /*"generic"*/
785   }
786 }
787 
788 static void getAMDProcessorTypeAndSubtype(unsigned Family, unsigned Model,
789                                           unsigned Features, unsigned *Type,
790                                           unsigned *Subtype) {
791   // FIXME: this poorly matches the generated SubtargetFeatureKV table.  There
792   // appears to be no way to generate the wide variety of AMD-specific targets
793   // from the information returned from CPUID.
794   switch (Family) {
795   case 4:
796     *Type = X86::AMD_i486;
797     break;
798   case 5:
799     *Type = X86::AMDPENTIUM;
800     switch (Model) {
801     case 6:
802     case 7:
803       *Subtype = X86::AMDPENTIUM_K6;
804       break; // "k6"
805     case 8:
806       *Subtype = X86::AMDPENTIUM_K62;
807       break; // "k6-2"
808     case 9:
809     case 13:
810       *Subtype = X86::AMDPENTIUM_K63;
811       break; // "k6-3"
812     case 10:
813       *Subtype = X86::AMDPENTIUM_GEODE;
814       break; // "geode"
815     }
816     break;
817   case 6:
818     if (Features & (1 << X86::FEATURE_SSE)) {
819       *Type = X86::AMD_ATHLON_XP;
820       break; // "athlon-xp"
821     }
822     *Type = X86::AMD_ATHLON;
823     break; // "athlon"
824   case 15:
825     if (Features & (1 << X86::FEATURE_SSE3)) {
826       *Type = X86::AMD_K8SSE3;
827       break; // "k8-sse3"
828     }
829     *Type = X86::AMD_K8;
830     break; // "k8"
831   case 16:
832     *Type = X86::AMDFAM10H; // "amdfam10"
833     switch (Model) {
834     case 2:
835       *Subtype = X86::AMDFAM10H_BARCELONA;
836       break;
837     case 4:
838       *Subtype = X86::AMDFAM10H_SHANGHAI;
839       break;
840     case 8:
841       *Subtype = X86::AMDFAM10H_ISTANBUL;
842       break;
843     }
844     break;
845   case 20:
846     *Type = X86::AMD_BTVER1;
847     break; // "btver1";
848   case 21:
849     *Type = X86::AMDFAM15H;
850     if (Model >= 0x60 && Model <= 0x7f) {
851       *Subtype = X86::AMDFAM15H_BDVER4;
852       break; // "bdver4"; 60h-7Fh: Excavator
853     }
854     if (Model >= 0x30 && Model <= 0x3f) {
855       *Subtype = X86::AMDFAM15H_BDVER3;
856       break; // "bdver3"; 30h-3Fh: Steamroller
857     }
858     if ((Model >= 0x10 && Model <= 0x1f) || Model == 0x02) {
859       *Subtype = X86::AMDFAM15H_BDVER2;
860       break; // "bdver2"; 02h, 10h-1Fh: Piledriver
861     }
862     if (Model <= 0x0f) {
863       *Subtype = X86::AMDFAM15H_BDVER1;
864       break; // "bdver1"; 00h-0Fh: Bulldozer
865     }
866     break;
867   case 22:
868     *Type = X86::AMD_BTVER2;
869     break; // "btver2"
870   case 23:
871     *Type = X86::AMDFAM17H;
872     *Subtype = X86::AMDFAM17H_ZNVER1;
873     break;
874   default:
875     break; // "generic"
876   }
877 }
878 
879 static void getAvailableFeatures(unsigned ECX, unsigned EDX, unsigned MaxLeaf,
880                                  unsigned *FeaturesOut, unsigned *Features2Out,
881                                  unsigned *Features3Out) {
882   unsigned Features = 0;
883   unsigned Features2 = 0;
884   unsigned Features3 = 0;
885   unsigned EAX, EBX;
886 
887   auto setFeature = [&](unsigned F) {
888     if (F < 32)
889       Features |= 1 << F;
890     else if (F < 64)
891       Features2 |= 1 << (F - 32);
892     else if (F < 96)
893       Features3 |= 1 << (F - 64);
894     else
895       llvm_unreachable("Unexpected FeatureBit");
896   };
897 
898   if ((EDX >> 15) & 1)
899     setFeature(X86::FEATURE_CMOV);
900   if ((EDX >> 23) & 1)
901     setFeature(X86::FEATURE_MMX);
902   if ((EDX >> 25) & 1)
903     setFeature(X86::FEATURE_SSE);
904   if ((EDX >> 26) & 1)
905     setFeature(X86::FEATURE_SSE2);
906 
907   if ((ECX >> 0) & 1)
908     setFeature(X86::FEATURE_SSE3);
909   if ((ECX >> 1) & 1)
910     setFeature(X86::FEATURE_PCLMUL);
911   if ((ECX >> 9) & 1)
912     setFeature(X86::FEATURE_SSSE3);
913   if ((ECX >> 12) & 1)
914     setFeature(X86::FEATURE_FMA);
915   if ((ECX >> 19) & 1)
916     setFeature(X86::FEATURE_SSE4_1);
917   if ((ECX >> 20) & 1)
918     setFeature(X86::FEATURE_SSE4_2);
919   if ((ECX >> 23) & 1)
920     setFeature(X86::FEATURE_POPCNT);
921   if ((ECX >> 25) & 1)
922     setFeature(X86::FEATURE_AES);
923 
924   if ((ECX >> 22) & 1)
925     setFeature(X86::FEATURE_MOVBE);
926 
927   // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
928   // indicates that the AVX registers will be saved and restored on context
929   // switch, then we have full AVX support.
930   const unsigned AVXBits = (1 << 27) | (1 << 28);
931   bool HasAVX = ((ECX & AVXBits) == AVXBits) && !getX86XCR0(&EAX, &EDX) &&
932                 ((EAX & 0x6) == 0x6);
933   bool HasAVX512Save = HasAVX && ((EAX & 0xe0) == 0xe0);
934 
935   if (HasAVX)
936     setFeature(X86::FEATURE_AVX);
937 
938   bool HasLeaf7 =
939       MaxLeaf >= 0x7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
940 
941   if (HasLeaf7 && ((EBX >> 3) & 1))
942     setFeature(X86::FEATURE_BMI);
943   if (HasLeaf7 && ((EBX >> 5) & 1) && HasAVX)
944     setFeature(X86::FEATURE_AVX2);
945   if (HasLeaf7 && ((EBX >> 9) & 1))
946     setFeature(X86::FEATURE_BMI2);
947   if (HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save)
948     setFeature(X86::FEATURE_AVX512F);
949   if (HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save)
950     setFeature(X86::FEATURE_AVX512DQ);
951   if (HasLeaf7 && ((EBX >> 19) & 1))
952     setFeature(X86::FEATURE_ADX);
953   if (HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save)
954     setFeature(X86::FEATURE_AVX512IFMA);
955   if (HasLeaf7 && ((EBX >> 23) & 1))
956     setFeature(X86::FEATURE_CLFLUSHOPT);
957   if (HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save)
958     setFeature(X86::FEATURE_AVX512PF);
959   if (HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save)
960     setFeature(X86::FEATURE_AVX512ER);
961   if (HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save)
962     setFeature(X86::FEATURE_AVX512CD);
963   if (HasLeaf7 && ((EBX >> 29) & 1))
964     setFeature(X86::FEATURE_SHA);
965   if (HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save)
966     setFeature(X86::FEATURE_AVX512BW);
967   if (HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save)
968     setFeature(X86::FEATURE_AVX512VL);
969 
970   if (HasLeaf7 && ((ECX >> 1) & 1) && HasAVX512Save)
971     setFeature(X86::FEATURE_AVX512VBMI);
972   if (HasLeaf7 && ((ECX >> 6) & 1) && HasAVX512Save)
973     setFeature(X86::FEATURE_AVX512VBMI2);
974   if (HasLeaf7 && ((ECX >> 8) & 1))
975     setFeature(X86::FEATURE_GFNI);
976   if (HasLeaf7 && ((ECX >> 10) & 1) && HasAVX)
977     setFeature(X86::FEATURE_VPCLMULQDQ);
978   if (HasLeaf7 && ((ECX >> 11) & 1) && HasAVX512Save)
979     setFeature(X86::FEATURE_AVX512VNNI);
980   if (HasLeaf7 && ((ECX >> 12) & 1) && HasAVX512Save)
981     setFeature(X86::FEATURE_AVX512BITALG);
982   if (HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save)
983     setFeature(X86::FEATURE_AVX512VPOPCNTDQ);
984 
985   if (HasLeaf7 && ((EDX >> 2) & 1) && HasAVX512Save)
986     setFeature(X86::FEATURE_AVX5124VNNIW);
987   if (HasLeaf7 && ((EDX >> 3) & 1) && HasAVX512Save)
988     setFeature(X86::FEATURE_AVX5124FMAPS);
989 
990   unsigned MaxExtLevel;
991   getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
992 
993   bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
994                      !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
995   if (HasExtLeaf1 && ((ECX >> 6) & 1))
996     setFeature(X86::FEATURE_SSE4_A);
997   if (HasExtLeaf1 && ((ECX >> 11) & 1))
998     setFeature(X86::FEATURE_XOP);
999   if (HasExtLeaf1 && ((ECX >> 16) & 1))
1000     setFeature(X86::FEATURE_FMA4);
1001 
1002   if (HasExtLeaf1 && ((EDX >> 29) & 1))
1003     setFeature(X86::FEATURE_EM64T);
1004 
1005   *FeaturesOut  = Features;
1006   *Features2Out = Features2;
1007   *Features3Out = Features3;
1008 }
1009 
1010 StringRef sys::getHostCPUName() {
1011   unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
1012   unsigned MaxLeaf, Vendor;
1013 
1014 #if defined(__GNUC__) || defined(__clang__)
1015   //FIXME: include cpuid.h from clang or copy __get_cpuid_max here
1016   // and simplify it to not invoke __cpuid (like cpu_model.c in
1017   // compiler-rt/lib/builtins/cpu_model.c?
1018   // Opting for the second option.
1019   if(!isCpuIdSupported())
1020     return "generic";
1021 #endif
1022   if (getX86CpuIDAndInfo(0, &MaxLeaf, &Vendor, &ECX, &EDX) || MaxLeaf < 1)
1023     return "generic";
1024   getX86CpuIDAndInfo(0x1, &EAX, &EBX, &ECX, &EDX);
1025 
1026   unsigned Brand_id = EBX & 0xff;
1027   unsigned Family = 0, Model = 0;
1028   unsigned Features = 0, Features2 = 0, Features3 = 0;
1029   detectX86FamilyModel(EAX, &Family, &Model);
1030   getAvailableFeatures(ECX, EDX, MaxLeaf, &Features, &Features2, &Features3);
1031 
1032   unsigned Type = 0;
1033   unsigned Subtype = 0;
1034 
1035   if (Vendor == SIG_INTEL) {
1036     getIntelProcessorTypeAndSubtype(Family, Model, Brand_id, Features,
1037                                     Features2, Features3, &Type, &Subtype);
1038   } else if (Vendor == SIG_AMD) {
1039     getAMDProcessorTypeAndSubtype(Family, Model, Features, &Type, &Subtype);
1040   }
1041 
1042   // Check subtypes first since those are more specific.
1043 #define X86_CPU_SUBTYPE(ARCHNAME, ENUM) \
1044   if (Subtype == X86::ENUM) \
1045     return ARCHNAME;
1046 #include "llvm/Support/X86TargetParser.def"
1047 
1048   // Now check types.
1049 #define X86_CPU_TYPE(ARCHNAME, ENUM) \
1050   if (Type == X86::ENUM) \
1051     return ARCHNAME;
1052 #include "llvm/Support/X86TargetParser.def"
1053 
1054   return "generic";
1055 }
1056 
1057 #elif defined(__APPLE__) && (defined(__ppc__) || defined(__powerpc__))
1058 StringRef sys::getHostCPUName() {
1059   host_basic_info_data_t hostInfo;
1060   mach_msg_type_number_t infoCount;
1061 
1062   infoCount = HOST_BASIC_INFO_COUNT;
1063   mach_port_t hostPort = mach_host_self();
1064   host_info(hostPort, HOST_BASIC_INFO, (host_info_t)&hostInfo,
1065             &infoCount);
1066   mach_port_deallocate(mach_task_self(), hostPort);
1067 
1068   if (hostInfo.cpu_type != CPU_TYPE_POWERPC)
1069     return "generic";
1070 
1071   switch (hostInfo.cpu_subtype) {
1072   case CPU_SUBTYPE_POWERPC_601:
1073     return "601";
1074   case CPU_SUBTYPE_POWERPC_602:
1075     return "602";
1076   case CPU_SUBTYPE_POWERPC_603:
1077     return "603";
1078   case CPU_SUBTYPE_POWERPC_603e:
1079     return "603e";
1080   case CPU_SUBTYPE_POWERPC_603ev:
1081     return "603ev";
1082   case CPU_SUBTYPE_POWERPC_604:
1083     return "604";
1084   case CPU_SUBTYPE_POWERPC_604e:
1085     return "604e";
1086   case CPU_SUBTYPE_POWERPC_620:
1087     return "620";
1088   case CPU_SUBTYPE_POWERPC_750:
1089     return "750";
1090   case CPU_SUBTYPE_POWERPC_7400:
1091     return "7400";
1092   case CPU_SUBTYPE_POWERPC_7450:
1093     return "7450";
1094   case CPU_SUBTYPE_POWERPC_970:
1095     return "970";
1096   default:;
1097   }
1098 
1099   return "generic";
1100 }
1101 #elif defined(__linux__) && (defined(__ppc__) || defined(__powerpc__))
1102 StringRef sys::getHostCPUName() {
1103   std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1104   StringRef Content = P ? P->getBuffer() : "";
1105   return detail::getHostCPUNameForPowerPC(Content);
1106 }
1107 #elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1108 StringRef sys::getHostCPUName() {
1109   std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1110   StringRef Content = P ? P->getBuffer() : "";
1111   return detail::getHostCPUNameForARM(Content);
1112 }
1113 #elif defined(__linux__) && defined(__s390x__)
1114 StringRef sys::getHostCPUName() {
1115   std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1116   StringRef Content = P ? P->getBuffer() : "";
1117   return detail::getHostCPUNameForS390x(Content);
1118 }
1119 #else
1120 StringRef sys::getHostCPUName() { return "generic"; }
1121 #endif
1122 
1123 #if defined(__linux__) && defined(__x86_64__)
1124 // On Linux, the number of physical cores can be computed from /proc/cpuinfo,
1125 // using the number of unique physical/core id pairs. The following
1126 // implementation reads the /proc/cpuinfo format on an x86_64 system.
1127 static int computeHostNumPhysicalCores() {
1128   // Read /proc/cpuinfo as a stream (until EOF reached). It cannot be
1129   // mmapped because it appears to have 0 size.
1130   llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Text =
1131       llvm::MemoryBuffer::getFileAsStream("/proc/cpuinfo");
1132   if (std::error_code EC = Text.getError()) {
1133     llvm::errs() << "Can't read "
1134                  << "/proc/cpuinfo: " << EC.message() << "\n";
1135     return -1;
1136   }
1137   SmallVector<StringRef, 8> strs;
1138   (*Text)->getBuffer().split(strs, "\n", /*MaxSplit=*/-1,
1139                              /*KeepEmpty=*/false);
1140   int CurPhysicalId = -1;
1141   int CurCoreId = -1;
1142   SmallSet<std::pair<int, int>, 32> UniqueItems;
1143   for (auto &Line : strs) {
1144     Line = Line.trim();
1145     if (!Line.startswith("physical id") && !Line.startswith("core id"))
1146       continue;
1147     std::pair<StringRef, StringRef> Data = Line.split(':');
1148     auto Name = Data.first.trim();
1149     auto Val = Data.second.trim();
1150     if (Name == "physical id") {
1151       assert(CurPhysicalId == -1 &&
1152              "Expected a core id before seeing another physical id");
1153       Val.getAsInteger(10, CurPhysicalId);
1154     }
1155     if (Name == "core id") {
1156       assert(CurCoreId == -1 &&
1157              "Expected a physical id before seeing another core id");
1158       Val.getAsInteger(10, CurCoreId);
1159     }
1160     if (CurPhysicalId != -1 && CurCoreId != -1) {
1161       UniqueItems.insert(std::make_pair(CurPhysicalId, CurCoreId));
1162       CurPhysicalId = -1;
1163       CurCoreId = -1;
1164     }
1165   }
1166   return UniqueItems.size();
1167 }
1168 #elif defined(__APPLE__) && defined(__x86_64__)
1169 #include <sys/param.h>
1170 #include <sys/sysctl.h>
1171 
1172 // Gets the number of *physical cores* on the machine.
1173 static int computeHostNumPhysicalCores() {
1174   uint32_t count;
1175   size_t len = sizeof(count);
1176   sysctlbyname("hw.physicalcpu", &count, &len, NULL, 0);
1177   if (count < 1) {
1178     int nm[2];
1179     nm[0] = CTL_HW;
1180     nm[1] = HW_AVAILCPU;
1181     sysctl(nm, 2, &count, &len, NULL, 0);
1182     if (count < 1)
1183       return -1;
1184   }
1185   return count;
1186 }
1187 #else
1188 // On other systems, return -1 to indicate unknown.
1189 static int computeHostNumPhysicalCores() { return -1; }
1190 #endif
1191 
1192 int sys::getHostNumPhysicalCores() {
1193   static int NumCores = computeHostNumPhysicalCores();
1194   return NumCores;
1195 }
1196 
1197 #if defined(__i386__) || defined(_M_IX86) || \
1198     defined(__x86_64__) || defined(_M_X64)
1199 bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1200   unsigned EAX = 0, EBX = 0, ECX = 0, EDX = 0;
1201   unsigned MaxLevel;
1202   union {
1203     unsigned u[3];
1204     char c[12];
1205   } text;
1206 
1207   if (getX86CpuIDAndInfo(0, &MaxLevel, text.u + 0, text.u + 2, text.u + 1) ||
1208       MaxLevel < 1)
1209     return false;
1210 
1211   getX86CpuIDAndInfo(1, &EAX, &EBX, &ECX, &EDX);
1212 
1213   Features["cmov"]   = (EDX >> 15) & 1;
1214   Features["mmx"]    = (EDX >> 23) & 1;
1215   Features["sse"]    = (EDX >> 25) & 1;
1216   Features["sse2"]   = (EDX >> 26) & 1;
1217 
1218   Features["sse3"]   = (ECX >>  0) & 1;
1219   Features["pclmul"] = (ECX >>  1) & 1;
1220   Features["ssse3"]  = (ECX >>  9) & 1;
1221   Features["cx16"]   = (ECX >> 13) & 1;
1222   Features["sse4.1"] = (ECX >> 19) & 1;
1223   Features["sse4.2"] = (ECX >> 20) & 1;
1224   Features["movbe"]  = (ECX >> 22) & 1;
1225   Features["popcnt"] = (ECX >> 23) & 1;
1226   Features["aes"]    = (ECX >> 25) & 1;
1227   Features["rdrnd"]  = (ECX >> 30) & 1;
1228 
1229   // If CPUID indicates support for XSAVE, XRESTORE and AVX, and XGETBV
1230   // indicates that the AVX registers will be saved and restored on context
1231   // switch, then we have full AVX support.
1232   bool HasAVXSave = ((ECX >> 27) & 1) && ((ECX >> 28) & 1) &&
1233                     !getX86XCR0(&EAX, &EDX) && ((EAX & 0x6) == 0x6);
1234   // AVX512 requires additional context to be saved by the OS.
1235   bool HasAVX512Save = HasAVXSave && ((EAX & 0xe0) == 0xe0);
1236 
1237   Features["avx"]   = HasAVXSave;
1238   Features["fma"]   = ((ECX >> 12) & 1) && HasAVXSave;
1239   // Only enable XSAVE if OS has enabled support for saving YMM state.
1240   Features["xsave"] = ((ECX >> 26) & 1) && HasAVXSave;
1241   Features["f16c"]  = ((ECX >> 29) & 1) && HasAVXSave;
1242 
1243   unsigned MaxExtLevel;
1244   getX86CpuIDAndInfo(0x80000000, &MaxExtLevel, &EBX, &ECX, &EDX);
1245 
1246   bool HasExtLeaf1 = MaxExtLevel >= 0x80000001 &&
1247                      !getX86CpuIDAndInfo(0x80000001, &EAX, &EBX, &ECX, &EDX);
1248   Features["sahf"]   = HasExtLeaf1 && ((ECX >>  0) & 1);
1249   Features["lzcnt"]  = HasExtLeaf1 && ((ECX >>  5) & 1);
1250   Features["sse4a"]  = HasExtLeaf1 && ((ECX >>  6) & 1);
1251   Features["prfchw"] = HasExtLeaf1 && ((ECX >>  8) & 1);
1252   Features["xop"]    = HasExtLeaf1 && ((ECX >> 11) & 1) && HasAVXSave;
1253   Features["lwp"]    = HasExtLeaf1 && ((ECX >> 15) & 1);
1254   Features["fma4"]   = HasExtLeaf1 && ((ECX >> 16) & 1) && HasAVXSave;
1255   Features["tbm"]    = HasExtLeaf1 && ((ECX >> 21) & 1);
1256   Features["mwaitx"] = HasExtLeaf1 && ((ECX >> 29) & 1);
1257 
1258   Features["64bit"]  = HasExtLeaf1 && ((EDX >> 29) & 1);
1259 
1260   // Miscellaneous memory related features, detected by
1261   // using the 0x80000008 leaf of the CPUID instruction
1262   bool HasExtLeaf8 = MaxExtLevel >= 0x80000008 &&
1263                      !getX86CpuIDAndInfo(0x80000008, &EAX, &EBX, &ECX, &EDX);
1264   Features["clzero"]   = HasExtLeaf8 && ((EBX >> 0) & 1);
1265   Features["wbnoinvd"] = HasExtLeaf8 && ((EBX >> 9) & 1);
1266 
1267   bool HasLeaf7 =
1268       MaxLevel >= 7 && !getX86CpuIDAndInfoEx(0x7, 0x0, &EAX, &EBX, &ECX, &EDX);
1269 
1270   Features["fsgsbase"]   = HasLeaf7 && ((EBX >>  0) & 1);
1271   Features["sgx"]        = HasLeaf7 && ((EBX >>  2) & 1);
1272   Features["bmi"]        = HasLeaf7 && ((EBX >>  3) & 1);
1273   // AVX2 is only supported if we have the OS save support from AVX.
1274   Features["avx2"]       = HasLeaf7 && ((EBX >>  5) & 1) && HasAVXSave;
1275   Features["bmi2"]       = HasLeaf7 && ((EBX >>  8) & 1);
1276   Features["invpcid"]    = HasLeaf7 && ((EBX >> 10) & 1);
1277   Features["rtm"]        = HasLeaf7 && ((EBX >> 11) & 1);
1278   // AVX512 is only supported if the OS supports the context save for it.
1279   Features["avx512f"]    = HasLeaf7 && ((EBX >> 16) & 1) && HasAVX512Save;
1280   Features["avx512dq"]   = HasLeaf7 && ((EBX >> 17) & 1) && HasAVX512Save;
1281   Features["rdseed"]     = HasLeaf7 && ((EBX >> 18) & 1);
1282   Features["adx"]        = HasLeaf7 && ((EBX >> 19) & 1);
1283   Features["avx512ifma"] = HasLeaf7 && ((EBX >> 21) & 1) && HasAVX512Save;
1284   Features["clflushopt"] = HasLeaf7 && ((EBX >> 23) & 1);
1285   Features["clwb"]       = HasLeaf7 && ((EBX >> 24) & 1);
1286   Features["avx512pf"]   = HasLeaf7 && ((EBX >> 26) & 1) && HasAVX512Save;
1287   Features["avx512er"]   = HasLeaf7 && ((EBX >> 27) & 1) && HasAVX512Save;
1288   Features["avx512cd"]   = HasLeaf7 && ((EBX >> 28) & 1) && HasAVX512Save;
1289   Features["sha"]        = HasLeaf7 && ((EBX >> 29) & 1);
1290   Features["avx512bw"]   = HasLeaf7 && ((EBX >> 30) & 1) && HasAVX512Save;
1291   Features["avx512vl"]   = HasLeaf7 && ((EBX >> 31) & 1) && HasAVX512Save;
1292 
1293   Features["prefetchwt1"]     = HasLeaf7 && ((ECX >>  0) & 1);
1294   Features["avx512vbmi"]      = HasLeaf7 && ((ECX >>  1) & 1) && HasAVX512Save;
1295   Features["pku"]             = HasLeaf7 && ((ECX >>  4) & 1);
1296   Features["waitpkg"]         = HasLeaf7 && ((ECX >>  5) & 1);
1297   Features["avx512vbmi2"]     = HasLeaf7 && ((ECX >>  6) & 1) && HasAVX512Save;
1298   Features["shstk"]           = HasLeaf7 && ((ECX >>  7) & 1);
1299   Features["gfni"]            = HasLeaf7 && ((ECX >>  8) & 1);
1300   Features["vaes"]            = HasLeaf7 && ((ECX >>  9) & 1) && HasAVXSave;
1301   Features["vpclmulqdq"]      = HasLeaf7 && ((ECX >> 10) & 1) && HasAVXSave;
1302   Features["avx512vnni"]      = HasLeaf7 && ((ECX >> 11) & 1) && HasAVX512Save;
1303   Features["avx512bitalg"]    = HasLeaf7 && ((ECX >> 12) & 1) && HasAVX512Save;
1304   Features["avx512vpopcntdq"] = HasLeaf7 && ((ECX >> 14) & 1) && HasAVX512Save;
1305   Features["rdpid"]           = HasLeaf7 && ((ECX >> 22) & 1);
1306   Features["cldemote"]        = HasLeaf7 && ((ECX >> 25) & 1);
1307   Features["movdiri"]         = HasLeaf7 && ((ECX >> 27) & 1);
1308   Features["movdir64b"]       = HasLeaf7 && ((ECX >> 28) & 1);
1309 
1310   // There are two CPUID leafs which information associated with the pconfig
1311   // instruction:
1312   // EAX=0x7, ECX=0x0 indicates the availability of the instruction (via the 18th
1313   // bit of EDX), while the EAX=0x1b leaf returns information on the
1314   // availability of specific pconfig leafs.
1315   // The target feature here only refers to the the first of these two.
1316   // Users might need to check for the availability of specific pconfig
1317   // leaves using cpuid, since that information is ignored while
1318   // detecting features using the "-march=native" flag.
1319   // For more info, see X86 ISA docs.
1320   Features["pconfig"] = HasLeaf7 && ((EDX >> 18) & 1);
1321 
1322   bool HasLeafD = MaxLevel >= 0xd &&
1323                   !getX86CpuIDAndInfoEx(0xd, 0x1, &EAX, &EBX, &ECX, &EDX);
1324 
1325   // Only enable XSAVE if OS has enabled support for saving YMM state.
1326   Features["xsaveopt"] = HasLeafD && ((EAX >> 0) & 1) && HasAVXSave;
1327   Features["xsavec"]   = HasLeafD && ((EAX >> 1) & 1) && HasAVXSave;
1328   Features["xsaves"]   = HasLeafD && ((EAX >> 3) & 1) && HasAVXSave;
1329 
1330   bool HasLeaf14 = MaxLevel >= 0x14 &&
1331                   !getX86CpuIDAndInfoEx(0x14, 0x0, &EAX, &EBX, &ECX, &EDX);
1332 
1333   Features["ptwrite"] = HasLeaf14 && ((EBX >> 4) & 1);
1334 
1335   return true;
1336 }
1337 #elif defined(__linux__) && (defined(__arm__) || defined(__aarch64__))
1338 bool sys::getHostCPUFeatures(StringMap<bool> &Features) {
1339   std::unique_ptr<llvm::MemoryBuffer> P = getProcCpuinfoContent();
1340   if (!P)
1341     return false;
1342 
1343   SmallVector<StringRef, 32> Lines;
1344   P->getBuffer().split(Lines, "\n");
1345 
1346   SmallVector<StringRef, 32> CPUFeatures;
1347 
1348   // Look for the CPU features.
1349   for (unsigned I = 0, E = Lines.size(); I != E; ++I)
1350     if (Lines[I].startswith("Features")) {
1351       Lines[I].split(CPUFeatures, ' ');
1352       break;
1353     }
1354 
1355 #if defined(__aarch64__)
1356   // Keep track of which crypto features we have seen
1357   enum { CAP_AES = 0x1, CAP_PMULL = 0x2, CAP_SHA1 = 0x4, CAP_SHA2 = 0x8 };
1358   uint32_t crypto = 0;
1359 #endif
1360 
1361   for (unsigned I = 0, E = CPUFeatures.size(); I != E; ++I) {
1362     StringRef LLVMFeatureStr = StringSwitch<StringRef>(CPUFeatures[I])
1363 #if defined(__aarch64__)
1364                                    .Case("asimd", "neon")
1365                                    .Case("fp", "fp-armv8")
1366                                    .Case("crc32", "crc")
1367 #else
1368                                    .Case("half", "fp16")
1369                                    .Case("neon", "neon")
1370                                    .Case("vfpv3", "vfp3")
1371                                    .Case("vfpv3d16", "d16")
1372                                    .Case("vfpv4", "vfp4")
1373                                    .Case("idiva", "hwdiv-arm")
1374                                    .Case("idivt", "hwdiv")
1375 #endif
1376                                    .Default("");
1377 
1378 #if defined(__aarch64__)
1379     // We need to check crypto separately since we need all of the crypto
1380     // extensions to enable the subtarget feature
1381     if (CPUFeatures[I] == "aes")
1382       crypto |= CAP_AES;
1383     else if (CPUFeatures[I] == "pmull")
1384       crypto |= CAP_PMULL;
1385     else if (CPUFeatures[I] == "sha1")
1386       crypto |= CAP_SHA1;
1387     else if (CPUFeatures[I] == "sha2")
1388       crypto |= CAP_SHA2;
1389 #endif
1390 
1391     if (LLVMFeatureStr != "")
1392       Features[LLVMFeatureStr] = true;
1393   }
1394 
1395 #if defined(__aarch64__)
1396   // If we have all crypto bits we can add the feature
1397   if (crypto == (CAP_AES | CAP_PMULL | CAP_SHA1 | CAP_SHA2))
1398     Features["crypto"] = true;
1399 #endif
1400 
1401   return true;
1402 }
1403 #else
1404 bool sys::getHostCPUFeatures(StringMap<bool> &Features) { return false; }
1405 #endif
1406 
1407 std::string sys::getProcessTriple() {
1408   std::string TargetTripleString = updateTripleOSVersion(LLVM_HOST_TRIPLE);
1409   Triple PT(Triple::normalize(TargetTripleString));
1410 
1411   if (sizeof(void *) == 8 && PT.isArch32Bit())
1412     PT = PT.get64BitArchVariant();
1413   if (sizeof(void *) == 4 && PT.isArch64Bit())
1414     PT = PT.get32BitArchVariant();
1415 
1416   return PT.str();
1417 }
1418