From 99ff84f27aafed1748aeca11bdc136c61794495a Mon Sep 17 00:00:00 2001 From: Cheng Gong Date: Wed, 25 Oct 2023 13:11:20 -0500 Subject: [PATCH] [EnzymeExt] Prefix AbstractKrylovSubspaceMethod to avoid undefvarerror --- ext/LinearSolveEnzymeExt.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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