Skip to content

Commit f5c6ad2

Browse files
committed
Fix Zygote
1 parent eff8ff5 commit f5c6ad2

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

DifferentiationInterface/ext/DifferentiationInterfaceZygoteExt/DifferentiationInterfaceZygoteExt.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ end
159159

160160
function DI.hvp(
161161
f,
162-
prep::DI.ForwardOverReverseHVPPrep,
162+
prep::DI.ForwardOverAnythingHVPPrep,
163163
backend::AutoZygote,
164164
x,
165165
tx::NTuple,
@@ -171,7 +171,7 @@ end
171171
function DI.hvp!(
172172
f,
173173
tg::NTuple,
174-
prep::DI.ForwardOverReverseHVPPrep,
174+
prep::DI.ForwardOverAnythingHVPPrep,
175175
backend::AutoZygote,
176176
x,
177177
tx::NTuple,
@@ -184,7 +184,7 @@ end
184184

185185
function DI.gradient_and_hvp(
186186
f,
187-
prep::DI.ForwardOverReverseHVPPrep,
187+
prep::DI.ForwardOverAnythingHVPPrep,
188188
backend::AutoZygote,
189189
x,
190190
tx::NTuple,
@@ -199,7 +199,7 @@ function DI.gradient_and_hvp!(
199199
f,
200200
grad,
201201
tg::NTuple,
202-
prep::DI.ForwardOverReverseHVPPrep,
202+
prep::DI.ForwardOverAnythingHVPPrep,
203203
backend::AutoZygote,
204204
x,
205205
tx::NTuple,

0 commit comments

Comments
 (0)