examples: cpu feature checks (#171)

Add necessary feature checks to the dot and geohash examples to prevent illegal instruction errors.

Fixes #170 #153
This commit is contained in:
Michael McLoughlin
2021-04-06 22:24:58 -07:00
committed by GitHub
parent 599bdd1269
commit 8f3c4ea0bf
4 changed files with 21 additions and 1 deletions

View File

@@ -3,17 +3,28 @@ package dot
import (
"math/rand"
"testing"
"golang.org/x/sys/cpu"
)
//go:generate go run asm.go -out dot.s -stubs stub.go
func RequireISAs(t *testing.T) {
t.Helper()
if !(cpu.X86.HasAVX && cpu.X86.HasFMA) {
t.Skip("requires AVX and FMA3 instruction sets")
}
}
func TestEmpty(t *testing.T) {
RequireISAs(t)
if Dot(nil, nil) != 0.0 {
t.Fatal("expect dot product of empty vectors to be zero")
}
}
func TestLengths(t *testing.T) {
RequireISAs(t)
const epsilon = 0.00001
for n := 0; n < 1000; n++ {
x, y := RandomVector(n), RandomVector(n)

View File

@@ -1,10 +1,17 @@
package geohash
import "testing"
import (
"testing"
"golang.org/x/sys/cpu"
)
//go:generate go run asm.go -out geohash.s -stubs stub.go
func TestEncodeIntMountEverest(t *testing.T) {
if !(cpu.X86.HasSSE2 && cpu.X86.HasBMI2) {
t.Skip("requires SSE2 and BMI2 instruction sets")
}
if EncodeInt(27.988056, 86.925278) != 0xceb7f254240fd612 {
t.Fail()
}