kime-cpu 0.0.9

The CPU backend for kime: x86 and ARM kernels, FP32 and INT8.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

//! 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.
//!
//! One of the six crates where `unsafe` is allowed. Every block carries a `// SAFETY:` comment
//! that names the invariant which makes it sound.

pub mod attention;
pub mod compat;
pub mod gemm;
pub mod ops;
pub mod par;
pub mod plan;
pub mod pool;
pub mod qgemm;
#[cfg(test)]
mod testing;

pub use compat::{Compat, Input, Output, executor, executor_from, executor_with};
pub use plan::{CpuBackend, CpuPlan};