diff --git a/ext/LinearSolveEnzymeExt.jl b/ext/LinearSolveEnzymeExt.jl index 607189e10..fea332dcd 100644 --- a/ext/LinearSolveEnzymeExt.jl +++ b/ext/LinearSolveEnzymeExt.jl @@ -144,7 +144,7 @@ function EnzymeCore.EnzymeRules.reverse(config, func::Const{typeof(LinearSolve.s _linsolve.cacheval' \ dy elseif _linsolve.cacheval isa Tuple && _linsolve.cacheval[1] isa Factorization _linsolve.cacheval[1]' \ dy - elseif _linsolve.alg isa AbstractKrylovSubspaceMethod + elseif _linsolve.alg isa LinearSolve.AbstractKrylovSubspaceMethod # Doesn't modify `A`, so it's safe to just reuse it invprob = LinearSolve.LinearProblem(transpose(_linsolve.A), dy) solve(invprob; @@ -163,4 +163,4 @@ function EnzymeCore.EnzymeRules.reverse(config, func::Const{typeof(LinearSolve.s return (nothing,) end -end \ No newline at end of file +end