From 85db9c34f78c796f50275d41f1a95bc1e0fb981b Mon Sep 17 00:00:00 2001 From: Samuel Omlin Date: Wed, 4 Dec 2024 15:39:06 +0100 Subject: [PATCH] add Const for Enzyme.autodiff_deferred function arg --- test/ParallelKernel/test_parallel.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/ParallelKernel/test_parallel.jl b/test/ParallelKernel/test_parallel.jl index 2f619538..a8594e23 100644 --- a/test/ParallelKernel/test_parallel.jl +++ b/test/ParallelKernel/test_parallel.jl @@ -114,7 +114,7 @@ macro compute_with_aliases(A) esc(:(ix + (iz -1)*size($A,1) return end @parallel configcall=f!(A, B, a) AD.autodiff_deferred!(Enzyme.Reverse, f!, Const, DuplicatedNoNeed(A, Ā), DuplicatedNoNeed(B, B̄), Const(a)) # NOTE: f! is automatically promoted to Const. - Enzyme.autodiff_deferred(Enzyme.Reverse, g!, Const, DuplicatedNoNeed(A_ref, Ā_ref), DuplicatedNoNeed(B_ref, B̄_ref), Const(a)) + Enzyme.autodiff_deferred(Enzyme.Reverse, Const(g!), Const, DuplicatedNoNeed(A_ref, Ā_ref), DuplicatedNoNeed(B_ref, B̄_ref), Const(a)) @test Array(Ā) ≈ Ā_ref @test Array(B̄) ≈ B̄_ref end