2024-01-10 12:29:58 +08:00
|
|
|
package gpu
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"golang.org/x/sys/cpu"
|
|
|
|
|
)
|
|
|
|
|
|
2024-06-06 03:07:20 +08:00
|
|
|
func GetCPUCapability() CPUCapability {
|
2024-01-10 12:29:58 +08:00
|
|
|
if cpu.X86.HasAVX2 {
|
2024-05-16 06:13:16 +08:00
|
|
|
return CPUCapabilityAVX2
|
2024-01-10 12:29:58 +08:00
|
|
|
}
|
|
|
|
|
if cpu.X86.HasAVX {
|
2024-05-16 06:13:16 +08:00
|
|
|
return CPUCapabilityAVX
|
2024-01-10 12:29:58 +08:00
|
|
|
}
|
|
|
|
|
// else LCD
|
2024-06-06 03:07:20 +08:00
|
|
|
return CPUCapabilityNone
|
2024-01-10 12:29:58 +08:00
|
|
|
}
|