Set const to avoid type instability issues

SciML · Sep 12, 2024 · 210f656 · 210f656
1 parent 9158744
commit 210f656
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/test/enzyme.jl b/test/enzyme.jl
@@ -176,27 +176,28 @@ Enzyme.autodiff(Reverse, f3, Duplicated(copy(A), dA), Duplicated(copy(b1), db1),
 =#
 
 A = rand(n, n);
+const _A = rand(n, n);
 dA = zeros(n, n);
-b1 = rand(n);
+const _b1 = rand(n);
 for alg in (
     LUFactorization(),
     RFLUFactorization()    # KrylovJL_GMRES(), fails
 )
     @show alg
     function fb(b)
-        prob = LinearProblem(A, b)
+        prob = LinearProblem(_A, b)
 
         sol1 = solve(prob, alg)
 
         sum(sol1.u)
     end
     fb(b1)
 
-    fd_jac = FiniteDiff.finite_difference_jacobian(fb, b1) |> vec
+    fd_jac = FiniteDiff.finite_difference_jacobian(fb, _b1) |> vec
     @show fd_jac
 
     en_jac = map(onehot(b1)) do db1
-        eres = Enzyme.autodiff(Forward, fb, Duplicated(copy(b1), db1))
+        eres = Enzyme.autodiff(Forward, fb, Duplicated(copy(_b1), db1))
         eres[1]
     end |> collect
     @show en_jac