Skip to content

Commit

Permalink
simplify gpu code
Browse files Browse the repository at this point in the history
  • Loading branch information
ChrisRackauckas committed Oct 4, 2023
1 parent affc130 commit 9ea818d
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions ext/LinearSolveCUDAExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -17,13 +17,6 @@ function SciMLBase.solve!(cache::LinearSolve.LinearCache, alg::CudaOffloadFactor
end

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}
A = A.A
end

fact = lu(CUDA.CuArray(A))
return fact
end
Expand Down

0 comments on commit 9ea818d

Please sign in to comment.