Skip to content

Commit 8378e26

Browse files
committed
Additional fixes
1 parent 90fcbae commit 8378e26

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

ext/AbstractDifferentiationChainRulesCoreExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import AbstractDifferentiation as AD
44
using ChainRulesCore: ChainRulesCore
55

66
AD.@primitive function pullback_function(ba::AD.ReverseRuleConfigBackend, f, xs...)
7-
_, back = ChainRulesCore.rrule_via_ad(ab.ruleconfig, f, xs...)
7+
_, back = ChainRulesCore.rrule_via_ad(ba.ruleconfig, f, xs...)
88
pullback(vs) = Base.tail(back(vs))
99
pullback(vs::Tuple{Any}) = Base.tail(back(first(vs)))
1010
return pullback

ext/AbstractDifferentiationTrackerExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ else
77
using ..Tracker: Tracker
88
end
99

10-
function AD.second_lowest(::TrackerBackend)
10+
function AD.second_lowest(::AD.TrackerBackend)
1111
return throw(ArgumentError("Tracker backend does not support nested differentiation."))
1212
end
1313

0 commit comments

Comments
 (0)