1
2
3
4
5
6
7
8
9 package simd_test
10
11 import (
12 "math"
13 "simd/archsimd"
14 "testing"
15 )
16
17 func testInt8x16Reduce(t *testing.T, f func(_ archsimd.Int8x16) int8, want func(_ []int8) int8) {
18 n := 16
19 t.Helper()
20 forSlice(t, int8s, n, func(x []int8) bool {
21 t.Helper()
22 a := archsimd.LoadInt8x16(x)
23 g := f(a)
24 w := want(x)
25 if g != w {
26 t.Errorf("got %v, want %v, input %v", g, w, x)
27 return false
28 }
29 return true
30 })
31 }
32
33 func testInt16x8Reduce(t *testing.T, f func(_ archsimd.Int16x8) int16, want func(_ []int16) int16) {
34 n := 8
35 t.Helper()
36 forSlice(t, int16s, n, func(x []int16) bool {
37 t.Helper()
38 a := archsimd.LoadInt16x8(x)
39 g := f(a)
40 w := want(x)
41 if g != w {
42 t.Errorf("got %v, want %v, input %v", g, w, x)
43 return false
44 }
45 return true
46 })
47 }
48
49 func testInt32x4Reduce(t *testing.T, f func(_ archsimd.Int32x4) int32, want func(_ []int32) int32) {
50 n := 4
51 t.Helper()
52 forSlice(t, int32s, n, func(x []int32) bool {
53 t.Helper()
54 a := archsimd.LoadInt32x4(x)
55 g := f(a)
56 w := want(x)
57 if g != w {
58 t.Errorf("got %v, want %v, input %v", g, w, x)
59 return false
60 }
61 return true
62 })
63 }
64
65 func testInt64x2Reduce(t *testing.T, f func(_ archsimd.Int64x2) int64, want func(_ []int64) int64) {
66 n := 2
67 t.Helper()
68 forSlice(t, int64s, n, func(x []int64) bool {
69 t.Helper()
70 a := archsimd.LoadInt64x2(x)
71 g := f(a)
72 w := want(x)
73 if g != w {
74 t.Errorf("got %v, want %v, input %v", g, w, x)
75 return false
76 }
77 return true
78 })
79 }
80
81 func testUint8x16Reduce(t *testing.T, f func(_ archsimd.Uint8x16) uint8, want func(_ []uint8) uint8) {
82 n := 16
83 t.Helper()
84 forSlice(t, uint8s, n, func(x []uint8) bool {
85 t.Helper()
86 a := archsimd.LoadUint8x16(x)
87 g := f(a)
88 w := want(x)
89 if g != w {
90 t.Errorf("got %v, want %v, input %v", g, w, x)
91 return false
92 }
93 return true
94 })
95 }
96
97 func testUint16x8Reduce(t *testing.T, f func(_ archsimd.Uint16x8) uint16, want func(_ []uint16) uint16) {
98 n := 8
99 t.Helper()
100 forSlice(t, uint16s, n, func(x []uint16) bool {
101 t.Helper()
102 a := archsimd.LoadUint16x8(x)
103 g := f(a)
104 w := want(x)
105 if g != w {
106 t.Errorf("got %v, want %v, input %v", g, w, x)
107 return false
108 }
109 return true
110 })
111 }
112
113 func testUint32x4Reduce(t *testing.T, f func(_ archsimd.Uint32x4) uint32, want func(_ []uint32) uint32) {
114 n := 4
115 t.Helper()
116 forSlice(t, uint32s, n, func(x []uint32) bool {
117 t.Helper()
118 a := archsimd.LoadUint32x4(x)
119 g := f(a)
120 w := want(x)
121 if g != w {
122 t.Errorf("got %v, want %v, input %v", g, w, x)
123 return false
124 }
125 return true
126 })
127 }
128
129 func testUint64x2Reduce(t *testing.T, f func(_ archsimd.Uint64x2) uint64, want func(_ []uint64) uint64) {
130 n := 2
131 t.Helper()
132 forSlice(t, uint64s, n, func(x []uint64) bool {
133 t.Helper()
134 a := archsimd.LoadUint64x2(x)
135 g := f(a)
136 w := want(x)
137 if g != w {
138 t.Errorf("got %v, want %v, input %v", g, w, x)
139 return false
140 }
141 return true
142 })
143 }
144
145 func testFloat32x4Reduce(t *testing.T, f func(_ archsimd.Float32x4) float32, want func(_ []float32) float32) {
146 n := 4
147 t.Helper()
148 forSlice(t, float32s, n, func(x []float32) bool {
149 t.Helper()
150 a := archsimd.LoadFloat32x4(x)
151 g := f(a)
152 w := want(x)
153 if g != w && !(math.IsNaN(float64(g)) && math.IsNaN(float64(w))) {
154 t.Errorf("got %v, want %v, input %v", g, w, x)
155 return false
156 }
157 return true
158 })
159 }
160
161 func testFloat64x2Reduce(t *testing.T, f func(_ archsimd.Float64x2) float64, want func(_ []float64) float64) {
162 n := 2
163 t.Helper()
164 forSlice(t, float64s, n, func(x []float64) bool {
165 t.Helper()
166 a := archsimd.LoadFloat64x2(x)
167 g := f(a)
168 w := want(x)
169 if g != w && !(math.IsNaN(float64(g)) && math.IsNaN(float64(w))) {
170 t.Errorf("got %v, want %v, input %v", g, w, x)
171 return false
172 }
173 return true
174 })
175 }
176
View as plain text
