gplu 0.1.0

Sparse LU factorization with partial pivoting
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::scalar::Scalar;
use crate::solve::lu_solve;
use crate::LU;
use rayon::iter::ParallelIterator;
use rayon::slice::ParallelSliceMut;

/// Solve `Ax=b` for one or more right-hand-sides given the numeric
/// factorization of A from `factor`.
pub fn par_solve<S: Scalar + Send + Sync>(
    lu: &LU<S>,
    rhs: &mut [S],
    trans: bool,
) -> Result<(), String> {
    let n = lu.n;
    if rhs.len() % n != 0 {
        return Err(format!(
            "len rhs ({}) must be a multiple of n ({})",
            rhs.len(),
            n
        ));
    }

    rhs.par_chunks_exact_mut(n)
        .try_for_each_with(vec![S::zero(); n], |work, b| -> Result<(), String> {
            lu_solve(lu, b, work, trans)
        })
}