diff --git a/main.jl b/main.jl index 209c4ae..53cb56d 100644 --- a/main.jl +++ b/main.jl @@ -7,7 +7,7 @@ import FiniteDifferences: central_fdm import ForwardDiff import ReverseDiff import Mooncake -import Enzyme: set_runtime_activity, Forward, Reverse +import Enzyme: set_runtime_activity, Forward, Reverse, Const import Zygote # AD backends to test. @@ -18,8 +18,8 @@ ADTYPES = Dict( "ReverseDiffCompiled" => AutoReverseDiff(; compile=true), "MooncakeReverse" => AutoMooncake(), "MooncakeForward" => AutoMooncakeForward(), - "EnzymeForward" => AutoEnzyme(; mode=set_runtime_activity(Forward, true)), - "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse, true)), + "EnzymeForward" => AutoEnzyme(; mode=set_runtime_activity(Forward, true), function_annotation=Const), + "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse, true), function_annotation=Const), "Zygote" => AutoZygote(), )