Skip to content

Commit 6c501a5

Browse files
Merge pull request #402 from enigne/enzymeext_krylov
[EnzymeExt] Prefix AbstractKrylovSubspaceMethod to avoid undefvarerror
2 parents 4ee3e1e + 99ff84f commit 6c501a5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/LinearSolveEnzymeExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ function EnzymeCore.EnzymeRules.reverse(config, func::Const{typeof(LinearSolve.s
144144
_linsolve.cacheval' \ dy
145145
elseif _linsolve.cacheval isa Tuple && _linsolve.cacheval[1] isa Factorization
146146
_linsolve.cacheval[1]' \ dy
147-
elseif _linsolve.alg isa AbstractKrylovSubspaceMethod
147+
elseif _linsolve.alg isa LinearSolve.AbstractKrylovSubspaceMethod
148148
# Doesn't modify `A`, so it's safe to just reuse it
149149
invprob = LinearSolve.LinearProblem(transpose(_linsolve.A), dy)
150150
solve(invprob;
@@ -163,4 +163,4 @@ function EnzymeCore.EnzymeRules.reverse(config, func::Const{typeof(LinearSolve.s
163163
return (nothing,)
164164
end
165165

166-
end
166+
end

0 commit comments

Comments
 (0)