kryst 3.2.1

Krylov subspace and preconditioned iterative solvers for dense and sparse linear systems, with shared and distributed memory parallelism.
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

#![cfg(feature = "backend-faer")]

use super::*;
use crate::algebra::prelude::*;
#[cfg(not(feature = "complex"))]
use faer::Mat;

struct Dummy;

impl Preconditioner for Dummy {
    fn setup(&mut self, _pmat: &dyn LinOp<S = S>) -> Result<(), KError> {
        Ok(())
    }

    fn apply(&self, _side: PcSide, r: &[S], z: &mut [S]) -> Result<(), KError> {
        z.copy_from_slice(r);
        Ok(())
    }
}

#[test]
#[cfg(not(feature = "complex"))]
fn default_direct_solve_errors() {
    let mut pc = Dummy;
    let a = Mat::<f64>::zeros(1, 1);
    let mut x = [S::zero()];
    let rhs = [S::from_real(1.0)];
    let err = pc.direct_solve(&a, &rhs, &mut x).unwrap_err();
    match err {
        KError::SolveError(msg) => assert!(msg.contains("direct_solve not supported")),
        _ => panic!("unexpected error variant"),
    }
}

#[test]
fn default_apply_mut_forwards_to_apply() {
    use std::sync::atomic::{AtomicUsize, Ordering};
    struct CountPc {
        calls: AtomicUsize,
    }
    impl Preconditioner for CountPc {
        fn setup(&mut self, _a: &dyn LinOp<S = S>) -> Result<(), KError> {
            Ok(())
        }
        fn apply(&self, _side: PcSide, _x: &[S], _y: &mut [S]) -> Result<(), KError> {
            self.calls.fetch_add(1, Ordering::Relaxed);
            Ok(())
        }
    }
    let mut pc = CountPc {
        calls: AtomicUsize::new(0),
    };
    let x = [S::zero(); 2];
    let mut y = [S::zero(); 2];
    pc.apply_mut(PcSide::Left, &x, &mut y).unwrap();
    assert_eq!(pc.calls.load(Ordering::Relaxed), 1);
}

mod asm_amg;
#[cfg(feature = "mpi")]
mod asm_mpi;
mod block_jacobi;
mod classical;
mod coarsen;
mod deflation;
mod direct_apply;
mod ilu_csr;
mod ilu_history;
mod jacobi;
mod legacy_bridge;
mod near_nullspace;
mod nodal;
mod post_interp;
mod spd;