-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add the usage of SYSTEM_CPU_SET_INFORMATION
- Loading branch information
Showing
9 changed files
with
732 additions
and
63 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| package main | ||
|
|
||
| import "github.com/intel-go/cpuid" | ||
|
|
||
| func isAMD() bool { | ||
| return cpuid.VendorIdentificatorString == "AuthenticAMD" | ||
| } | ||
|
|
||
| func isIntel() bool { | ||
| return cpuid.VendorIdentificatorString == "GenuineIntel" | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,150 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "log" | ||
|
|
||
| "golang.org/x/sys/windows" | ||
| ) | ||
|
|
||
| // AMD 5900x | ||
| func Fake5900x() { | ||
| size := 0x20 * 64 | ||
| cpuSet := make([]SYSTEM_CPU_SET_INFORMATION, size) | ||
| cpuSet[0].Size = 32 | ||
| var lastCoreIndex byte | ||
| var count = 768 | ||
| var index = 0x100 | ||
| for i := 0; i < count; i++ { | ||
| cs := cpuSet[i].CpuSet() | ||
| cs.Id = uint32(index + i) | ||
| cs.LogicalProcessorIndex = byte(i) | ||
| if i%2 != 0 { | ||
| cs.CoreIndex = lastCoreIndex | ||
| } else { | ||
| cs.CoreIndex = byte(i) | ||
| lastCoreIndex = byte(i) | ||
| } | ||
| if i > 11 { | ||
| cs.LastLevelCacheIndex = 12 | ||
| } | ||
| } | ||
| SystemCpuSets = cpuSet[:count] | ||
| } | ||
|
|
||
| // Intel i9 13900 | ||
| func Fake13900() { | ||
| size := 0x20 * 64 | ||
| cpuSet := make([]SYSTEM_CPU_SET_INFORMATION, size) | ||
| cpuSet[0].Size = 32 | ||
| var lastCoreIndex byte | ||
| var count = 1024 | ||
| var index = 0x100 | ||
| for i := 0; i < count; i++ { | ||
| cs := cpuSet[i].CpuSet() | ||
| cs.Id = uint32(index + i) | ||
| cs.LogicalProcessorIndex = byte(i) | ||
|
|
||
| if i < 16 && i%2 != 0 { | ||
| cs.CoreIndex = lastCoreIndex | ||
| } else { | ||
| cs.CoreIndex = byte(i) | ||
| lastCoreIndex = byte(i) | ||
| } | ||
| if i < 16 { | ||
| cs.EfficiencyClass = 1 | ||
| } | ||
|
|
||
| } | ||
| SystemCpuSets = cpuSet[:count] | ||
| } | ||
|
|
||
| var SystemCpuSets = []SYSTEM_CPU_SET_INFORMATION{} | ||
|
|
||
| const ( | ||
| ToolTipTextNumaNode = "A group-relative value indicating which NUMA node a CPU Set is on. All CPU Sets in a given group that are on the same NUMA node will have the same value for this field." | ||
| ToolTipTextLastLevelCache = "A group-relative value indicating which CPU Sets share at least one level of cache with each other. This value is the same for all CPU Sets in a group that are on processors that share cache with each other." | ||
| ToolTipTextEfficiencyClass = "A value indicating the intrinsic energy efficiency of a processor for systems that support heterogeneous processors (such as ARM big.LITTLE systems). CPU Sets with higher numerical values of this field have home processors that are faster but less power-efficient than ones with lower values." | ||
| ) | ||
|
|
||
| type CpuSets struct { | ||
| HyperThreading bool | ||
| CoreCount int | ||
| MaxThreadsPerCore int | ||
| NumaNode bool // A group-relative value indicating which NUMA node a CPU Set is on. All CPU Sets in a given group that are on the same NUMA node will have the same value for this field. | ||
| LastLevelCache bool // A group-relative value indicating which CPU Sets share at least one level of cache with each other. This value is the same for all CPU Sets in a group that are on processors that share cache with each other. | ||
| EfficiencyClass bool // A value indicating the intrinsic energy efficiency of a processor for systems that support heterogeneous processors (such as ARM big.LITTLE systems). CPU Sets with higher numerical values of this field have home processors that are faster but less power-efficient than ones with lower values. | ||
| CPU []CpuSet | ||
| } | ||
|
|
||
| type CpuSet struct { | ||
| Id uint32 | ||
| CoreIndex byte | ||
| LogicalProcessorIndex byte | ||
| LastLevelCacheIndex byte // A group-relative value indicating which CPU Sets share at least one level of cache with each other. This value is the same for all CPU Sets in a group that are on processors that share cache with each other. | ||
| EfficiencyClass byte // A value indicating the intrinsic energy efficiency of a processor for systems that support heterogeneous processors (such as ARM big.LITTLE systems). CPU Sets with higher numerical values of this field have home processors that are faster but less power-efficient than ones with lower values. | ||
| NumaNodeIndex byte // A group-relative value indicating which NUMA node a CPU Set is on. All CPU Sets in a given group that are on the same NUMA node will have the same value for this field. | ||
| } | ||
|
|
||
| func (cs *CpuSets) Init() { | ||
| size := 0x20 * 64 | ||
| SystemCpuSets = make([]SYSTEM_CPU_SET_INFORMATION, size) | ||
|
|
||
| var length uint32 | ||
| var hProcess windows.Handle | ||
| success := GetSystemCpuSetInformation(&SystemCpuSets[0], uint32(size), &length, uintptr(hProcess), 0) | ||
| if success != 1 { | ||
| log.Println("err") | ||
| } else { | ||
| SystemCpuSets = SystemCpuSets[:length] | ||
| } | ||
|
|
||
| /// debug | ||
| // Fake13900() | ||
| // Fake5900x() | ||
|
|
||
| cs.CoreCount = int(uint32(len(SystemCpuSets)) / SystemCpuSets[0].Size) | ||
| var lastEfficiencyClass, lastLevelCache, lastNumaNodeIndex byte | ||
|
|
||
| for i := 0; i < cs.CoreCount; i++ { | ||
| cpu := SystemCpuSets[i].CpuSet() | ||
| if i == 0 { // The EfficiencyClass starts with 1 on the Intel Gen12+ | ||
| lastEfficiencyClass = cpu.EfficiencyClass | ||
| } | ||
|
|
||
| cs.CPU = append(cs.CPU, CpuSet{ | ||
| Id: cpu.Id, | ||
| CoreIndex: cpu.CoreIndex, | ||
| LogicalProcessorIndex: cpu.LogicalProcessorIndex, | ||
| EfficiencyClass: cpu.EfficiencyClass, | ||
| LastLevelCacheIndex: cpu.LastLevelCacheIndex, | ||
| NumaNodeIndex: cpu.NumaNodeIndex, | ||
| }) | ||
|
|
||
| // fmt.Printf("(%02d) [%d/%x] %02d/%02d Eff%d CCD%d NUMA%d\n", i, cpu.Id, cpu.Id, cpu.CoreIndex, cpu.LogicalProcessorIndex, cpu.EfficiencyClass, cpu.LastLevelCacheIndex, cpu.NumaNodeIndex) | ||
|
|
||
| if cpu.CoreIndex != cpu.LogicalProcessorIndex { | ||
| if !cs.HyperThreading { | ||
| cs.HyperThreading = true | ||
| } | ||
| if cs.MaxThreadsPerCore < int(cpu.LogicalProcessorIndex-cpu.CoreIndex) { | ||
| cs.MaxThreadsPerCore = int(cpu.LogicalProcessorIndex - cpu.CoreIndex) | ||
| } | ||
| } | ||
|
|
||
| if !cs.EfficiencyClass && lastEfficiencyClass != cpu.EfficiencyClass { | ||
| cs.EfficiencyClass = true | ||
| } | ||
|
|
||
| if !cs.LastLevelCache && lastLevelCache != cpu.LastLevelCacheIndex { | ||
| cs.LastLevelCache = true | ||
| } | ||
|
|
||
| if !cs.NumaNode && lastNumaNodeIndex != cpu.NumaNodeIndex { | ||
| cs.NumaNode = true | ||
| } | ||
|
|
||
| lastEfficiencyClass = cpu.EfficiencyClass | ||
| lastLevelCache = cpu.LastLevelCacheIndex | ||
| lastNumaNodeIndex = cpu.NumaNodeIndex | ||
| } | ||
| } |
Oops, something went wrong.