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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

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

use crate::context::pc_context::PcConfig;
use crate::error::KError;
use crate::preconditioner::Preconditioner;

pub fn try_build(cfg: &PcConfig) -> Result<Option<Box<dyn Preconditioner>>, KError> {
    use crate::preconditioner::builders as b;

    match cfg {
        PcConfig::None => Ok(None),
        PcConfig::Jacobi => b::build_jacobi().map(Some),
        PcConfig::BlockJacobi { block } => b::build_block_jacobi(*block).map(Some),

        PcConfig::Ilu0 { conditioning } => {
            b::build_ilu0_with_conditioning(conditioning.clone()).map(Some)
        }
        PcConfig::Iluk {
            level,
            conditioning,
        } => b::build_iluk_with_conditioning(*level, conditioning.clone()).map(Some),
        PcConfig::Ilut {
            drop_tol,
            max_fill,
            reordering,
            conditioning,
        } => b::build_ilut_with_conditioning(
            *drop_tol,
            *max_fill,
            reordering.clone(),
            conditioning.clone(),
        )
        .map(Some),
        PcConfig::Ilutp {
            drop_tol,
            max_fill,
            perm_tol,
            reordering,
            conditioning,
        } => b::build_ilutp_with_conditioning(
            *max_fill,
            *drop_tol,
            *perm_tol,
            reordering.clone(),
            conditioning.clone(),
        )
        .map(Some),
        PcConfig::Milu0 { conditioning } => {
            b::build_milu0_with_conditioning(conditioning.clone()).map(Some)
        }

        PcConfig::Sor {
            omega,
            sweeps,
            mat_side,
        } => b::build_sor(*omega, *sweeps, *mat_side).map(Some),

        PcConfig::Chebyshev {
            degree,
            eig_lo,
            eig_hi,
        } => b::build_chebyshev(*degree, *eig_lo, *eig_hi).map(Some),

        PcConfig::Asm {
            overlap,
            subdomain_hint,
            block_solver,
            mode,
            weighting,
            inner_pc,
        } => b::build_asm(
            *overlap,
            *subdomain_hint,
            block_solver.clone(),
            mode.clone(),
            weighting.clone(),
            *inner_pc,
        )
        .map(Some),
        PcConfig::Amg {
            config,
            conditioning,
        } => {
            let mut cfg = config.clone();
            cfg.conditioning = conditioning.clone();
            b::build_amg(cfg).map(Some)
        }
        PcConfig::ApproxInv {
            kind,
            levels,
            max_per_col,
            drop_tol,
            reg,
            max_cond,
            parallel,
        } => {
            use crate::preconditioner::approxinv_csr::{
                ApproxInvKind, ApproxInvParams, FsaiCsr, SpaiCsr,
            };
            let params = ApproxInvParams {
                kind: *kind,
                levels: *levels,
                max_per_col: *max_per_col,
                drop_tol: *drop_tol,
                reg: *reg,
                max_cond: *max_cond,
                parallel: *parallel,
            };
            let pc: Box<dyn Preconditioner> = match *kind {
                ApproxInvKind::FSAI => Box::new(FsaiCsr::new_with_params(params)),
                ApproxInvKind::SPAI => Box::new(SpaiCsr::new_with_params(params)),
            };
            Ok(Some(pc))
        }

        PcConfig::Lu => {
            #[cfg(feature = "dense-direct")]
            {
                b::build_lu().map(Some)
            }
            #[cfg(not(feature = "dense-direct"))]
            {
                Ok(None)
            }
        }
        PcConfig::Qr => {
            #[cfg(feature = "dense-direct")]
            {
                b::build_qr().map(Some)
            }
            #[cfg(not(feature = "dense-direct"))]
            {
                Ok(None)
            }
        }
        #[cfg(feature = "superlu_dist")]
        PcConfig::SuperLuDist => b::build_superlu_dist().map(Some),
    }
}