1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
package xxh3
import "testing"
func TestVectorCompat(t *testing.T) {
check := func(b []byte) {
t.Helper()
for i := range b {
b[i] = byte(i)
}
var avx512Sum, avx2Sum, sse2Sum, genericSum uint64
withAVX512(func() { avx512Sum = Hash(b) })
withAVX2(func() { avx2Sum = Hash(b) })
withSSE2(func() { sse2Sum = Hash(b) })
withGeneric(func() { genericSum = Hash(b) })
if avx2Sum != sse2Sum || avx2Sum != genericSum || sse2Sum != genericSum || avx512Sum != genericSum {
t.Errorf("data : %d", len(b))
t.Errorf("avx512: %016x", avx512Sum)
t.Errorf("avx2 : %016x", avx2Sum)
t.Errorf("sse2 : %016x", sse2Sum)
t.Errorf("scalar: %016x", genericSum)
t.FailNow()
}
}
t.Logf("avx512: %v", hasAVX512)
t.Logf("avx2: %v", hasAVX2)
t.Logf("sse2: %v", hasSSE2)
for _, n := range []int{
0, 1,
63, 64, 65,
127, 128, 129,
191, 192, 193,
239, 240, 241,
255, 256, 257,
319, 320, 321,
383, 384, 385,
447, 448, 449,
511, 512, 513,
575, 576, 577,
639, 640, 641,
703, 704, 705,
767, 768, 769,
831, 832, 833,
895, 896, 897,
959, 960, 961,
1023, 1024, 1025,
4932, 10233, 19238,
30000, 32 << 10, 1 + 32<<10,
100 * 1024,
1e6, 1e7, 1e8,
} {
check(make([]byte, n))
}
// Test more extensively around avx512Switch.
for n := avx512Switch; n < avx512Switch+(_block*5); n++ {
check(make([]byte, n))
}
}
|
