Skip to content

Commit 9630121

Browse files
fix other algorithms
1 parent cbb5f1d commit 9630121

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

ext/LinearSolveEnzymeExt.jl

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -114,17 +114,16 @@ function EnzymeCore.EnzymeRules.reverse(config, func::Const{typeof(LinearSolve.s
114114
z = if _linsolve.cacheval isa Factorization
115115
_linsolve.cacheval' \ dy
116116
elseif linsolve.cacheval isa Tuple && linsolve.cacheval[1] isa Factorization
117-
linsolve.cacheval[1]' \ dy
117+
_linsolve.cacheval[1]' \ dy
118118
elseif linsolve.alg isa AbstractKrylovSubspaceMethod
119119
# Doesn't modify `A`, so it's safe to just reuse it
120-
invprob = LinearSolve.LinearProblem(transpose(linsolve.A), dy)
120+
invprob = LinearSolve.LinearProblem(transpose(_linsolve.A), dy)
121121
solve(invprob;
122-
abstol = linsolve.val.abstol,
123-
reltol = linsolve.val.reltol,
124-
verbose = linsolve.val.verbose,
125-
isfresh = freshbefore)
122+
abstol = _linsolve.val.abstol,
123+
reltol = _linsolve.val.reltol,
124+
verbose = _linsolve.val.verbose)
126125
else
127-
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")
126+
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")
128127
end
129128

130129
dA .-= z * transpose(y)

0 commit comments

Comments
 (0)