Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
name = "ArrayInterface"
uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
version = "7.20.0"
version = "7.20.1"

[deps]
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
Expand All @@ -24,7 +24,7 @@ Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
ArrayInterfaceBandedMatricesExt = "BandedMatrices"
ArrayInterfaceBlockBandedMatricesExt = "BlockBandedMatrices"
ArrayInterfaceCUDAExt = "CUDA"
ArrayInterfaceCUDSSExt = "CUDSS"
ArrayInterfaceCUDSSExt = ["CUDSS", "CUDA"]
ArrayInterfaceChainRulesCoreExt = "ChainRulesCore"
ArrayInterfaceChainRulesExt = "ChainRules"
ArrayInterfaceGPUArraysCoreExt = "GPUArraysCore"
Expand All @@ -39,7 +39,7 @@ Adapt = "4"
BandedMatrices = "1"
BlockBandedMatrices = "0.13"
CUDA = "5"
CUDSS = "0.2, 0.3, 0.4"
CUDSS = "0.2, 0.3, 0.4, 0.5, 0.6"
ChainRules = "1"
ChainRulesCore = "1"
ChainRulesTestUtils = "1"
Expand Down
12 changes: 8 additions & 4 deletions ext/ArrayInterfaceCUDSSExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,18 @@ module ArrayInterfaceCUDSSExt

using ArrayInterface
using CUDSS
using CUDA

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)
n = size(A, 1)

# Use standard CUDA types (CuVector) instead of deprecated CudssMatrix
x = CUDA.CuVector{T}(undef, n)
b = CUDA.CuVector{T}(undef, n)

fact = CudssSolver(A, "G", 'F')
cudss("analysis", fact, x, b)
fact
end
Expand Down