ferrum_kernels/backend/mod.rs
1//! Unified Backend trait for CUDA, Metal, and CPU compute.
2//!
3//! Each backend implements the same set of transformer-layer primitives
4//! (GEMM, norms, RoPE, attention, activations). `layer_forward()` and
5//! `ModelRunner` are generic over `Backend`, so one forward path serves
6//! all hardware targets.
7
8mod traits;
9pub use traits::*;
10
11pub mod cpu;
12
13#[cfg(feature = "metal")]
14pub mod metal;
15
16#[cfg(feature = "cuda")]
17pub mod cuda;