Merge pull request #498 from SciML/myb/cuarray

ChrisRackauckas · web-flow · commit a114c4fca4cd · 2020-04-30T16:11:28.000-04:00
Add default_factorize &amp; `\` support for CuArray
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "DiffEqBase"
 uuid = "2b5f629d-d688-5b77-993f-72d75c75574e"
 authors = ["Chris Rackauckas <accounts@chrisrackauckas.com>"]
-version = "6.30.4"
+version = "6.31.0"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
diff --git a/src/init.jl b/src/init.jl
@@ -183,7 +183,11 @@ function __init__()
       _x = UpperTriangular(_qr.R) \ (_qr.Q' * reshape(b,length(b),1))
       x .= vec(_x)
       CuArrays.unsafe_free!(_x)
+      return x
     end
+    # make `\` work
+    LinearAlgebra.ldiv!(F::CuArrays.CUSOLVER.CuQR, b::CuArrays.CuArray) = (x = similar(b); ldiv!(x, F, b); x)
+    default_factorize(A::CuArrays.CuArray) = qr(A)
     function findall_events(affect!,affect_neg!,prev_sign::CuArrays.CuArray,next_sign::CuArrays.CuArray)
       hasaffect::Bool = affect! !== nothing
       hasaffectneg::Bool = affect_neg! !== nothing
diff --git a/src/linear_nonlinear.jl b/src/linear_nonlinear.jl
@@ -1,4 +1,8 @@
 using LinearAlgebra, SparseArrays, SuiteSparse
+
+# This is only used for oop stiff solvers
+default_factorize(A) = lu(A)
+
 mutable struct LinSolveFactorize{F}
   factorization::F
   A

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,8 @@`
`1`	`1`	`using LinearAlgebra, SparseArrays, SuiteSparse`
	`2`	`+`
	`3`	`+# This is only used for oop stiff solvers`
	`4`	`+default_factorize(A) = lu(A)`
	`5`	`+`
`2`	`6`	`mutable struct LinSolveFactorize{F}`
`3`	`7`	`factorization::F`
`4`	`8`	`A`