4141 end
4242 function ChainRulesCore. rrule (:: typeof (identity), x)
4343 function identity_pullback (ȳ)
44- return (NO_FIELDS , ȳ)
44+ return (NoTangent () , ȳ)
4545 end
4646 return x, identity_pullback
4747 end
6767 x̄_ret = InplaceableThunk (
6868 @thunk (ȳ), ā -> (inplace_used = true ; ā .+ = ȳ)
6969 )
70- return (NO_FIELDS , x̄_ret)
70+ return (NoTangent () , x̄_ret)
7171 end
7272 return identity (x), identity_pullback
7373 end
9393 function my_identity_pullback (ȳ)
9494 # only the in-place part is incorrect
9595 x̄_ret = InplaceableThunk (@thunk (ȳ), ā -> ā .+ = 200 .* ȳ)
96- return (NO_FIELDS , x̄_ret)
96+ return (NoTangent () , x̄_ret)
9797 end
9898 return my_identity (x), my_identity_pullback
9999 end
106106 @testset " check inferred" begin
107107 ChainRulesCore. frule ((_, Δx), :: typeof (f_inferrable), x) = (x, Δx)
108108 function ChainRulesCore. rrule (:: typeof (f_inferrable), x)
109- f_inferrable_pullback (Δy) = (NO_FIELDS , Δy)
109+ f_inferrable_pullback (Δy) = (NoTangent () , Δy)
110110 return x, f_inferrable_pullback
111111 end
112112
123123 return (x, x > 0 ? Float64 (Δx) : Float32 (Δx))
124124 end
125125 function ChainRulesCore. rrule (:: typeof (f_noninferrable_frule), x)
126- f_noninferrable_frule_pullback (Δy) = (NO_FIELDS , Δy)
126+ f_noninferrable_frule_pullback (Δy) = (NoTangent () , Δy)
127127 return x, f_noninferrable_frule_pullback
128128 end
129129
@@ -144,10 +144,10 @@ end
144144 ChainRulesCore. frule ((_, Δx), :: typeof (f_noninferrable_rrule), x) = (x, Δx)
145145 function ChainRulesCore. rrule (:: typeof (f_noninferrable_rrule), x)
146146 if x > 0
147- f_noninferrable_rrule_pullback (Δy) = (NO_FIELDS , Δy)
147+ f_noninferrable_rrule_pullback (Δy) = (NoTangent () , Δy)
148148 return x, f_noninferrable_rrule_pullback
149149 else
150- return x, _ -> (NO_FIELDS , Δy) # this is not hit by the used point
150+ return x, _ -> (NoTangent () , Δy) # this is not hit by the used point
151151 end
152152 end
153153
167167 @testset " check not inferred in pullback" begin
168168 function ChainRulesCore. rrule (:: typeof (f_noninferrable_pullback), x)
169169 function f_noninferrable_pullback_pullback (Δy)
170- return (NO_FIELDS , x > 0 ? Float64 (Δy) : Float32 (Δy))
170+ return (NoTangent () , x > 0 ? Float64 (Δy) : Float32 (Δy))
171171 end
172172 return x, f_noninferrable_pullback_pullback
173173 end
182182 function ChainRulesCore. rrule (:: typeof (f_noninferrable_thunk), x, y)
183183 function f_noninferrable_thunk_pullback (Δz)
184184 ∂x = @thunk (x > 0 ? Float64 (Δz) : Float32 (Δz))
185- return (NO_FIELDS , ∂x, Δz)
185+ return (NoTangent () , ∂x, Δz)
186186 end
187187 return x + y, f_noninferrable_thunk_pullback
188188 end
198198 return (x > 0 ? Float64 (x) : Float32 (x), x > 0 ? Float64 (Δx) : Float32 (Δx))
199199 end
200200 function ChainRulesCore. rrule (:: typeof (f_inferrable_pullback_only), x)
201- f_inferrable_pullback_only_pullback (Δy) = (NO_FIELDS , oftype (x, Δy))
201+ f_inferrable_pullback_only_pullback (Δy) = (NoTangent () , oftype (x, Δy))
202202 return x > 0 ? Float64 (x) : Float32 (x), f_inferrable_pullback_only_pullback
203203 end
204204 test_frule (f_inferrable_pullback_only, 2.0 ; check_inferred= true )
212212 # define rrule using ChainRulesCore's v0.9.0 convention, conjugating the derivative
213213 # in the rrule
214214 function ChainRulesCore. rrule (:: typeof (sinconj), x)
215- sinconj_pullback (ΔΩ) = (NO_FIELDS , conj (cos (x)) * ΔΩ)
215+ sinconj_pullback (ΔΩ) = (NoTangent () , conj (cos (x)) * ΔΩ)
216216 return sin (x), sinconj_pullback
217217 end
218218
225225 ChainRulesCore. frule ((_, dx, dy), :: typeof (fst), x, y) = (x, dx)
226226 function ChainRulesCore. rrule (:: typeof (fst), x, y)
227227 function fst_pullback (Δx)
228- return (NO_FIELDS , Δx, ZeroTangent ())
228+ return (NoTangent () , Δx, ZeroTangent ())
229229 end
230230 return x, fst_pullback
231231 end
242242 @testset " single input, multiple output" begin
243243 simo (x) = (x, 2 x)
244244 function ChainRulesCore. rrule (simo, x)
245- simo_pullback ((a, b)) = (NO_FIELDS , a .+ 2 .* b)
245+ simo_pullback ((a, b)) = (NoTangent () , a .+ 2 .* b)
246246 return simo (x), simo_pullback
247247 end
248248 function ChainRulesCore. frule ((_, ẋ), simo, x)
264264 ChainRulesCore. frule ((_, dx), :: typeof (first), xs:: Tuple ) = (first (xs), first (dx))
265265 function ChainRulesCore. rrule (:: typeof (first), x:: Tuple )
266266 function first_pullback (Δx)
267- return (NO_FIELDS , Tangent {typeof(x)} (Δx, falses (length (x) - 1 )... ))
267+ return (NoTangent () , Tangent {typeof(x)} (Δx, falses (length (x) - 1 )... ))
268268 end
269269 return first (x), first_pullback
270270 end
294294 ChainRulesCore. frule ((_, Δx, _), :: typeof (fsymtest), x, s) = (x, Δx)
295295 function ChainRulesCore. rrule (:: typeof (fsymtest), x, s)
296296 function fsymtest_pullback (Δx)
297- return NO_FIELDS , Δx, NoTangent ()
297+ return NoTangent () , Δx, NoTangent ()
298298 end
299299 return x, fsymtest_pullback
300300 end
314314 end
315315 function ChainRulesCore. rrule (:: typeof (futestkws), x; err= true )
316316 function futestkws_pullback (Δx)
317- return (NO_FIELDS , Δx)
317+ return (NoTangent () , Δx)
318318 end
319319 return futestkws (x; err= err), futestkws_pullback
320320 end
348348 end
349349 function ChainRulesCore. rrule (:: typeof (fbtestkws), x, y; err= true )
350350 function fbtestkws_pullback (Δx)
351- return (NO_FIELDS , Δx, ZeroTangent ())
351+ return (NoTangent () , Δx, ZeroTangent ())
352352 end
353353 return fbtestkws (x, y; err= err), fbtestkws_pullback
354354 end
381381
382382 function ChainRulesCore. rrule (:: typeof (primalapprox), x)
383383 function primalapprox_pullback (Δx)
384- return (NO_FIELDS , Δx)
384+ return (NoTangent () , Δx)
385385 end
386386 return x + sqrt (eps (x)), primalapprox_pullback
387387 end
@@ -391,21 +391,21 @@ end
391391 end
392392
393393 @testset " frule with mutation" begin
394- function ChainRulesCore. frule ((_, ẋ ), :: typeof (finplace!), x; y= [1 ])
394+ function ChainRulesCore. frule ((_, ẋ ), :: typeof (finplace!), x; y= [1 ])
395395 y[1 ] *= 2
396396 x .*= y[1 ]
397- ẋ .*= 2 # hardcoded to match y defined below
398- return x, ẋ
397+ ẋ .*= 2 # hardcoded to match y defined below
398+ return x, ẋ
399399 end
400400
401401 # these pass in tangents explictly so that we can check them after
402402 x = randn (3 )
403- ẋ = [4.0 , 5.0 , 6.0 ]
404- xcopy, ẋcopy = copy (x), copy (ẋ )
403+ ẋ = [4.0 , 5.0 , 6.0 ]
404+ xcopy, ẋcopy = copy (x), copy (ẋ )
405405 y = [1 , 2 ]
406- test_frule (finplace!, x ⊢ ẋ ; fkwargs= (y= y,))
406+ test_frule (finplace!, x ⊢ ẋ ; fkwargs= (y= y,))
407407 @test x == xcopy
408- @test ẋ == ẋcopy
408+ @test ẋ == ẋcopy
409409 @test y == [1 , 2 ]
410410 end
411411
450450 ∂iter = TestIterator (
451451 ∂data, Base. IteratorSize (iter), Base. IteratorEltype (iter)
452452 )
453- return (NO_FIELDS , ∂iter)
453+ return (NoTangent () , ∂iter)
454454 end
455455 return iterfun (iter), iterfun_pullback
456456 end
471471 end
472472 function ChainRulesCore. rrule (:: typeof (my_identity1), x)
473473 function identity_pullback (ȳ)
474- return (NO_FIELDS , ȳ)
474+ return (NoTangent () , ȳ)
475475 end
476476 return 2.5 * x, identity_pullback
477477 end
487487 end
488488 function ChainRulesCore. rrule (:: typeof (my_identity2), x)
489489 function identity_pullback (ȳ)
490- return (NO_FIELDS , 31.8 * ȳ)
490+ return (NoTangent () , 31.8 * ȳ)
491491 end
492492 return x, identity_pullback
493493 end
505505
506506 rev_trouble ((x, y)) = y
507507 function ChainRulesCore. rrule (:: typeof (rev_trouble), (x, y):: P ) where {P}
508- rev_trouble_pullback (ȳ) = (NO_FIELDS , Tangent {P} (ZeroTangent (), ȳ))
508+ rev_trouble_pullback (ȳ) = (NoTangent () , Tangent {P} (ZeroTangent (), ȳ))
509509 return y, rev_trouble_pullback
510510 end
511511 test_rrule (rev_trouble, (3 , 3.0 ) ⊢ Tangent {Tuple{Int,Float64}} (ZeroTangent (), 1.0 ))
517517 function foo_pullback (Δy)
518518 da = zeros (size (a))
519519 da[i] = Δy
520- return NO_FIELDS , da, ZeroTangent ()
520+ return NoTangent () , da, ZeroTangent ()
521521 end
522522 return foo (a, i), foo_pullback
523523 end
0 commit comments