mirror of
https://github.com/mudler/LocalAI.git
synced 2025-05-28 14:35:00 +00:00
feat(startup): show CPU/GPU information with --debug (#2241)
Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
parent
117c9873e1
commit
b69ff46c7e
6 changed files with 95 additions and 3 deletions
38
pkg/xsysinfo/cpu.go
Normal file
38
pkg/xsysinfo/cpu.go
Normal file
|
@ -0,0 +1,38 @@
|
|||
package xsysinfo
|
||||
|
||||
import (
|
||||
"sort"
|
||||
|
||||
"github.com/jaypipes/ghw"
|
||||
"github.com/klauspost/cpuid/v2"
|
||||
)
|
||||
|
||||
func CPUCapabilities() ([]string, error) {
|
||||
cpu, err := ghw.CPU()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
caps := map[string]struct{}{}
|
||||
|
||||
for _, proc := range cpu.Processors {
|
||||
for _, c := range proc.Capabilities {
|
||||
|
||||
caps[c] = struct{}{}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
ret := []string{}
|
||||
for c := range caps {
|
||||
ret = append(ret, c)
|
||||
}
|
||||
|
||||
// order
|
||||
sort.Strings(ret)
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func HasCPUCaps(ids ...cpuid.FeatureID) bool {
|
||||
return cpuid.CPU.Supports(ids...)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue