lapack-traits 0.4.0

Generic real/complex scalar trait wrappers for BLAS and LAPACK routines
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
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

use crate::Scalar;
use cblas::{Layout, Part, Transpose, cherk, zherk};
use num_complex::Complex32 as c32;
use num_complex::Complex64 as c64;
use crate::blas::syrk::Tsyrk;

pub trait Therk : Scalar + Tsyrk{
    /// Hermitian rank k update
    /// For real scalars, herk casts Transpose::Conjugate into Transpose::Ordinary and is
    /// completely equivalent to syrk
    unsafe fn herk(layout: Layout,
                   uplo: Part,
                   trans: Transpose,
                   n: i32,
                   k: i32,
                   alpha: Self::Real,
                   a: &[Self],
                   lda: i32,
                   beta: Self::Real,
                   c: &mut [Self],
                   ldc: i32,
    );
}

macro_rules! impl_therk_real{
    ($N: ty) => (
        impl Therk for $N{
            #[inline]
            unsafe fn herk(
                layout: Layout,
                uplo: Part,
                trans: Transpose,
                n: i32,
                k: i32,
                alpha: Self::Real,
                a: &[Self],
                lda: i32,
                beta: Self::Real,
                c: &mut [Self],
                ldc: i32
            )
            {
                let real_trans = if let Transpose::Conjugate = trans { Transpose::Ordinary }
                                 else { trans };
                <$N>::syrk(layout, uplo, real_trans, n, k, alpha, a, lda, beta, c, ldc)
            }
        }
    )
}

macro_rules! impl_therk_complex{
    ($N: ty, $therk: path) => (
        impl Therk for $N{
            #[inline]
            unsafe fn herk(
                layout: Layout,
                uplo: Part,
                trans: Transpose,
                n: i32,
                k: i32,
                alpha: Self::Real,
                a: &[Self],
                lda: i32,
                beta: Self::Real,
                c: &mut [Self],
                ldc: i32
            )
            {
                $therk(layout, uplo, trans, n, k, alpha, a, lda, beta, c, ldc)
            }
        }
    )
}

impl_therk_real!(f32);
impl_therk_real!(f64);
impl_therk_complex!(c32, cherk);
impl_therk_complex!(c64, zherk);