Missing EnzymeRule

[wsmoses]

ERROR: LoadError: MethodError: no method matching forward(::EnzymeCore.EnzymeRules.FwdConfigWidth{1, false, false, false, false}, ::Duplicated{FunctionWrappersWrappers.FunctionWrappersWrapper{Tuple{FunctionWrappers.FunctionWrapper{Nothing, Tuple{Vector{Float64}, Vector{Float64}, Vector{Float64}}}, FunctionWrappers.FunctionWrapper{Nothing, Tuple{Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}, Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}, Vector{Float64}}}, FunctionWrappers.FunctionWrapper{Nothing, Tuple{Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}, Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}, Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}}}, FunctionWrappers.FunctionWrapper{Nothing, Tuple{Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}, Vector{Float64}, Vector{ForwardDiff.Dual{ForwardDiff.Tag{NonlinearSolveBase.NonlinearSolveTag, Float64}, Float64, 1}}}}}, FunctionWrappersWrappers.AllowNonIsBits, FunctionWrappersWrappers.SingleCacheStorage}}, ::Type{Const{Nothing}}, ::Duplicated{Vector{Float64}}, ::Duplicated{Vector{Float64}}, ::Duplicated{Vector{Float64}})
The function `forward` exists, but no method is defined for this combination of argument types.

Closest candidates are:
  forward(::EnzymeCore.EnzymeRules.FwdConfig, ::Const{typeof(hypot)}, ::Any, ::Annotation, ::Annotation, ::Annotation, Annotation...) where N
   @ Enzyme /mnt3/wmoses/git/Enzyme.jl2/src/internal_rules/math.jl:6
  forward(::EnzymeCore.EnzymeRules.FwdConfigWidth{W, false, false, RuntimeActivity, StrongZero}, ::Const{<:FunctionWrappersWrappers.FunctionWrappersWrapper}, ::Type{<:Annotation}, ::Annotation...) where {W, N, RuntimeActivity, StrongZero}
   @ FunctionWrappersWrappersEnzymeExt ~/.julia/packages/FunctionWrappersWrappers/YcpKm/ext/FunctionWrappersWrappersEnzymeExt.jl:108
  forward(::Any, ::Const{<:KernelAbstractions.Kernel{KernelAbstractions.CPU}}, ::Type{Const{Nothing}}, ::Any...; ndrange, workgroupsize)
   @ EnzymeExt ~/.julia/packages/KernelAbstractions/ecO4B/ext/EnzymeCore08Ext.jl:14
  ...

Stacktrace:
  [1] custom_rule_method_error
    @ /mnt3/wmoses/git/Enzyme.jl2/src/rules/customrules.jl:974 [inlined]
  [2] __init
    @ ~/.julia/packages/NonlinearSolveFirstOrder/kipKw/src/solve.jl:134 [inlined]
  [3] #__solve#157
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:299
  [4] __solve
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:291 [inlined]
  [5] #solve_call#152
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:174 [inlined]
  [6] solve_call
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:140 [inlined]
  [7] solve_call
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:0 [inlined]
  [8] fwddiffejulia_solve_call_14784_inner_4wrap
    @ ~/.julia/packages/NonlinearSolveBase/gEITD/src/solve.jl:0
  [9] macro expansion
    @ /mnt3/wmoses/git/Enzyme.jl2/src/compiler.jl:6780 [inlined]
 [10] enzyme_call
    @ /mnt3/wmoses/git/Enzyme.jl2/src/compiler.jl:6259 [inlined]
 [11] ForwardModeThunk
    @ /mnt3/wmoses/git/Enzyme.jl2/src/compiler.jl:6159 [inlined]
 [12] macro expansion

@ChrisRackauckas can you have @ChrisRackauckas-Claude give this a go

cc @oscardssmith

[wsmoses]

comes from:

using Enzyme

 Enzyme.Compiler.VERBOSE_ERRORS[] = true

using NonlinearSolve, Enzyme, SciMLSensitivity
using LinearAlgebra

function f!(residual, u, p)
    nothing
end

function loss_inplace(p, cst)
    first(solve(NonlinearProblem(f!, cst, p), NewtonRaphson()).u)
end

# Enzyme.Compiler.DumpPostCheck[] = true

#Enzyme.Compiler.DumpPreOpt[] = true

Enzyme.API.printall!(true)
Enzyme.Compiler.DumpPostEnzyme[] = true

loss_inplace([2.0], [1.0])
dx =  Duplicated([2.0], [0.0])
cst = Duplicated([1.0], [0.0])
Enzyme.autodiff(Enzyme.Forward, loss_inplace, Duplicated, dx,cst)