Struct PartitionedCsr 

pub struct PartitionedCsr {
    pub partitions: Vec<GpuCsrMatrix>,
    pub row_offsets: Vec<usize>,
    pub halo_rows: Vec<Vec<usize>>,
    pub n_total_rows: usize,
    pub n_cols: usize,
    /* private fields */
}

A CSR matrix partitioned across multiple logical workers using a row-striped decomposition with optional halo rows.

Each entry of partitions holds the local matrix for that worker, which covers [partition_row_start[w], partition_row_start[w] + partitions[w].n_rows).

Fields

partitions: Vec<GpuCsrMatrix>

Local matrix per worker (may include halo rows at the boundaries).

row_offsets: Vec<usize>

Global row index of the first owned (non-halo) row for each worker.

halo_rows: Vec<Vec<usize>>

For each worker: the list of global row indices that are ghost rows (owned by another worker but needed by this worker).

n_total_rows: usize

Total number of rows in the original matrix.

n_cols: usize

Number of columns in the original matrix.

Implementations

impl PartitionedCsr

pub fn from_csr(matrix: &GpuCsrMatrix, config: &DistributedCsrConfig) -> Self

Partition matrix into config.n_workers pieces using a row-striped decomposition.

Each partition gets its owned rows plus up to config.overlap halo rows from each adjacent partition.

pub fn spmv(&self, x: &[f64]) -> SparseResult<Vec<f64>>

Compute y = A * x using the distributed representation.

Each worker computes SpMV on its local matrix (which includes halo rows) and contributes only the owned rows to the global result.

Errors

Returns SparseError::DimensionMismatch when x.len() != n_cols.

pub fn to_csr(&self) -> GpuCsrMatrix

Reassemble all partitions’ owned rows into a single GpuCsrMatrix.

pub fn load_balance_quality(&self) -> f64

Measure load balance quality.

Returns std_dev(nnz_per_partition) / mean(nnz_per_partition). A value of 0.0 means perfect balance; lower is better.

Returns 0.0 for degenerate cases (0 or 1 workers, or 0 total nnz).

Trait Implementations

impl Clone for PartitionedCsr

fn clone(&self) -> PartitionedCsr

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PartitionedCsr

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.