Merge pull request #433 from JuliaArrays/cudss

Add CUDSS extension for LU factorization instances

Author: ChrisRackauckas

Project.toml

@@ -12,6 +12,7 @@ SuiteSparse = "4607b0f0-06f3-5cda-b6b1-a6196a1729e9"
 BandedMatrices = "aae01518-5342-5314-be14-df237901396f"
 BlockBandedMatrices = "ffab5731-97b5-5995-9138-79e8c1846df0"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+CUDSS = "45b445bb-4962-46a0-9369-b4df9d0f772e"
 ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
@@ -22,6 +23,7 @@ Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 ArrayInterfaceBandedMatricesExt = "BandedMatrices"
 ArrayInterfaceBlockBandedMatricesExt = "BlockBandedMatrices"
 ArrayInterfaceCUDAExt = "CUDA"
+ArrayInterfaceCUDSSExt = "CUDSS"
 ArrayInterfaceChainRulesExt = "ChainRules"
 ArrayInterfaceGPUArraysCoreExt = "GPUArraysCore"
 ArrayInterfaceReverseDiffExt = "ReverseDiff"

ext/ArrayInterfaceCUDSSExt.jl

@@ -0,0 +1,17 @@
+module ArrayInterfaceCUDSSExt
+
+using ArrayInterface
+using CUDSS
+
+function ArrayInterface.lu_instance(A::CUDSS.CuSparseMatrixCSR)
+    ArrayInterface.LinearAlgebra.checksquare(A)
+    fact = CudssSolver(A, "G", 'F')
+    T = eltype(A)
+    n = size(A,1)
+    x = CudssMatrix(T, n)
+    b = CudssMatrix(T, n)
+    cudss("analysis", fact, x, b)
+    fact
+end
+
+end