Skip to content

Commit 99ff84f

Browse files
committed
[EnzymeExt] Prefix AbstractKrylovSubspaceMethod to avoid undefvarerror
1 parent 4ee3e1e commit 99ff84f

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)