Skip to main content

rstsr_common/layout/
mod.rs

1pub mod broadcast;
2pub mod dim;
3pub mod indexer;
4pub mod iterator;
5pub mod layoutbase;
6pub mod matmul;
7pub mod rearrangement;
8pub mod reshape;
9pub mod shape;
10pub mod slice;
11
12pub mod exports {
13    use super::*;
14    pub use broadcast::*;
15    pub use dim::*;
16    pub use indexer::*;
17    pub use iterator::*;
18    pub use layoutbase::*;
19    pub use matmul::*;
20    pub use rearrangement::*;
21    pub use reshape::*;
22    pub use shape::*;
23    pub use slice::*;
24
25    pub trait DimDevAPI: DimBaseAPI + DimShapeAPI + DimLayoutContigAPI {}
26
27    impl<const N: usize> DimDevAPI for Ix<N> {}
28    impl DimDevAPI for IxD {}
29
30    pub trait DimAPI:
31        DimDevAPI
32        + DimIntoAPI<IxD>
33        + DimIntoAPI<Ix0>
34        + DimIntoAPI<Ix1>
35        + DimIntoAPI<Ix2>
36        + DimIntoAPI<Ix3>
37        + DimIntoAPI<Ix4>
38        + DimIntoAPI<Ix5>
39        + DimIntoAPI<Ix6>
40        + DimIntoAPI<Ix7>
41        + DimIntoAPI<Ix8>
42        + DimIntoAPI<Ix9>
43        + DimIntoAPI<Self>
44    {
45    }
46
47    impl<const N: usize> DimAPI for Ix<N> {}
48    impl DimAPI for IxD {}
49}