Code coverage not tracking custom EnzymeRules.augmented_primal and EnzymeRules.reverse implementations

kongdd · October 23, 2025, 3:08am

Directly call not work for unknown reason:

test/test-reverse_hard.jl

master


      
          autodiff(
            Reverse,
            solver,
            Const,
            Duplicated(make_zero(state_init), d_res), # 输入
            Const(f!),
            Const(state_init),
            Duplicated(params, d_param)               # 输出
          )

Finally, I tested those function in a hard way:

github.com/jl-pkgs/ModelAutoGrads.jl

test/test-reverse_hard.jl

master


      
          config = RevConfigWidth{1,false,false,(false, false, false, false, false),false,false}()
          func = Const{typeof(fixed_point!)}(ModelAutoGrad.fixed_point!)
          
          primal = EnzymeRules.augmented_primal(
            config,
            func,
            Const{Nothing},
            Duplicated(make_zero(state_init), d_res), # 输入
            Const(f!),
            Const(state_init),
            Duplicated(params, d_param)               # 输出
          )
          tape = primal.tape
          
          EnzymeRules.reverse(
            config,
            func,
            Const{Nothing},
            tape,
            Duplicated(make_zero(state_init), d_res), # 输入

This file has been truncated. show original

Topic		Replies	Views
How can we create more basic Enzyme custom rules? Numerics question , autodiff , enzyme , advice	3	330	March 27, 2024
Trouble writing custom rule in Enzyme: `AssertionError: !(overwritten[end])` Modelling & Simulations physics , reversediff , autodiff , geometric-algebra , enzyme	3	336	February 20, 2024
ChainRules: Replacing DiffRules in the Julia AD world Package Announcements package	43	4430	March 12, 2019
Enzyme custom rule alloactions Performance autodiff , allocations , enzyme	0	55	April 28, 2025
What's the state of Automatic Differentiation in Julia January 2023? General Usage autodiff	41	12567	June 21, 2024

Code coverage not tracking custom EnzymeRules.augmented_primal and EnzymeRules.reverse implementations

Related topics