Fix bug in handling Keylov solvers

SciML · Nov 7, 2023 · 4265053 · 4265053
1 parent 5abd3f2
commit 4265053
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/ext/LinearSolveEnzymeExt.jl b/ext/LinearSolveEnzymeExt.jl
@@ -34,7 +34,7 @@ function EnzymeCore.EnzymeRules.forward(func::Const{typeof(LinearSolve.solve!)},
     if RT <: Const
         return res
     end
-    if linsolve.alg isa LinearSolve.AbstractKrylovSubspaceMethod
+    if linsolve.val.alg isa LinearSolve.AbstractKrylovSubspaceMethod
         error("Algorithm $(_linsolve.alg) is currently not supported by Enzyme rules on LinearSolve.jl. Please open an issue on LinearSolve.jl detailing which algorithm is missing the adjoint handling")
     end
     b = deepcopy(linsolve.val.b)

diff --git a/test/enzyme.jl b/test/enzyme.jl
@@ -173,8 +173,7 @@ for alg in (
     RFLUFactorization(),
     # KrylovJL_GMRES(), fails
     )
-    alg_str = string(alg)
-    @show alg_str
+    @show alg
     function fb(b)
         prob = LinearProblem(A, b)