1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504 |
- package cpuid
- import (
- "math"
- "strings"
- )
- type Vendor int
- const (
- Other Vendor = iota
- Intel
- AMD
- VIA
- Transmeta
- NSC
- KVM
- MSVM
- VMware
- XenHVM
- Bhyve
- Hygon
- SiS
- RDC
- )
- const (
- CMOV = 1 << iota
- NX
- AMD3DNOW
- AMD3DNOWEXT
- MMX
- MMXEXT
- SSE
- SSE2
- SSE3
- SSSE3
- SSE4
- SSE4A
- SSE42
- AVX
- AVX2
- FMA3
- FMA4
- XOP
- F16C
- BMI1
- BMI2
- TBM
- LZCNT
- POPCNT
- AESNI
- CLMUL
- HTT
- HLE
- RTM
- RDRAND
- RDSEED
- ADX
- SHA
- AVX512F
- AVX512DQ
- AVX512IFMA
- AVX512PF
- AVX512ER
- AVX512CD
- AVX512BW
- AVX512VL
- AVX512VBMI
- AVX512VBMI2
- AVX512VNNI
- AVX512VPOPCNTDQ
- GFNI
- VAES
- AVX512BITALG
- VPCLMULQDQ
- AVX512BF16
- AVX512VP2INTERSECT
- MPX
- ERMS
- RDTSCP
- CX16
- SGX
- SGXLC
- IBPB
- STIBP
- VMX
-
- SSE2SLOW
- SSE3SLOW
- ATOM
- )
- var flagNames = map[Flags]string{
- CMOV: "CMOV",
- NX: "NX",
- AMD3DNOW: "AMD3DNOW",
- AMD3DNOWEXT: "AMD3DNOWEXT",
- MMX: "MMX",
- MMXEXT: "MMXEXT",
- SSE: "SSE",
- SSE2: "SSE2",
- SSE3: "SSE3",
- SSSE3: "SSSE3",
- SSE4: "SSE4.1",
- SSE4A: "SSE4A",
- SSE42: "SSE4.2",
- AVX: "AVX",
- AVX2: "AVX2",
- FMA3: "FMA3",
- FMA4: "FMA4",
- XOP: "XOP",
- F16C: "F16C",
- BMI1: "BMI1",
- BMI2: "BMI2",
- TBM: "TBM",
- LZCNT: "LZCNT",
- POPCNT: "POPCNT",
- AESNI: "AESNI",
- CLMUL: "CLMUL",
- HTT: "HTT",
- HLE: "HLE",
- RTM: "RTM",
- RDRAND: "RDRAND",
- RDSEED: "RDSEED",
- ADX: "ADX",
- SHA: "SHA",
- AVX512F: "AVX512F",
- AVX512DQ: "AVX512DQ",
- AVX512IFMA: "AVX512IFMA",
- AVX512PF: "AVX512PF",
- AVX512ER: "AVX512ER",
- AVX512CD: "AVX512CD",
- AVX512BW: "AVX512BW",
- AVX512VL: "AVX512VL",
- AVX512VBMI: "AVX512VBMI",
- AVX512VBMI2: "AVX512VBMI2",
- AVX512VNNI: "AVX512VNNI",
- AVX512VPOPCNTDQ: "AVX512VPOPCNTDQ",
- GFNI: "GFNI",
- VAES: "VAES",
- AVX512BITALG: "AVX512BITALG",
- VPCLMULQDQ: "VPCLMULQDQ",
- AVX512BF16: "AVX512BF16",
- AVX512VP2INTERSECT: "AVX512VP2INTERSECT",
- MPX: "MPX",
- ERMS: "ERMS",
- RDTSCP: "RDTSCP",
- CX16: "CX16",
- SGX: "SGX",
- SGXLC: "SGXLC",
- IBPB: "IBPB",
- STIBP: "STIBP",
- VMX: "VMX",
-
- SSE2SLOW: "SSE2SLOW",
- SSE3SLOW: "SSE3SLOW",
- ATOM: "ATOM",
- }
- const (
-
- FP ArmFlags = 1 << iota
- ASIMD
- EVTSTRM
- AES
- PMULL
- SHA1
- SHA2
- CRC32
- ATOMICS
- FPHP
- ASIMDHP
- ARMCPUID
- ASIMDRDM
- JSCVT
- FCMA
- LRCPC
- DCPOP
- SHA3
- SM3
- SM4
- ASIMDDP
- SHA512
- SVE
- GPA
- )
- var flagNamesArm = map[ArmFlags]string{
- FP: "FP",
- ASIMD: "ASIMD",
- EVTSTRM: "EVTSTRM",
- AES: "AES",
- PMULL: "PMULL",
- SHA1: "SHA1",
- SHA2: "SHA2",
- CRC32: "CRC32",
- ATOMICS: "ATOMICS",
- FPHP: "FPHP",
- ASIMDHP: "ASIMDHP",
- ARMCPUID: "CPUID",
- ASIMDRDM: "ASIMDRDM",
- JSCVT: "JSCVT",
- FCMA: "FCMA",
- LRCPC: "LRCPC",
- DCPOP: "DCPOP",
- SHA3: "SHA3",
- SM3: "SM3",
- SM4: "SM4",
- ASIMDDP: "ASIMDDP",
- SHA512: "SHA512",
- SVE: "SVE",
- GPA: "GPA",
- }
- type CPUInfo struct {
- BrandName string
- VendorID Vendor
- VendorString string
- Features Flags
- Arm ArmFlags
- PhysicalCores int
- ThreadsPerCore int
- LogicalCores int
- Family int
- Model int
- CacheLine int
- Hz int64
- Cache struct {
- L1I int
- L1D int
- L2 int
- L3 int
- }
- SGX SGXSupport
- maxFunc uint32
- maxExFunc uint32
- }
- var cpuid func(op uint32) (eax, ebx, ecx, edx uint32)
- var cpuidex func(op, op2 uint32) (eax, ebx, ecx, edx uint32)
- var xgetbv func(index uint32) (eax, edx uint32)
- var rdtscpAsm func() (eax, ebx, ecx, edx uint32)
- var CPU CPUInfo
- func init() {
- initCPU()
- Detect()
- }
- func Detect() {
-
- CPU.ThreadsPerCore = 1
- CPU.Cache.L1I = -1
- CPU.Cache.L1D = -1
- CPU.Cache.L2 = -1
- CPU.Cache.L3 = -1
- addInfo(&CPU)
- }
- func (c CPUInfo) Cmov() bool {
- return c.Features&CMOV != 0
- }
- func (c CPUInfo) Amd3dnow() bool {
- return c.Features&AMD3DNOW != 0
- }
- func (c CPUInfo) Amd3dnowExt() bool {
- return c.Features&AMD3DNOWEXT != 0
- }
- func (c CPUInfo) VMX() bool {
- return c.Features&VMX != 0
- }
- func (c CPUInfo) MMX() bool {
- return c.Features&MMX != 0
- }
- func (c CPUInfo) MMXExt() bool {
- return c.Features&MMXEXT != 0
- }
- func (c CPUInfo) SSE() bool {
- return c.Features&SSE != 0
- }
- func (c CPUInfo) SSE2() bool {
- return c.Features&SSE2 != 0
- }
- func (c CPUInfo) SSE3() bool {
- return c.Features&SSE3 != 0
- }
- func (c CPUInfo) SSSE3() bool {
- return c.Features&SSSE3 != 0
- }
- func (c CPUInfo) SSE4() bool {
- return c.Features&SSE4 != 0
- }
- func (c CPUInfo) SSE42() bool {
- return c.Features&SSE42 != 0
- }
- func (c CPUInfo) AVX() bool {
- return c.Features&AVX != 0
- }
- func (c CPUInfo) AVX2() bool {
- return c.Features&AVX2 != 0
- }
- func (c CPUInfo) FMA3() bool {
- return c.Features&FMA3 != 0
- }
- func (c CPUInfo) FMA4() bool {
- return c.Features&FMA4 != 0
- }
- func (c CPUInfo) XOP() bool {
- return c.Features&XOP != 0
- }
- func (c CPUInfo) F16C() bool {
- return c.Features&F16C != 0
- }
- func (c CPUInfo) BMI1() bool {
- return c.Features&BMI1 != 0
- }
- func (c CPUInfo) BMI2() bool {
- return c.Features&BMI2 != 0
- }
- func (c CPUInfo) TBM() bool {
- return c.Features&TBM != 0
- }
- func (c CPUInfo) Lzcnt() bool {
- return c.Features&LZCNT != 0
- }
- func (c CPUInfo) Popcnt() bool {
- return c.Features&POPCNT != 0
- }
- func (c CPUInfo) HTT() bool {
- return c.Features&HTT != 0
- }
- func (c CPUInfo) SSE2Slow() bool {
- return c.Features&SSE2SLOW != 0
- }
- func (c CPUInfo) SSE3Slow() bool {
- return c.Features&SSE3SLOW != 0
- }
- func (c CPUInfo) AesNi() bool {
- return c.Features&AESNI != 0
- }
- func (c CPUInfo) Clmul() bool {
- return c.Features&CLMUL != 0
- }
- func (c CPUInfo) NX() bool {
- return c.Features&NX != 0
- }
- func (c CPUInfo) SSE4A() bool {
- return c.Features&SSE4A != 0
- }
- func (c CPUInfo) HLE() bool {
- return c.Features&HLE != 0
- }
- func (c CPUInfo) RTM() bool {
- return c.Features&RTM != 0
- }
- func (c CPUInfo) Rdrand() bool {
- return c.Features&RDRAND != 0
- }
- func (c CPUInfo) Rdseed() bool {
- return c.Features&RDSEED != 0
- }
- func (c CPUInfo) ADX() bool {
- return c.Features&ADX != 0
- }
- func (c CPUInfo) SHA() bool {
- return c.Features&SHA != 0
- }
- func (c CPUInfo) AVX512F() bool {
- return c.Features&AVX512F != 0
- }
- func (c CPUInfo) AVX512DQ() bool {
- return c.Features&AVX512DQ != 0
- }
- func (c CPUInfo) AVX512IFMA() bool {
- return c.Features&AVX512IFMA != 0
- }
- func (c CPUInfo) AVX512PF() bool {
- return c.Features&AVX512PF != 0
- }
- func (c CPUInfo) AVX512ER() bool {
- return c.Features&AVX512ER != 0
- }
- func (c CPUInfo) AVX512CD() bool {
- return c.Features&AVX512CD != 0
- }
- func (c CPUInfo) AVX512BW() bool {
- return c.Features&AVX512BW != 0
- }
- func (c CPUInfo) AVX512VL() bool {
- return c.Features&AVX512VL != 0
- }
- func (c CPUInfo) AVX512VBMI() bool {
- return c.Features&AVX512VBMI != 0
- }
- func (c CPUInfo) AVX512VBMI2() bool {
- return c.Features&AVX512VBMI2 != 0
- }
- func (c CPUInfo) AVX512VNNI() bool {
- return c.Features&AVX512VNNI != 0
- }
- func (c CPUInfo) AVX512VPOPCNTDQ() bool {
- return c.Features&AVX512VPOPCNTDQ != 0
- }
- func (c CPUInfo) GFNI() bool {
- return c.Features&GFNI != 0
- }
- func (c CPUInfo) VAES() bool {
- return c.Features&VAES != 0
- }
- func (c CPUInfo) AVX512BITALG() bool {
- return c.Features&AVX512BITALG != 0
- }
- func (c CPUInfo) VPCLMULQDQ() bool {
- return c.Features&VPCLMULQDQ != 0
- }
- func (c CPUInfo) AVX512BF16() bool {
- return c.Features&AVX512BF16 != 0
- }
- func (c CPUInfo) AVX512VP2INTERSECT() bool {
- return c.Features&AVX512VP2INTERSECT != 0
- }
- func (c CPUInfo) MPX() bool {
- return c.Features&MPX != 0
- }
- func (c CPUInfo) ERMS() bool {
- return c.Features&ERMS != 0
- }
- func (c CPUInfo) RDTSCP() bool {
- return c.Features&RDTSCP != 0
- }
- func (c CPUInfo) CX16() bool {
- return c.Features&CX16 != 0
- }
- func (c CPUInfo) TSX() bool {
- return c.Features&(HLE|RTM) == HLE|RTM
- }
- func (c CPUInfo) Atom() bool {
- return c.Features&ATOM != 0
- }
- func (c CPUInfo) Intel() bool {
- return c.VendorID == Intel
- }
- func (c CPUInfo) AMD() bool {
- return c.VendorID == AMD
- }
- func (c CPUInfo) Hygon() bool {
- return c.VendorID == Hygon
- }
- func (c CPUInfo) Transmeta() bool {
- return c.VendorID == Transmeta
- }
- func (c CPUInfo) NSC() bool {
- return c.VendorID == NSC
- }
- func (c CPUInfo) VIA() bool {
- return c.VendorID == VIA
- }
- func (c CPUInfo) RTCounter() uint64 {
- if !c.RDTSCP() {
- return 0
- }
- a, _, _, d := rdtscpAsm()
- return uint64(a) | (uint64(d) << 32)
- }
- func (c CPUInfo) Ia32TscAux() uint32 {
- if !c.RDTSCP() {
- return 0
- }
- _, _, ecx, _ := rdtscpAsm()
- return ecx
- }
- func (c CPUInfo) LogicalCPU() int {
- if c.maxFunc < 1 {
- return -1
- }
- _, ebx, _, _ := cpuid(1)
- return int(ebx >> 24)
- }
- func hertz(model string) int64 {
- mfi := maxFunctionID()
- if mfi >= 0x15 {
- eax, ebx, ecx, _ := cpuid(0x15)
- if eax != 0 && ebx != 0 && ecx != 0 {
- return int64((int64(ecx) * int64(ebx)) / int64(eax))
- }
- }
-
-
-
-
-
-
- hz := strings.LastIndex(model, "Hz")
- if hz < 3 {
- return -1
- }
- var multiplier int64
- switch model[hz-1] {
- case 'M':
- multiplier = 1000 * 1000
- case 'G':
- multiplier = 1000 * 1000 * 1000
- case 'T':
- multiplier = 1000 * 1000 * 1000 * 1000
- }
- if multiplier == 0 {
- return -1
- }
- freq := int64(0)
- divisor := int64(0)
- decimalShift := int64(1)
- var i int
- for i = hz - 2; i >= 0 && model[i] != ' '; i-- {
- if model[i] >= '0' && model[i] <= '9' {
- freq += int64(model[i]-'0') * decimalShift
- decimalShift *= 10
- } else if model[i] == '.' {
- if divisor != 0 {
- return -1
- }
- divisor = decimalShift
- } else {
- return -1
- }
- }
-
- if i < 0 {
- return -1
- }
- if divisor != 0 {
- return (freq * multiplier) / divisor
- }
- return freq * multiplier
- }
- func (c CPUInfo) VM() bool {
- switch c.VendorID {
- case MSVM, KVM, VMware, XenHVM, Bhyve:
- return true
- }
- return false
- }
- type Flags uint64
- type ArmFlags uint64
- func (f Flags) String() string {
- return strings.Join(f.Strings(), ",")
- }
- func (f Flags) Strings() []string {
- r := make([]string, 0, 20)
- for i := uint(0); i < 64; i++ {
- key := Flags(1 << i)
- val := flagNames[key]
- if f&key != 0 {
- r = append(r, val)
- }
- }
- return r
- }
- func (f ArmFlags) String() string {
- return strings.Join(f.Strings(), ",")
- }
- func (f ArmFlags) Strings() []string {
- r := make([]string, 0, 20)
- for i := uint(0); i < 64; i++ {
- key := ArmFlags(1 << i)
- val := flagNamesArm[key]
- if f&key != 0 {
- r = append(r, val)
- }
- }
- return r
- }
- func maxExtendedFunction() uint32 {
- eax, _, _, _ := cpuid(0x80000000)
- return eax
- }
- func maxFunctionID() uint32 {
- a, _, _, _ := cpuid(0)
- return a
- }
- func brandName() string {
- if maxExtendedFunction() >= 0x80000004 {
- v := make([]uint32, 0, 48)
- for i := uint32(0); i < 3; i++ {
- a, b, c, d := cpuid(0x80000002 + i)
- v = append(v, a, b, c, d)
- }
- return strings.Trim(string(valAsString(v...)), " ")
- }
- return "unknown"
- }
- func threadsPerCore() int {
- mfi := maxFunctionID()
- vend, _ := vendorID()
- if mfi < 0x4 || (vend != Intel && vend != AMD) {
- return 1
- }
- if mfi < 0xb {
- if vend != Intel {
- return 1
- }
- _, b, _, d := cpuid(1)
- if (d & (1 << 28)) != 0 {
-
- v := (b >> 16) & 255
- if v > 1 {
- a4, _, _, _ := cpuid(4)
-
- v2 := (a4 >> 26) + 1
- if v2 > 0 {
- return int(v) / int(v2)
- }
- }
- }
- return 1
- }
- _, b, _, _ := cpuidex(0xb, 0)
- if b&0xffff == 0 {
- return 1
- }
- return int(b & 0xffff)
- }
- func logicalCores() int {
- mfi := maxFunctionID()
- v, _ := vendorID()
- switch v {
- case Intel:
-
- if mfi < 0xb {
- if mfi < 1 {
- return 0
- }
-
-
-
- _, ebx, _, _ := cpuid(1)
- logical := (ebx >> 16) & 0xff
- return int(logical)
- }
- _, b, _, _ := cpuidex(0xb, 1)
- return int(b & 0xffff)
- case AMD, Hygon:
- _, b, _, _ := cpuid(1)
- return int((b >> 16) & 0xff)
- default:
- return 0
- }
- }
- func familyModel() (int, int) {
- if maxFunctionID() < 0x1 {
- return 0, 0
- }
- eax, _, _, _ := cpuid(1)
- family := ((eax >> 8) & 0xf) + ((eax >> 20) & 0xff)
- model := ((eax >> 4) & 0xf) + ((eax >> 12) & 0xf0)
- return int(family), int(model)
- }
- func physicalCores() int {
- v, _ := vendorID()
- switch v {
- case Intel:
- return logicalCores() / threadsPerCore()
- case AMD, Hygon:
- lc := logicalCores()
- tpc := threadsPerCore()
- if lc > 0 && tpc > 0 {
- return lc / tpc
- }
-
- if maxExtendedFunction() >= 0x80000008 {
- _, _, c, _ := cpuid(0x80000008)
- return int(c&0xff) + 1
- }
- }
- return 0
- }
- var vendorMapping = map[string]Vendor{
- "AMDisbetter!": AMD,
- "AuthenticAMD": AMD,
- "CentaurHauls": VIA,
- "GenuineIntel": Intel,
- "TransmetaCPU": Transmeta,
- "GenuineTMx86": Transmeta,
- "Geode by NSC": NSC,
- "VIA VIA VIA ": VIA,
- "KVMKVMKVMKVM": KVM,
- "Microsoft Hv": MSVM,
- "VMwareVMware": VMware,
- "XenVMMXenVMM": XenHVM,
- "bhyve bhyve ": Bhyve,
- "HygonGenuine": Hygon,
- "Vortex86 SoC": SiS,
- "SiS SiS SiS ": SiS,
- "RiseRiseRise": SiS,
- "Genuine RDC": RDC,
- }
- func vendorID() (Vendor, string) {
- _, b, c, d := cpuid(0)
- v := string(valAsString(b, d, c))
- vend, ok := vendorMapping[v]
- if !ok {
- return Other, v
- }
- return vend, v
- }
- func cacheLine() int {
- if maxFunctionID() < 0x1 {
- return 0
- }
- _, ebx, _, _ := cpuid(1)
- cache := (ebx & 0xff00) >> 5
- if cache == 0 && maxExtendedFunction() >= 0x80000006 {
- _, _, ecx, _ := cpuid(0x80000006)
- cache = ecx & 0xff
- }
-
- return int(cache)
- }
- func (c *CPUInfo) cacheSize() {
- c.Cache.L1D = -1
- c.Cache.L1I = -1
- c.Cache.L2 = -1
- c.Cache.L3 = -1
- vendor, _ := vendorID()
- switch vendor {
- case Intel:
- if maxFunctionID() < 4 {
- return
- }
- for i := uint32(0); ; i++ {
- eax, ebx, ecx, _ := cpuidex(4, i)
- cacheType := eax & 15
- if cacheType == 0 {
- break
- }
- cacheLevel := (eax >> 5) & 7
- coherency := int(ebx&0xfff) + 1
- partitions := int((ebx>>12)&0x3ff) + 1
- associativity := int((ebx>>22)&0x3ff) + 1
- sets := int(ecx) + 1
- size := associativity * partitions * coherency * sets
- switch cacheLevel {
- case 1:
- if cacheType == 1 {
-
- c.Cache.L1D = size
- } else if cacheType == 2 {
-
- c.Cache.L1I = size
- } else {
- if c.Cache.L1D < 0 {
- c.Cache.L1I = size
- }
- if c.Cache.L1I < 0 {
- c.Cache.L1I = size
- }
- }
- case 2:
- c.Cache.L2 = size
- case 3:
- c.Cache.L3 = size
- }
- }
- case AMD, Hygon:
-
- if maxExtendedFunction() < 0x80000005 {
- return
- }
- _, _, ecx, edx := cpuid(0x80000005)
- c.Cache.L1D = int(((ecx >> 24) & 0xFF) * 1024)
- c.Cache.L1I = int(((edx >> 24) & 0xFF) * 1024)
- if maxExtendedFunction() < 0x80000006 {
- return
- }
- _, _, ecx, _ = cpuid(0x80000006)
- c.Cache.L2 = int(((ecx >> 16) & 0xFFFF) * 1024)
-
- if maxExtendedFunction() < 0x8000001D {
- return
- }
- for i := uint32(0); i < math.MaxUint32; i++ {
- eax, ebx, ecx, _ := cpuidex(0x8000001D, i)
- level := (eax >> 5) & 7
- cacheNumSets := ecx + 1
- cacheLineSize := 1 + (ebx & 2047)
- cachePhysPartitions := 1 + ((ebx >> 12) & 511)
- cacheNumWays := 1 + ((ebx >> 22) & 511)
- typ := eax & 15
- size := int(cacheNumSets * cacheLineSize * cachePhysPartitions * cacheNumWays)
- if typ == 0 {
- return
- }
- switch level {
- case 1:
- switch typ {
- case 1:
-
- c.Cache.L1D = size
- case 2:
-
- c.Cache.L1I = size
- default:
- if c.Cache.L1D < 0 {
- c.Cache.L1I = size
- }
- if c.Cache.L1I < 0 {
- c.Cache.L1I = size
- }
- }
- case 2:
- c.Cache.L2 = size
- case 3:
- c.Cache.L3 = size
- }
- }
- }
- return
- }
- type SGXEPCSection struct {
- BaseAddress uint64
- EPCSize uint64
- }
- type SGXSupport struct {
- Available bool
- LaunchControl bool
- SGX1Supported bool
- SGX2Supported bool
- MaxEnclaveSizeNot64 int64
- MaxEnclaveSize64 int64
- EPCSections []SGXEPCSection
- }
- func hasSGX(available, lc bool) (rval SGXSupport) {
- rval.Available = available
- if !available {
- return
- }
- rval.LaunchControl = lc
- a, _, _, d := cpuidex(0x12, 0)
- rval.SGX1Supported = a&0x01 != 0
- rval.SGX2Supported = a&0x02 != 0
- rval.MaxEnclaveSizeNot64 = 1 << (d & 0xFF)
- rval.MaxEnclaveSize64 = 1 << ((d >> 8) & 0xFF)
- rval.EPCSections = make([]SGXEPCSection, 0)
- for subleaf := uint32(2); subleaf < 2+8; subleaf++ {
- eax, ebx, ecx, edx := cpuidex(0x12, subleaf)
- leafType := eax & 0xf
- if leafType == 0 {
-
- break
- } else if leafType == 1 {
-
- baseAddress := uint64(eax&0xfffff000) + (uint64(ebx&0x000fffff) << 32)
- size := uint64(ecx&0xfffff000) + (uint64(edx&0x000fffff) << 32)
- section := SGXEPCSection{BaseAddress: baseAddress, EPCSize: size}
- rval.EPCSections = append(rval.EPCSections, section)
- }
- }
- return
- }
- func support() Flags {
- mfi := maxFunctionID()
- vend, _ := vendorID()
- if mfi < 0x1 {
- return 0
- }
- rval := uint64(0)
- _, _, c, d := cpuid(1)
- if (d & (1 << 15)) != 0 {
- rval |= CMOV
- }
- if (d & (1 << 23)) != 0 {
- rval |= MMX
- }
- if (d & (1 << 25)) != 0 {
- rval |= MMXEXT
- }
- if (d & (1 << 25)) != 0 {
- rval |= SSE
- }
- if (d & (1 << 26)) != 0 {
- rval |= SSE2
- }
- if (c & 1) != 0 {
- rval |= SSE3
- }
- if (c & (1 << 5)) != 0 {
- rval |= VMX
- }
- if (c & 0x00000200) != 0 {
- rval |= SSSE3
- }
- if (c & 0x00080000) != 0 {
- rval |= SSE4
- }
- if (c & 0x00100000) != 0 {
- rval |= SSE42
- }
- if (c & (1 << 25)) != 0 {
- rval |= AESNI
- }
- if (c & (1 << 1)) != 0 {
- rval |= CLMUL
- }
- if c&(1<<23) != 0 {
- rval |= POPCNT
- }
- if c&(1<<30) != 0 {
- rval |= RDRAND
- }
- if c&(1<<29) != 0 {
- rval |= F16C
- }
- if c&(1<<13) != 0 {
- rval |= CX16
- }
- if vend == Intel && (d&(1<<28)) != 0 && mfi >= 4 {
- if threadsPerCore() > 1 {
- rval |= HTT
- }
- }
- if vend == AMD && (d&(1<<28)) != 0 && mfi >= 4 {
- if threadsPerCore() > 1 {
- rval |= HTT
- }
- }
-
- if c&(1<<26) != 0 && c&(1<<27) != 0 && c&(1<<28) != 0 {
-
- eax, _ := xgetbv(0)
- if (eax & 0x6) == 0x6 {
- rval |= AVX
- if (c & 0x00001000) != 0 {
- rval |= FMA3
- }
- }
- }
-
- if mfi >= 7 {
- _, ebx, ecx, edx := cpuidex(7, 0)
- eax1, _, _, _ := cpuidex(7, 1)
- if (rval&AVX) != 0 && (ebx&0x00000020) != 0 {
- rval |= AVX2
- }
- if (ebx & 0x00000008) != 0 {
- rval |= BMI1
- if (ebx & 0x00000100) != 0 {
- rval |= BMI2
- }
- }
- if ebx&(1<<2) != 0 {
- rval |= SGX
- }
- if ebx&(1<<4) != 0 {
- rval |= HLE
- }
- if ebx&(1<<9) != 0 {
- rval |= ERMS
- }
- if ebx&(1<<11) != 0 {
- rval |= RTM
- }
- if ebx&(1<<14) != 0 {
- rval |= MPX
- }
- if ebx&(1<<18) != 0 {
- rval |= RDSEED
- }
- if ebx&(1<<19) != 0 {
- rval |= ADX
- }
- if ebx&(1<<29) != 0 {
- rval |= SHA
- }
- if edx&(1<<26) != 0 {
- rval |= IBPB
- }
- if ecx&(1<<30) != 0 {
- rval |= SGXLC
- }
- if edx&(1<<27) != 0 {
- rval |= STIBP
- }
-
- if c&((1<<26)|(1<<27)) == (1<<26)|(1<<27) {
-
- eax, _ := xgetbv(0)
-
-
-
- if (eax>>5)&7 == 7 && (eax>>1)&3 == 3 {
- if ebx&(1<<16) != 0 {
- rval |= AVX512F
- }
- if ebx&(1<<17) != 0 {
- rval |= AVX512DQ
- }
- if ebx&(1<<21) != 0 {
- rval |= AVX512IFMA
- }
- if ebx&(1<<26) != 0 {
- rval |= AVX512PF
- }
- if ebx&(1<<27) != 0 {
- rval |= AVX512ER
- }
- if ebx&(1<<28) != 0 {
- rval |= AVX512CD
- }
- if ebx&(1<<30) != 0 {
- rval |= AVX512BW
- }
- if ebx&(1<<31) != 0 {
- rval |= AVX512VL
- }
-
- if ecx&(1<<1) != 0 {
- rval |= AVX512VBMI
- }
- if ecx&(1<<6) != 0 {
- rval |= AVX512VBMI2
- }
- if ecx&(1<<8) != 0 {
- rval |= GFNI
- }
- if ecx&(1<<9) != 0 {
- rval |= VAES
- }
- if ecx&(1<<10) != 0 {
- rval |= VPCLMULQDQ
- }
- if ecx&(1<<11) != 0 {
- rval |= AVX512VNNI
- }
- if ecx&(1<<12) != 0 {
- rval |= AVX512BITALG
- }
- if ecx&(1<<14) != 0 {
- rval |= AVX512VPOPCNTDQ
- }
-
- if edx&(1<<8) != 0 {
- rval |= AVX512VP2INTERSECT
- }
-
- if eax1&(1<<5) != 0 {
- rval |= AVX512BF16
- }
- }
- }
- }
- if maxExtendedFunction() >= 0x80000001 {
- _, _, c, d := cpuid(0x80000001)
- if (c & (1 << 5)) != 0 {
- rval |= LZCNT
- rval |= POPCNT
- }
- if (d & (1 << 31)) != 0 {
- rval |= AMD3DNOW
- }
- if (d & (1 << 30)) != 0 {
- rval |= AMD3DNOWEXT
- }
- if (d & (1 << 23)) != 0 {
- rval |= MMX
- }
- if (d & (1 << 22)) != 0 {
- rval |= MMXEXT
- }
- if (c & (1 << 6)) != 0 {
- rval |= SSE4A
- }
- if d&(1<<20) != 0 {
- rval |= NX
- }
- if d&(1<<27) != 0 {
- rval |= RDTSCP
- }
-
- if vend != Intel &&
- rval&SSE2 != 0 && (c&0x00000040) == 0 {
- rval |= SSE2SLOW
- }
-
- if (rval & AVX) != 0 {
- if (c & 0x00000800) != 0 {
- rval |= XOP
- }
- if (c & 0x00010000) != 0 {
- rval |= FMA4
- }
- }
- if vend == Intel {
- family, model := familyModel()
- if family == 6 && (model == 9 || model == 13 || model == 14) {
-
- if (rval & SSE2) != 0 {
- rval |= SSE2SLOW
- }
- if (rval & SSE3) != 0 {
- rval |= SSE3SLOW
- }
- }
-
- if family == 6 && model == 28 {
- rval |= ATOM
- }
- }
- }
- return Flags(rval)
- }
- func valAsString(values ...uint32) []byte {
- r := make([]byte, 4*len(values))
- for i, v := range values {
- dst := r[i*4:]
- dst[0] = byte(v & 0xff)
- dst[1] = byte((v >> 8) & 0xff)
- dst[2] = byte((v >> 16) & 0xff)
- dst[3] = byte((v >> 24) & 0xff)
- switch {
- case dst[0] == 0:
- return r[:i*4]
- case dst[1] == 0:
- return r[:i*4+1]
- case dst[2] == 0:
- return r[:i*4+2]
- case dst[3] == 0:
- return r[:i*4+3]
- }
- }
- return r
- }
- func (c CPUInfo) ArmFP() bool {
- return c.Arm&FP != 0
- }
- func (c CPUInfo) ArmASIMD() bool {
- return c.Arm&ASIMD != 0
- }
- func (c CPUInfo) ArmEVTSTRM() bool {
- return c.Arm&EVTSTRM != 0
- }
- func (c CPUInfo) ArmAES() bool {
- return c.Arm&AES != 0
- }
- func (c CPUInfo) ArmPMULL() bool {
- return c.Arm&PMULL != 0
- }
- func (c CPUInfo) ArmSHA1() bool {
- return c.Arm&SHA1 != 0
- }
- func (c CPUInfo) ArmSHA2() bool {
- return c.Arm&SHA2 != 0
- }
- func (c CPUInfo) ArmCRC32() bool {
- return c.Arm&CRC32 != 0
- }
- func (c CPUInfo) ArmATOMICS() bool {
- return c.Arm&ATOMICS != 0
- }
- func (c CPUInfo) ArmFPHP() bool {
- return c.Arm&FPHP != 0
- }
- func (c CPUInfo) ArmASIMDHP() bool {
- return c.Arm&ASIMDHP != 0
- }
- func (c CPUInfo) ArmASIMDRDM() bool {
- return c.Arm&ASIMDRDM != 0
- }
- func (c CPUInfo) ArmJSCVT() bool {
- return c.Arm&JSCVT != 0
- }
- func (c CPUInfo) ArmFCMA() bool {
- return c.Arm&FCMA != 0
- }
- func (c CPUInfo) ArmLRCPC() bool {
- return c.Arm&LRCPC != 0
- }
- func (c CPUInfo) ArmDCPOP() bool {
- return c.Arm&DCPOP != 0
- }
- func (c CPUInfo) ArmSHA3() bool {
- return c.Arm&SHA3 != 0
- }
- func (c CPUInfo) ArmSM3() bool {
- return c.Arm&SM3 != 0
- }
- func (c CPUInfo) ArmSM4() bool {
- return c.Arm&SM4 != 0
- }
- func (c CPUInfo) ArmASIMDDP() bool {
- return c.Arm&ASIMDDP != 0
- }
- func (c CPUInfo) ArmSHA512() bool {
- return c.Arm&SHA512 != 0
- }
- func (c CPUInfo) ArmSVE() bool {
- return c.Arm&SVE != 0
- }
- func (c CPUInfo) ArmGPA() bool {
- return c.Arm&GPA != 0
- }
|