Make the algorithm type be inactive

SciML · Sep 12, 2024 · c5fe914 · c5fe914
1 parent 39b3717
commit c5fe914
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ext/LinearSolveEnzymeExt.jl b/ext/LinearSolveEnzymeExt.jl
@@ -8,8 +8,7 @@ using Enzyme
 
 using EnzymeCore
 
-@inline EnzymeCore.EnzymeRules.inactive_type(v::Type{LinearSolve.KrylovJL}) = true
-@inline EnzymeCore.EnzymeRules.inactive_type(v::Type{LinearSolve.Krylov.GmresSolver}) = true
+@inline EnzymeCore.EnzymeRules.inactive_type(::Type{<:LinearSolve.SciMLLinearSolveAlgorithm}) = true
 
 function EnzymeCore.EnzymeRules.forward(
         func::Const{typeof(LinearSolve.init)}, ::Type{RT}, prob::EnzymeCore.Annotation{LP},