makarna/pkg/backend/cpu/simd_avx512.go

//go:build amd64

package cpu

// hasAVX512Kernel signals whether the AVX-512 assembly implementation is
// available in this build tag.
const hasAVX512Kernel = true

//go:noescape
func dotAVX512(a *float32, b *float32, n int) float32

//go:noescape
func axpyAVX512(alpha float32, x *float32, y *float32, n int)

func dotFloat32AVX512(a, b []float32) float32 {
	if len(a) == 0 {
		return 0
	}
	return dotAVX512(&a[0], &b[0], len(a))
}

func axpyFloat32AVX512(alpha float32, x, y []float32) {
	if len(x) == 0 {
		return
	}
	axpyAVX512(alpha, &x[0], &y[0], len(x))
}