burn-lm-cli 0.0.1

Burn Large Models Engine - CLI.
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

#[derive(Debug, Clone, PartialEq, Eq, clap::ValueEnum, strum::Display, strum::EnumIter)]
pub(crate) enum BackendValues {
    // candle ----------------------------------------------------------------
    #[strum(to_string = "candle-accelerate")]
    CandleAccelerate,

    #[strum(to_string = "candle-cpu")]
    CandleCpu,

    #[cfg(not(target_os = "macos"))]
    #[strum(to_string = "candle-cuda")]
    CandleCuda,

    #[cfg(target_os = "macos")]
    #[strum(to_string = "candle-metal")]
    CandleMetal,

    // cuda ------------------------------------------------------------------
    #[cfg(not(target_os = "macos"))]
    #[strum(to_string = "cuda")]
    Cuda,

    // rocm -------------------------------------------------------------------
    #[cfg(target_os = "linux")]
    #[strum(to_string = "rocm")]
    Rocm,

    // ndarray ---------------------------------------------------------------
    #[strum(to_string = "ndarray")]
    Ndarray,

    #[strum(to_string = "ndarray-blas-accelerate")]
    NdarrayBlasAccelerate,

    #[strum(to_string = "ndarray-blas-netlib")]
    NdarrayBlasNetlib,

    #[strum(to_string = "ndarray-blas-openblas")]
    NdarrayBlasOpenblas,

    // libtorch --------------------------------------------------------------
    #[strum(to_string = "libtorch-cpu")]
    LibTorchCpu,

    #[strum(to_string = "libtorch")]
    LibTorch,

    // vulkan ----------------------------------------------------------------
    #[strum(to_string = "vulkan")]
    Vulkan,

    // Metal -----------------------------------------------------------------
    #[cfg(target_os = "macos")]
    #[strum(to_string = "metal")]
    Metal,

    // wgpu ------------------------------------------------------------------
    #[strum(to_string = "wgpu")]
    Wgpu,

    #[strum(to_string = "wgpu-cpu")]
    WgpuCpu,
}