ferrum-runtime 0.5.0

Backend implementations (Candle, CPU) for Ferrum inference
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

//! Backend implementations for different compute devices

#[cfg(feature = "candle")]
pub mod candle;

#[cfg(feature = "candle")]
pub mod candle_kernel_ops;

#[cfg(feature = "cpu")]
pub mod cpu;

// Re-export backend implementations
#[cfg(feature = "candle")]
pub use self::candle::*;

#[cfg(feature = "candle")]
pub use self::candle_kernel_ops::CandleKernelOps;

#[cfg(feature = "cpu")]
pub use self::cpu::*;