rstsr 0.7.2

An n-Dimension Rust Tensor Toolkit
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
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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156

pub mod rstsr_traits {
    pub use rstsr_core::prelude::rstsr_traits::*;
    #[cfg(feature = "linalg")]
    pub mod linalg {
        pub use rstsr_linalg_traits::prelude::rstsr_traits::*;
    }

    #[cfg(feature = "sci")]
    pub mod sci {
        pub use rstsr_sci_traits::prelude::rstsr_traits::*;
    }

    #[cfg(feature = "tblis")]
    pub mod tblis {
        pub use rstsr_tblis::prelude::rstsr_traits::*;
    }
}

pub mod rstsr_structs {
    pub use rstsr_core::prelude::rstsr_structs::*;
    #[cfg(feature = "linalg")]
    pub mod linalg {
        pub use rstsr_linalg_traits::prelude::rstsr_structs::*;
    }

    #[cfg(feature = "sci")]
    pub mod sci {
        pub use rstsr_sci_traits::prelude::rstsr_structs::*;
    }

    #[cfg(feature = "aocl")]
    pub use rstsr_aocl::DeviceAOCL;
    #[cfg(feature = "blis")]
    pub use rstsr_blis::DeviceBLIS;
    #[cfg(feature = "kml")]
    pub use rstsr_kml::DeviceKML;
    #[cfg(feature = "mkl")]
    pub use rstsr_mkl::DeviceMKL;
    #[cfg(feature = "openblas")]
    pub use rstsr_openblas::DeviceOpenBLAS;

    #[cfg(all(
        feature = "openblas",
        not(feature = "mkl"),
        not(feature = "blis"),
        not(feature = "aocl"),
        not(feature = "kml")
    ))]
    pub type DeviceBLAS = DeviceOpenBLAS;
    #[cfg(all(
        not(feature = "openblas"),
        feature = "mkl",
        not(feature = "blis"),
        not(feature = "aocl"),
        not(feature = "kml")
    ))]
    pub type DeviceBLAS = DeviceMKL;
    #[cfg(all(
        not(feature = "openblas"),
        not(feature = "mkl"),
        feature = "blis",
        not(feature = "aocl"),
        not(feature = "kml")
    ))]
    pub type DeviceBLAS = DeviceBLIS;
    #[cfg(all(
        not(feature = "openblas"),
        not(feature = "mkl"),
        not(feature = "blis"),
        feature = "aocl",
        not(feature = "kml")
    ))]
    pub type DeviceBLAS = DeviceAOCL;
    #[cfg(all(
        not(feature = "openblas"),
        not(feature = "mkl"),
        not(feature = "blis"),
        not(feature = "aocl"),
        feature = "kml"
    ))]
    pub type DeviceBLAS = DeviceKML;
}

pub mod rstsr_funcs {
    pub use rstsr_core::prelude::rstsr_funcs::*;
    #[cfg(feature = "linalg")]
    pub mod linalg {
        pub use rstsr_linalg_traits::prelude::rstsr_funcs::*;
    }

    #[cfg(feature = "sci")]
    pub mod sci {
        pub use rstsr_sci_traits::prelude::rstsr_funcs::*;
    }

    #[cfg(feature = "tblis")]
    pub mod tblis {
        pub use rstsr_tblis::prelude::rstsr_funcs::*;
    }
}

pub mod rstsr_macros {
    pub use rstsr_core::prelude::rstsr_macros::*;
}

// final re-exports
pub use rstsr_macros::*;
pub use rstsr_structs::*;
pub use rstsr_traits::*;

#[cfg(feature = "linalg")]
pub mod linalg {
    pub use rstsr_linalg_traits::prelude::rstsr_funcs::*;
    pub use rstsr_linalg_traits::prelude::rstsr_structs::*;
    pub use rstsr_linalg_traits::prelude::rstsr_traits::*;
}

#[cfg(feature = "sci")]
pub mod sci {
    pub use rstsr_sci_traits::prelude::rstsr_funcs::*;
    pub use rstsr_sci_traits::prelude::rstsr_mods::*;
    pub use rstsr_sci_traits::prelude::rstsr_structs::*;
    pub use rstsr_sci_traits::prelude::rstsr_traits::*;
}

#[cfg(feature = "tblis")]
pub mod tblis {
    pub use rstsr_tblis::prelude::rstsr_funcs::*;
    pub use rstsr_tblis::prelude::rstsr_traits::*;
}

pub mod rt {
    pub use super::rstsr_funcs;
    pub use super::rstsr_macros;
    pub use super::rstsr_structs;
    pub use super::rstsr_traits;

    pub use super::rstsr_funcs::*;
    pub use super::rstsr_macros::*;
    pub use super::rstsr_structs::*;
    pub use super::rstsr_traits::*;

    #[cfg(feature = "linalg")]
    pub use super::linalg;

    #[cfg(feature = "sci")]
    pub use super::sci;

    #[cfg(feature = "tblis")]
    pub use super::tblis;

    #[cfg(feature = "use_blas_traits")]
    pub use rstsr_blas_traits::prelude as blas;

    pub use rstsr_core::prelude::rt::{Error, Result};
}