Struct BsrMatrix 

pub struct BsrMatrix<T: GpuFloat> { /* private fields */ }

A sparse matrix in Block Sparse Row (BSR) format, stored on GPU.

The matrix has shape (rows, cols) where rows and cols are multiples of block_dim. It contains nnz_blocks non-zero dense blocks.

Implementations

impl<T: GpuFloat> BsrMatrix<T>

pub fn from_host( rows: u32, cols: u32, block_dim: u32, row_ptr: &[i32], col_idx: &[i32], values: &[T], ) -> SparseResult<Self>

Creates a BSR matrix from host-side arrays, uploading to GPU.

Arguments

• rows – Number of rows (must be a multiple of block_dim).
• cols – Number of columns (must be a multiple of block_dim).
• block_dim – Size of each dense block.
• row_ptr – Block row pointer array of length block_rows + 1.
• col_idx – Block column indices of length nnz_blocks.
• values – Dense block values (row-major within each block).

Errors

Returns SparseError::InvalidFormat if dimensions or array lengths are inconsistent.

pub fn to_host(&self) -> SparseResult<(Vec<i32>, Vec<i32>, Vec<T>)>

Downloads the BSR arrays from GPU to host memory.

Errors

Returns SparseError::Cuda on transfer failure.

pub fn rows(&self) -> u32

Returns the number of rows.

pub fn cols(&self) -> u32

Returns the number of columns.

pub fn nnz_blocks(&self) -> u32

Returns the number of non-zero blocks.

pub fn block_dim(&self) -> u32

Returns the block dimension.

pub fn block_rows(&self) -> u32

Returns the number of block rows.

pub fn block_cols(&self) -> u32

Returns the number of block columns.

pub fn scalar_nnz(&self) -> u32

Returns the total scalar non-zero count.

pub fn row_ptr(&self) -> &DeviceBuffer<i32>

Returns a reference to the block row pointer device buffer.

pub fn col_idx(&self) -> &DeviceBuffer<i32>

Returns a reference to the block column index device buffer.

pub fn values(&self) -> &DeviceBuffer<T>

Returns a reference to the values device buffer.