mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
cb7f2fa6fd
Related to #39095 https://go.dev/doc/modules/version-numbers Update pkg version according to golang dep version convention --------- Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
22 lines
446 B
Go
22 lines
446 B
Go
package distance
|
|
|
|
import (
|
|
"math"
|
|
|
|
"golang.org/x/sys/cpu"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/log"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/distance/asm"
|
|
)
|
|
|
|
func init() {
|
|
if cpu.X86.HasAVX2 {
|
|
log.Info("Hook avx for go simd distance computation")
|
|
IPImpl = asm.IP
|
|
L2Impl = asm.L2
|
|
CosineImpl = func(a []float32, b []float32) float32 {
|
|
return asm.IP(a, b) / float32(math.Sqrt(float64(asm.IP(a, a))*float64((asm.IP(b, b)))))
|
|
}
|
|
}
|
|
}
|