From df73d4d5682b99f19be81129772f897be7ba5cdb Mon Sep 17 00:00:00 2001 From: William Moses Date: Thu, 21 Aug 2025 10:06:06 +0200 Subject: [PATCH 1/2] Mark function as non-differentiable --- main.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.jl b/main.jl index 209c4ae..da956c6 100644 --- a/main.jl +++ b/main.jl @@ -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=Enzyme.Const), + "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse, true), function_annotation=Enzyme.Const), "Zygote" => AutoZygote(), ) From f000aaef8874632c0246cbcdb322a31ba0d8c982 Mon Sep 17 00:00:00 2001 From: William Moses Date: Thu, 21 Aug 2025 12:21:02 +0200 Subject: [PATCH 2/2] Update main.jl --- main.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main.jl b/main.jl index da956c6..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), function_annotation=Enzyme.Const), - "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse, true), function_annotation=Enzyme.Const), + "EnzymeForward" => AutoEnzyme(; mode=set_runtime_activity(Forward, true), function_annotation=Const), + "EnzymeReverse" => AutoEnzyme(; mode=set_runtime_activity(Reverse, true), function_annotation=Const), "Zygote" => AutoZygote(), )