opensrdk-linear-algebra 0.8.5

Standard linear algebra library using blas and lapack for OpenSRDK toolchain.
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use super::SymmetricTridiagonalMatrix;
use crate::{matrix::ge::Matrix, matrix::*};
use lapack::dstevd;

impl SymmetricTridiagonalMatrix {
    /// # Eigen decomposition
    /// return (lambda, pt)
    pub fn stevd(self) -> Result<(Vec<f64>, Matrix), MatrixError> {
        let (mut d, mut e) = self.eject();
        let n = d.len();
        let mut z = Matrix::new(n, n);
        let lwork = 1.max(1 + 4 * n + n.pow(2));
        let mut work = vec![0.0; lwork];
        let liwork = 1.max(3 + 5 * n);
        let mut iwork = vec![0; liwork];
        let mut info = 0;

        let n = n as i32;

        unsafe {
            dstevd(
                'V' as u8,
                n,
                &mut d,
                &mut e,
                &mut z.elems_mut(),
                n,
                &mut work,
                lwork as i32,
                &mut iwork,
                liwork as i32,
                &mut info,
            )
        }

        match info {
            0 => Ok((d, z)),
            _ => Err(MatrixError::LapackRoutineError {
                routine: "dstev".to_owned(),
                info,
            }),
        }
    }
}