2024-10-10 11:33:36 +00:00
|
|
|
package guts
|
2022-08-19 16:34:07 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"syscall"
|
|
|
|
|
|
|
|
"github.com/klauspost/cpuid/v2"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
haveAVX2 bool
|
|
|
|
haveAVX512 bool
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
haveAVX2 = cpuid.CPU.Supports(cpuid.AVX2)
|
|
|
|
haveAVX512 = cpuid.CPU.Supports(cpuid.AVX512F)
|
|
|
|
if !haveAVX512 {
|
|
|
|
// On some Macs, AVX512 detection is buggy, so fallback to sysctl
|
|
|
|
b, _ := syscall.Sysctl("hw.optional.avx512f")
|
|
|
|
haveAVX512 = len(b) > 0 && b[0] == 1
|
|
|
|
}
|
|
|
|
}
|