From 990fee978273cd2b05240661d70b7231e87b86e8 Mon Sep 17 00:00:00 2001 From: Vedant Puri Date: Sun, 2 Jul 2023 16:36:41 -0400 Subject: [PATCH] rm diffeqarrayoperator --- ext/LinearSolveCUDAExt.jl | 2 +- src/default.jl | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/ext/LinearSolveCUDAExt.jl b/ext/LinearSolveCUDAExt.jl index 08acdaa28..5af942e04 100644 --- a/ext/LinearSolveCUDAExt.jl +++ b/ext/LinearSolveCUDAExt.jl @@ -20,7 +20,7 @@ function LinearSolve.do_factorization(alg::CudaOffloadFactorization, A, b, u) A isa Union{AbstractMatrix, AbstractSciMLOperator} || error("LU is not defined for $(typeof(A))") - if A isa Union{MatrixOperator, DiffEqArrayOperator} + if A isa MatrixOperator A = A.A end diff --git a/src/default.jl b/src/default.jl index 3eb67627a..c4c869662 100644 --- a/src/default.jl +++ b/src/default.jl @@ -23,8 +23,7 @@ end # For SciML algorithms already using `defaultalg`, all assume square matrix. defaultalg(A, b) = defaultalg(A, b, OperatorAssumptions(true)) -function defaultalg(A::Union{DiffEqArrayOperator, MatrixOperator}, b, - assump::OperatorAssumptions) +function defaultalg(A::MatrixOperator, b, assump::OperatorAssumptions) defaultalg(A.A, b, assump) end