Skip to content

Add direct sparse gpu constructors for COO, CSC, CSR and BSR matrices #3022

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
nHackel wants to merge 6 commits into
base: master
Choose a base branch
from
Draft

Add direct sparse gpu constructors for COO, CSC, CSR and BSR matrices #3022

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
1909ff3
Add direct sparse gpu constructors for COO, CSC, CSR and BSR
nHackel Jan 23, 2026
f9c2af8
Add direct sparse gpu constructors for COO, CSC, CSR and BSR
nHackel Jan 23, 2026
6ea82ca
Merge branch 'nh/sparseConstructors' of github.com:nHackel/CUDA.jl in…
nHackel Mar 13, 2026
a0968ab
Fix invalid array types GPUSparseMatrixX constructors
nHackel Mar 13, 2026
a6cfd9d
Fix invalid array types for GPUSparseMatrixCOO and BSR
nHackel Mar 13, 2026
423fbb8
Fix incorrect module name
nHackel Mar 13, 2026
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
14 changes: 12 additions & 2 deletions lib/cusparse/array.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,9 @@ mutable struct CuSparseMatrixCSC{Tv, Ti} <: GPUArrays.AbstractGPUSparseMatrixCSC
new{Tv, Ti}(colPtr, rowVal, nzVal, dims, length(nzVal))
end
end
function GPUArrays.GPUSparseMatrixCSC(colPtr::CuVector{Ti}, rowVal::CuVector{Ti}, nzVal::CuVector{Tv}, dims::NTuple{2,<:Integer}) where {Tv, Ti <: Integer}
return CuSparseMatrixCSC{Tv, Ti}(colPtr, rowVal, nzVal, dims)
end
CuSparseMatrixCSC{Tv, Ti}(csc::CuSparseMatrixCSC{Tv, Ti}) where {Tv, Ti} = csc

SparseArrays.rowvals(g::T) where {T<:CuSparseVector} = nonzeroinds(g)
Expand Down Expand Up @@ -94,7 +97,9 @@ mutable struct CuSparseMatrixCSR{Tv, Ti} <: GPUArrays.AbstractGPUSparseMatrixCSR
new{Tv, Ti}(rowPtr, colVal, nzVal, dims, length(nzVal))
end
end

function GPUArrays.GPUSparseMatrixCSR(rowPtr::CuVector{Ti}, colVal::CuVector{Ti}, nzVal::CuVector{Tv}, dims::NTuple{2,<:Integer}) where {Tv, Ti <: Integer}
return CuSparseMatrixCSR{Tv, Ti}(rowPtr, colVal, nzVal, dims)
end
CuSparseMatrixCSR{Tv, Ti}(csr::CuSparseMatrixCSR{Tv, Ti}) where {Tv, Ti} = csr
CuSparseMatrixCSR(A::CuSparseMatrixCSR) = A

Expand Down Expand Up @@ -147,6 +152,9 @@ mutable struct CuSparseMatrixBSR{Tv, Ti} <: AbstractCuSparseMatrix{Tv, Ti}
new{Tv, Ti}(rowPtr, colVal, nzVal, dims, blockDim, dir, nnz)
end
end
function GPUArrays.GPUSparseMatrixBSR(rowPtr::CuVector{Ti}, colVal::CuVector{Ti}, nzVal::CuVector{Tv}, dims::NTuple{2,<:Integer}, blockDim::Integer, args...) where {Tv, Ti <: Integer}
return CuSparseMatrixBSR{Tv, Ti}(rowPtr, colVal, nzVal, dims, blockDim, args...)
end

CuSparseMatrixBSR(A::CuSparseMatrixBSR) = A

Expand Down Expand Up @@ -177,7 +185,9 @@ mutable struct CuSparseMatrixCOO{Tv, Ti} <: AbstractCuSparseMatrix{Tv, Ti}
new{Tv, Ti}(rowInd,colInd,nzVal,dims,nnz)
end
end

function GPUArrays.GPUSparseMatrixCOO(rowInd::CuVector{Ti}, colInd::CuVector{Ti}, nzVal::CuVector{Tv}, dims::NTuple{2,<:Integer}) where {Tv, Ti <: Integer}
return CuSparseMatrixCOO{Tv, Ti}(rowInd, colInd, nzVal, dims)
end
CuSparseMatrixCOO(A::CuSparseMatrixCOO) = A

mutable struct CuSparseArrayCSR{Tv, Ti, N} <: GPUArrays.AbstractGPUSparseArray{Tv, Ti, N}
Expand Down