Improve AD module

[omlins]

No description provided.

[omlins]

@vchuravy / @aelligp : concerning your previous contribution, could you please tell me why for example here g! is not promoted to Const(g!) either by you're promotion function or enzyme itself (here in the documentation f is not annotated):

Enzyme.autodiff_deferred(Enzyme.Reverse, g!, Const, DuplicatedNoNeed(A_ref, Ā_ref), DuplicatedNoNeed(B_ref, B̄_ref), Const(a))`

(I changed it back in the test. This is why the CI fails)

[vchuravy]

IIRC Enzyme no longer auto-promotes arguments, it might still take functions to be Const, but I am afk so I can't double-check.

ParallelStencil.jl/src/ParallelKernel/EnzymeExt/autodiff_gpu.jl

Line 28 in 0390670

args = promote_to_const(args...)

The custom promotion is only applied to the splatted in args and not the first arg which likely is the function.

[omlins]

IIRC Enzyme no longer auto-promotes arguments, it might still take functions to be Const, but I am afk so I can't double-check.

ParallelStencil.jl/src/ParallelKernel/EnzymeExt/autodiff_gpu.jl

Line 28 in 0390670

args = promote_to_const(args...)

The custom promotion is only applied to the splatted in args and not the first arg which likely is the function.

Thanks for your reply @vchuravy !