burn-rmexp-dyntensor 0.0.5

burn remote experiment dyntensor
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

use crate::errors::DynTensorError;

/// Dynamic to static rank dispatch handler.
pub trait RankHandler {
    type Output;

    /// Call the static-rank handler.
    fn call<const R: usize>(self) -> Result<Self::Output, DynTensorError>;
}

/// Dynamic rank dispatch.
///
/// Handles up to rank=12.
pub fn dispatch_rank<H: RankHandler>(
    rank: usize,
    handler: H,
) -> Result<H::Output, DynTensorError> {
    match rank {
        1 => handler.call::<1>(),
        2 => handler.call::<2>(),
        3 => handler.call::<3>(),
        4 => handler.call::<4>(),
        5 => handler.call::<5>(),
        6 => handler.call::<6>(),
        7 => handler.call::<7>(),
        8 => handler.call::<8>(),
        9 => handler.call::<9>(),
        10 => handler.call::<10>(),
        11 => handler.call::<11>(),
        12 => handler.call::<12>(),
        _ => Err(DynTensorError::UnsupportedRank {
            msg: "unsupported rank".to_string(),
            rank,
        }),
    }
}