Skip to content

Commit 605b12e

Browse files
committed
Fix Enzyme
1 parent cf73f95 commit 605b12e

File tree

1 file changed

+1
-1
lines changed
  • DifferentiationInterface/ext/DifferentiationInterfaceEnzymeExt

1 file changed

+1
-1
lines changed

DifferentiationInterface/ext/DifferentiationInterfaceEnzymeExt/utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ force_annotation(f::F) where {F} = Const(f)
5454
end
5555

5656
@inline function _translate(
57-
backend::AutoEnzyme, ::Mode, ::Val{B}, c::Union{DI.Cache,DI.ConstantOrCache}
57+
backend::AutoEnzyme, ::Mode, ::Val{B}, c::Union{DI.Cache,DI.GeneralizedConstantOrCache}
5858
) where {B}
5959
if B == 1
6060
return Duplicated(DI.unwrap(c), make_zero(DI.unwrap(c)))

0 commit comments

Comments
 (0)