162 lines
4.1 KiB
Ruby
162 lines
4.1 KiB
Ruby
# typed: true # rubocop:todo Sorbet/StrictSigil
|
|
# frozen_string_literal: true
|
|
|
|
module Hardware
|
|
class CPU
|
|
class << self
|
|
def optimization_flags
|
|
@optimization_flags ||= begin
|
|
flags = generic_optimization_flags.dup
|
|
flags[:native] = arch_flag(Homebrew::EnvConfig.arch)
|
|
flags
|
|
end
|
|
end
|
|
|
|
def family
|
|
return :arm if arm?
|
|
return :ppc if ppc?
|
|
return :dunno unless intel?
|
|
|
|
# See https://software.intel.com/en-us/articles/intel-architecture-and-processor-identification-with-cpuid-model-and-family-numbers
|
|
# and https://github.com/llvm/llvm-project/blob/main/llvm/lib/TargetParser/Host.cpp
|
|
# and https://en.wikipedia.org/wiki/List_of_Intel_CPU_microarchitectures#Roadmap
|
|
vendor_id = cpuinfo[/^vendor_id\s*: (.*)/, 1]
|
|
cpu_family = cpuinfo[/^cpu family\s*: ([0-9]+)/, 1].to_i
|
|
cpu_model = cpuinfo[/^model\s*: ([0-9]+)/, 1].to_i
|
|
unknown = :"unknown_0x#{cpu_family.to_s(16)}_0x#{cpu_model.to_s(16)}"
|
|
case vendor_id
|
|
when "GenuineIntel"
|
|
intel_family(cpu_family, cpu_model)
|
|
when "AuthenticAMD"
|
|
amd_family(cpu_family, cpu_model)
|
|
end || unknown
|
|
end
|
|
|
|
def intel_family(family, cpu_model)
|
|
case family
|
|
when 0x06
|
|
case cpu_model
|
|
when 0x3a, 0x3e
|
|
:ivybridge
|
|
when 0x2a, 0x2d
|
|
:sandybridge
|
|
when 0x25, 0x2c, 0x2f
|
|
:westmere
|
|
when 0x1a, 0x1e, 0x1f, 0x2e
|
|
:nehalem
|
|
when 0x17, 0x1d
|
|
:penryn
|
|
when 0x0f, 0x16
|
|
:merom
|
|
when 0x0d
|
|
:dothan
|
|
when 0x1c, 0x26, 0x27, 0x35, 0x36
|
|
:atom
|
|
when 0x3c, 0x3f, 0x45, 0x46
|
|
:haswell
|
|
when 0x3d, 0x47, 0x4f, 0x56
|
|
:broadwell
|
|
when 0x4e, 0x5e, 0x8e, 0x9e, 0xa5, 0xa6
|
|
:skylake
|
|
when 0x66
|
|
:cannonlake
|
|
when 0x6a, 0x6c, 0x7d, 0x7e
|
|
:icelake
|
|
when 0xa7
|
|
:rocketlake
|
|
when 0x8c, 0x8d
|
|
:tigerlake
|
|
when 0x97, 0x9a, 0xbe, 0xb7, 0xba, 0xbf, 0xaa, 0xac
|
|
:alderlake
|
|
when 0xc5, 0xb5, 0xc6, 0xbd
|
|
:arrowlake
|
|
when 0xcc
|
|
:pantherlake
|
|
when 0xad, 0xae
|
|
:graniterapids
|
|
when 0xcf, 0x8f
|
|
:sapphirerapids
|
|
end
|
|
when 0x0f
|
|
case cpu_model
|
|
when 0x06
|
|
:presler
|
|
when 0x03, 0x04
|
|
:prescott
|
|
end
|
|
end
|
|
end
|
|
|
|
def amd_family(family, cpu_model)
|
|
case family
|
|
when 0x06
|
|
:amd_k7
|
|
when 0x0f
|
|
:amd_k8
|
|
when 0x10
|
|
:amd_k10
|
|
when 0x11
|
|
:amd_k8_k10_hybrid
|
|
when 0x12
|
|
:amd_k12
|
|
when 0x14
|
|
:bobcat
|
|
when 0x15
|
|
:bulldozer
|
|
when 0x16
|
|
:jaguar
|
|
when 0x17
|
|
case cpu_model
|
|
when 0x10..0x2f
|
|
:zen
|
|
when 0x30..0x3f, 0x47, 0x60..0x7f, 0x84..0x87, 0x90..0xaf
|
|
:zen2
|
|
end
|
|
when 0x19
|
|
case cpu_model
|
|
when ..0x0f, 0x20..0x5f
|
|
:zen3
|
|
when 0x10..0x1f, 0x60..0x7f, 0xa0..0xaf
|
|
:zen4
|
|
end
|
|
when 0x1a
|
|
:zen5
|
|
end
|
|
end
|
|
|
|
# Supported CPU instructions
|
|
def flags
|
|
@flags ||= cpuinfo[/^(?:flags|Features)\s*: (.*)/, 1]&.split
|
|
@flags ||= []
|
|
end
|
|
|
|
# Compatibility with Mac method, which returns lowercase symbols
|
|
# instead of strings.
|
|
def features
|
|
@features ||= flags[1..].map(&:intern)
|
|
end
|
|
|
|
%w[aes altivec avx avx2 lm ssse3 sse4_2].each do |flag|
|
|
define_method(:"#{flag}?") do
|
|
T.bind(self, T.class_of(Hardware::CPU))
|
|
flags.include? flag
|
|
end
|
|
end
|
|
|
|
def sse3?
|
|
flags.include?("pni") || flags.include?("sse3")
|
|
end
|
|
|
|
def sse4?
|
|
flags.include? "sse4_1"
|
|
end
|
|
|
|
private
|
|
|
|
def cpuinfo
|
|
@cpuinfo ||= File.read("/proc/cpuinfo")
|
|
end
|
|
end
|
|
end
|
|
end
|