Skip to content

Commit 1dd0535

Browse files
authored
Update autodiff_gpu.jl
1 parent c50ac4a commit 1dd0535

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/ParallelKernel/EnzymeExt/autodiff_gpu.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,14 @@ end
2828
function ParallelStencil.ParallelKernel.AD.autodiff_deferred!(mode, f, args...) # NOTE: minimal specialization is required to avoid overwriting the default method
2929
f = promote_to_const(f)
3030
args = promote_to_const(args...)
31-
Enzyme.autodiff_deferred(mode, f, Const, args...)
31+
Enzyme.autodiff_deferred(mode, f, Enzyme.Const, args...)
3232
return
3333
end
3434

3535
function ParallelStencil.ParallelKernel.AD.autodiff_deferred_thunk!(mode, f, args...) # NOTE: minimal specialization is required to avoid overwriting the default method
3636
f = promote_to_const(f)
3737
args = promote_to_const(args...)
38-
Enzyme.autodiff_deferred_thunk(mode, f, Const, args...)
38+
Enzyme.autodiff_deferred_thunk(mode, f, Enzyme.Const, args...)
3939
return
4040
end
4141

0 commit comments

Comments
 (0)