just test finiteDiff vs Zygote

Author: cmrace

test/adjoint.jl

@@ -44,6 +44,10 @@ db12 = ForwardDiff.gradient(x -> f(eltype(x).(A), x), copy(b1))
 @test dA ≈ dA2
 @test db1 ≈ db12
 
+# Test complex numbers
+A = rand(n, n) + 1im*rand(n, n);
+b1 = rand(n) + 1im*rand(n, n);
+
 function f3(A, b1, b2; alg = KrylovJL_GMRES())
     prob = LinearProblem(A, b1)
     sol1 = solve(prob, alg)
@@ -66,6 +70,9 @@ db22 = FiniteDiff.finite_difference_gradient(
 @test db1 ≈ db12
 @test db2 ≈ db22
 
+A = rand(n, n);
+b1 = rand(n);
+
 function f4(A, b1, b2; alg = LUFactorization())
     prob = LinearProblem(A, b1)
     sol1 = solve(prob, alg; sensealg = LinearSolveAdjoint(; linsolve = KrylovJL_LSMR()))
@@ -85,9 +92,8 @@ db22 = ForwardDiff.gradient(x -> f4(eltype(x).(A), eltype(x).(b1), x), copy(b1))
 @test db1 ≈ db12
 @test db2 ≈ db22
 
-# Test complex numbers
-A = rand(n, n) + 1im * rand(n, n);
-b1 = rand(n) + 1im * rand(n);
+A = rand(n, n);
+b1 = rand(n);
 for alg in (
     LUFactorization(),
     RFLUFactorization(),
@@ -99,7 +105,7 @@ for alg in (
 
         sol1 = solve(prob, alg)
 
-        sum(abs2.(sol1.u))
+        sum(sol1.u)
     end
     fb(b1)
 
@@ -116,7 +122,7 @@ for alg in (
 
         sol1 = solve(prob, alg)
 
-        sum(abs2.(sol1.u))
+        sum(sol1.u)
     end
     fA(A)