kime_cpu/lib.rs
1//! The CPU backend. FP32 reference kernels first, because every other backend is tested against them, then INT8 kernels for AVX2, AVX-512 VNNI, AMX and NEON i8mm. See spec/10-cpu.md.
2//!
3//! One of the six crates where `unsafe` is allowed. Every block carries a `// SAFETY:` comment
4//! that names the invariant which makes it sound.
5
6pub mod attention;
7pub mod compat;
8pub mod gemm;
9pub mod ops;
10pub mod par;
11pub mod plan;
12pub mod pool;
13pub mod qgemm;
14#[cfg(test)]
15mod testing;
16
17pub use compat::{Compat, Input, Output, executor, executor_from, executor_with};
18pub use plan::{CpuBackend, CpuPlan};