Skip to content

Commit

Permalink
hotfix isgpu removal
Browse files Browse the repository at this point in the history
  • Loading branch information
ChrisRackauckas committed May 18, 2022
1 parent 90458f8 commit 34d1dfe
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "LinearSolve"
uuid = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
authors = ["SciML"]
version = "1.16.0"
version = "1.16.1"

[deps]
ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
Expand Down
8 changes: 4 additions & 4 deletions src/default.jl
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ function defaultalg(A,b)
# Special case on Arrays: avoid BLAS for RecursiveFactorization.jl when
# it makes sense according to the benchmarks, which is dependent on
# whether MKL or OpenBLAS is being used
if (A === nothing && !isgpu(b)) || A isa Matrix
if (A === nothing && b isa AbstractGPUArray) || A isa Matrix
if (A === nothing || eltype(A) <: Union{Float32,Float64,ComplexF32,ComplexF64}) &&
ArrayInterface.can_setindex(b)
if length(b) <= 10
Expand Down Expand Up @@ -39,7 +39,7 @@ function defaultalg(A,b)

# This catches the case where A is a CuMatrix
# Which does not have LU fully defined
elseif isgpu(A) || isgpu(b)
elseif A isa AbstractGPUArray || b isa AbstractGPUArray
alg = QRFactorization(false)

# Not factorizable operator, default to only using A*x
Expand Down Expand Up @@ -100,7 +100,7 @@ function SciMLBase.solve(cache::LinearCache, alg::Nothing,

# This catches the case where A is a CuMatrix
# Which does not have LU fully defined
elseif isgpu(A)
elseif A isa AbstractGPUArray
alg = QRFactorization(false)
SciMLBase.solve(cache, alg, args...; kwargs...)

Expand Down Expand Up @@ -158,7 +158,7 @@ function init_cacheval(alg::Nothing, A, b, u, Pl, Pr, maxiters, abstol, reltol,

# This catches the case where A is a CuMatrix
# Which does not have LU fully defined
elseif isgpu(A)
elseif A isa AbstractGPUArray
alg = QRFactorization(false)
init_cacheval(alg, A, b, u, Pl, Pr, maxiters, abstol, reltol, verbose)

Expand Down

0 comments on commit 34d1dfe

Please sign in to comment.