@@ -12,9 +12,9 @@ SIMDTests.test("init(repeating:)") {
1212 func foo1( x: Float ) -> SIMD4 < Float > {
1313 return SIMD4 < Float > ( repeating: 2 * x)
1414 }
15- let ( val1, bp1 ) = valueWithPullback ( at: 5 , in: foo1)
15+ let ( val1, pb1 ) = valueWithPullback ( at: 5 , in: foo1)
1616 expectEqual ( SIMD4 < Float > ( 10 , 10 , 10 , 10 ) , val1)
17- expectEqual ( 8 , bp1 ( g) )
17+ expectEqual ( 8 , pb1 ( g) )
1818}
1919
2020// FIXME(TF-1103): Derivative registration does not yet support
@@ -26,9 +26,9 @@ SIMDTests.test("Sum") {
2626 func foo1(x: SIMD4<Float>) -> Float {
2727 return x.sum()
2828 }
29- let (val1, bp1 ) = valueWithPullback(at: a, in: foo1)
29+ let (val1, pb1 ) = valueWithPullback(at: a, in: foo1)
3030 expectEqual(10, val1)
31- expectEqual(SIMD4<Float>(3, 3, 3, 3), bp1 (3))
31+ expectEqual(SIMD4<Float>(3, 3, 3, 3), pb1 (3))
3232}
3333*/
3434
@@ -39,9 +39,9 @@ SIMDTests.test("Identity") {
3939 func foo1( x: SIMD4 < Float > ) -> SIMD4 < Float > {
4040 return x
4141 }
42- let ( val1, bp1 ) = valueWithPullback ( at: a, in: foo1)
42+ let ( val1, pb1 ) = valueWithPullback ( at: a, in: foo1)
4343 expectEqual ( a, val1)
44- expectEqual ( g, bp1 ( g) )
44+ expectEqual ( g, pb1 ( g) )
4545}
4646
4747SIMDTests . test ( " Negate " ) {
@@ -51,9 +51,9 @@ SIMDTests.test("Negate") {
5151 func foo1( x: SIMD4 < Float > ) -> SIMD4 < Float > {
5252 return - x
5353 }
54- let ( val1, bp1 ) = valueWithPullback ( at: a, in: foo1)
54+ let ( val1, pb1 ) = valueWithPullback ( at: a, in: foo1)
5555 expectEqual ( - a, val1)
56- expectEqual ( - g, bp1 ( g) )
56+ expectEqual ( - g, pb1 ( g) )
5757}
5858
5959SIMDTests . test ( " subscript " ) {
@@ -63,9 +63,9 @@ SIMDTests.test("subscript") {
6363 return x [ 3 ]
6464 }
6565
66- let ( val1, bp1 ) = valueWithPullback ( at: a, in: foo1)
66+ let ( val1, pb1 ) = valueWithPullback ( at: a, in: foo1)
6767 expectEqual ( 4 , val1)
68- expectEqual ( SIMD4 < Float > ( 0 , 0 , 0 , 7 ) , bp1 ( 7 ) )
68+ expectEqual ( SIMD4 < Float > ( 0 , 0 , 0 , 7 ) , pb1 ( 7 ) )
6969}
7070
7171SIMDTests . test ( " Addition " ) {
@@ -76,25 +76,25 @@ SIMDTests.test("Addition") {
7676 func foo1( x: SIMD4 < Float > , y: SIMD4 < Float > ) -> SIMD4 < Float > {
7777 return x + y
7878 }
79- let ( val1, bp1 ) = valueWithPullback ( at: a, a, in: foo1)
79+ let ( val1, pb1 ) = valueWithPullback ( at: a, a, in: foo1)
8080 expectEqual ( SIMD4 < Float > ( 2 , 4 , 6 , 8 ) , val1)
81- expectEqual ( ( g, g) , bp1 ( g) )
81+ expectEqual ( ( g, g) , pb1 ( g) )
8282
8383 // SIMD + Scalar
8484 func foo2( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
8585 return x + y
8686 }
87- let ( val2, bp2 ) = valueWithPullback ( at: a, 5 , in: foo2)
87+ let ( val2, pb2 ) = valueWithPullback ( at: a, 5 , in: foo2)
8888 expectEqual ( SIMD4 < Float > ( 6 , 7 , 8 , 9 ) , val2)
89- expectEqual ( ( g, 4 ) , bp2 ( g) )
89+ expectEqual ( ( g, 4 ) , pb2 ( g) )
9090
9191 // Scalar + SIMD
9292 func foo3( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
9393 return y + x
9494 }
95- let ( val3, bp3 ) = valueWithPullback ( at: a, 5 , in: foo3)
95+ let ( val3, pb3 ) = valueWithPullback ( at: a, 5 , in: foo3)
9696 expectEqual ( SIMD4 < Float > ( 6 , 7 , 8 , 9 ) , val3)
97- expectEqual ( ( g, 4 ) , bp3 ( g) )
97+ expectEqual ( ( g, 4 ) , pb3 ( g) )
9898}
9999
100100SIMDTests . test ( " Subtraction " ) {
@@ -105,25 +105,25 @@ SIMDTests.test("Subtraction") {
105105 func foo1( x: SIMD4 < Float > , y: SIMD4 < Float > ) -> SIMD4 < Float > {
106106 return x - y
107107 }
108- let ( val1, bp1 ) = valueWithPullback ( at: a, a, in: foo1)
108+ let ( val1, pb1 ) = valueWithPullback ( at: a, a, in: foo1)
109109 expectEqual ( SIMD4 < Float > ( 0 , 0 , 0 , 0 ) , val1)
110- expectEqual ( ( g, - g) , bp1 ( g) )
110+ expectEqual ( ( g, - g) , pb1 ( g) )
111111
112112 // SIMD - Scalar
113113 func foo2( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
114114 return x - y
115115 }
116- let ( val2, bp2 ) = valueWithPullback ( at: a, 5 , in: foo2)
116+ let ( val2, pb2 ) = valueWithPullback ( at: a, 5 , in: foo2)
117117 expectEqual ( SIMD4 < Float > ( - 4 , - 3 , - 2 , - 1 ) , val2)
118- expectEqual ( ( g, - 4 ) , bp2 ( g) )
118+ expectEqual ( ( g, - 4 ) , pb2 ( g) )
119119
120120 // Scalar - SIMD
121121 func foo3( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
122122 return y - x
123123 }
124- let ( val3, bp3 ) = valueWithPullback ( at: a, 5 , in: foo3)
124+ let ( val3, pb3 ) = valueWithPullback ( at: a, 5 , in: foo3)
125125 expectEqual ( SIMD4 < Float > ( 4 , 3 , 2 , 1 ) , val3)
126- expectEqual ( ( - g, 4 ) , bp3 ( g) )
126+ expectEqual ( ( - g, 4 ) , pb3 ( g) )
127127}
128128
129129SIMDTests . test ( " Multiplication " ) {
@@ -134,25 +134,25 @@ SIMDTests.test("Multiplication") {
134134 func foo1( x: SIMD4 < Float > , y: SIMD4 < Float > ) -> SIMD4 < Float > {
135135 return x * y
136136 }
137- let ( val1, bp1 ) = valueWithPullback ( at: a, a, in: foo1)
137+ let ( val1, pb1 ) = valueWithPullback ( at: a, a, in: foo1)
138138 expectEqual ( a * a, val1)
139- expectEqual ( ( a, a) , bp1 ( g) )
139+ expectEqual ( ( a, a) , pb1 ( g) )
140140
141141 // SIMD * Scalar
142142 func foo2( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
143143 return x * y
144144 }
145- let ( val2, bp2 ) = valueWithPullback ( at: a, 5 , in: foo2)
145+ let ( val2, pb2 ) = valueWithPullback ( at: a, 5 , in: foo2)
146146 expectEqual ( a * 5 , val2)
147- expectEqual ( ( SIMD4 < Float > ( 5 , 5 , 5 , 5 ) , 10 ) , bp2 ( g) )
147+ expectEqual ( ( SIMD4 < Float > ( 5 , 5 , 5 , 5 ) , 10 ) , pb2 ( g) )
148148
149149 // Scalar * SIMD
150150 func foo3( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
151151 return y * x
152152 }
153- let ( val3, bp3 ) = valueWithPullback ( at: a, 5 , in: foo3)
153+ let ( val3, pb3 ) = valueWithPullback ( at: a, 5 , in: foo3)
154154 expectEqual ( a * 5 , val3)
155- expectEqual ( ( SIMD4 < Float > ( 5 , 5 , 5 , 5 ) , 10 ) , bp3 ( g) )
155+ expectEqual ( ( SIMD4 < Float > ( 5 , 5 , 5 , 5 ) , 10 ) , pb3 ( g) )
156156}
157157
158158SIMDTests . test ( " Division " ) {
@@ -165,29 +165,29 @@ SIMDTests.test("Division") {
165165 }
166166 let dlhs1 = g / a
167167 let drhs1 = - 1 / a
168- let ( val1, bp1 ) = valueWithPullback ( at: a, a, in: foo1)
168+ let ( val1, pb1 ) = valueWithPullback ( at: a, a, in: foo1)
169169 expectEqual ( a / a, val1)
170- expectEqual ( ( dlhs1, drhs1) , bp1 ( g) )
170+ expectEqual ( ( dlhs1, drhs1) , pb1 ( g) )
171171
172172 // SIMD / Scalar
173173 func foo2( x: SIMD4 < Float > , y: Float ) -> SIMD4 < Float > {
174174 return x / y
175175 }
176176 let dlhs2 = g / 5
177177 let drhs2 = ( - a / 25 * g) . sum ( )
178- let ( val2, bp2 ) = valueWithPullback ( at: a, 5 , in: foo2)
178+ let ( val2, pb2 ) = valueWithPullback ( at: a, 5 , in: foo2)
179179 expectEqual ( a / 5 , val2)
180- expectEqual ( ( dlhs2, drhs2) , bp2 ( g) )
180+ expectEqual ( ( dlhs2, drhs2) , pb2 ( g) )
181181
182182 // Scalar / SIMD
183183 func foo3( x: Float , y: SIMD4 < Float > ) -> SIMD4 < Float > {
184184 return x / y
185185 }
186186 let dlhs3 = ( g / a) . sum ( )
187187 let drhs3 = - 5 / ( a*a) * g
188- let ( val3, bp3 ) = valueWithPullback ( at: 5 , a, in: foo3)
188+ let ( val3, pb3 ) = valueWithPullback ( at: 5 , a, in: foo3)
189189 expectEqual ( 5 / a, val3)
190- expectEqual ( ( dlhs3, drhs3) , bp3 ( g) )
190+ expectEqual ( ( dlhs3, drhs3) , pb3 ( g) )
191191}
192192
193193SIMDTests . test ( " Generics " ) {
@@ -203,9 +203,9 @@ SIMDTests.test("Generics") {
203203 return SIMDType . init ( repeating: x)
204204 }
205205 func simd3Init( x: Double ) -> SIMD3 < Double > { testInit ( x: x) }
206- let ( val1, bp1 ) = valueWithPullback ( at: 10 , in: simd3Init)
206+ let ( val1, pb1 ) = valueWithPullback ( at: 10 , in: simd3Init)
207207 expectEqual ( SIMD3 < Double > ( 10 , 10 , 10 ) , val1)
208- expectEqual ( 3 , bp1 ( g) )
208+ expectEqual ( 3 , pb1 ( g) )
209209
210210 // SIMDType + SIMDType
211211 func testAddition< Scalar, SIMDType: SIMD > ( lhs: SIMDType , rhs: SIMDType )
@@ -220,9 +220,9 @@ SIMDTests.test("Generics") {
220220 func simd3Add( lhs: SIMD3 < Double > , rhs: SIMD3 < Double > ) -> SIMD3 < Double > {
221221 return testAddition ( lhs: lhs, rhs: rhs)
222222 }
223- let ( val2, bp2 ) = valueWithPullback ( at: a, a, in: simd3Add)
223+ let ( val2, pb2 ) = valueWithPullback ( at: a, a, in: simd3Add)
224224 expectEqual ( SIMD3 < Double > ( 2 , 4 , 6 ) , val2)
225- expectEqual ( ( g, g) , bp2 ( g) )
225+ expectEqual ( ( g, g) , pb2 ( g) )
226226
227227 // Scalar - SIMDType
228228 func testSubtraction< Scalar, SIMDType: SIMD > ( lhs: Scalar , rhs: SIMDType )
@@ -237,9 +237,9 @@ SIMDTests.test("Generics") {
237237 func simd3Subtract( lhs: Double , rhs: SIMD3 < Double > ) -> SIMD3 < Double > {
238238 return testSubtraction ( lhs: lhs, rhs: rhs)
239239 }
240- let ( val3, bp3 ) = valueWithPullback ( at: 5 , a, in: simd3Subtract)
240+ let ( val3, pb3 ) = valueWithPullback ( at: 5 , a, in: simd3Subtract)
241241 expectEqual ( SIMD3 < Double > ( 4 , 3 , 2 ) , val3)
242- expectEqual ( ( 3 , SIMD3 < Double > ( - 1 , - 1 , - 1 ) ) , bp3 ( g) )
242+ expectEqual ( ( 3 , SIMD3 < Double > ( - 1 , - 1 , - 1 ) ) , pb3 ( g) )
243243
244244 // SIMDType * Scalar
245245 func testMultipication< Scalar, SIMDType: SIMD > ( lhs: SIMDType , rhs: Scalar )
@@ -254,9 +254,9 @@ SIMDTests.test("Generics") {
254254 func simd3Multiply( lhs: SIMD3 < Double > , rhs: Double ) -> SIMD3 < Double > {
255255 return testMultipication ( lhs: lhs, rhs: rhs)
256256 }
257- let ( val4, bp4 ) = valueWithPullback ( at: a, 5 , in: simd3Multiply)
257+ let ( val4, pb4 ) = valueWithPullback ( at: a, 5 , in: simd3Multiply)
258258 expectEqual ( SIMD3 < Double > ( 5 , 10 , 15 ) , val4)
259- expectEqual ( ( SIMD3 < Double > ( 5 , 5 , 5 ) , 6 ) , bp4 ( g) )
259+ expectEqual ( ( SIMD3 < Double > ( 5 , 5 , 5 ) , 6 ) , pb4 ( g) )
260260
261261 // FIXME(TF-1103): Derivative registration does not yet support
262262 /*
@@ -269,9 +269,9 @@ SIMDTests.test("Generics") {
269269 return x.sum()
270270 }
271271 func simd3Sum(x: SIMD3<Double>) -> Double { testSum(x: x) }
272- let (val5, bp5 ) = valueWithPullback(at: a, in: simd3Sum)
272+ let (val5, pb5 ) = valueWithPullback(at: a, in: simd3Sum)
273273 expectEqual(6, val5)
274- expectEqual(SIMD3<Double>(7, 7, 7), bp5 (7))
274+ expectEqual(SIMD3<Double>(7, 7, 7), pb5 (7))
275275 */
276276}
277277
0 commit comments