Skip to content

Commit cb9e134

Browse files
dan-zhengdan-zheng
authored
[AutoDiff] NFC: fix "pullback" abbrevation. (swiftlang#32745)
"bp" -> "pb"
1 parent a11c06c commit cb9e134

File tree

1 file changed

+44
-44
lines changed

1 file changed

+44
-44
lines changed

‎test/AutoDiff/stdlib/simd.swift

Lines changed: 44 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -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

4747
SIMDTests.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

5959
SIMDTests.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

7171
SIMDTests.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

100100
SIMDTests.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

129129
SIMDTests.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

158158
SIMDTests.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

193193
SIMDTests.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

Comments
 (0)