infotheory/backends/rwkvzip/rwkv7/
mod.rs

1//! High-performance RWKV7 inference kernel.
2//!
3//! This module provides a portable SIMD RWKV7 implementation using `wide` so
4//! rustc/LLVM can pick the best ISA per target (x86_64, aarch64, wasm32 SIMD,
5//! or scalar fallback on non-SIMD CPUs).
6//!
7//! # Architecture
8//!
9//! - Matrix/vector operations are vectorized via `wide` (`f32x8`)
10//! - State updates are optimized for RWKV7 head dimension N=64
11//! - Memory layout is cache-friendly and alignment-aware
12//! - No external BLAS dependencies
13
14mod kernel;
15mod model;
16mod profiling;
17mod tensor;
18mod weights;
19
20pub use model::ScratchBuffers;
21pub use model::{Config, FullAdamState, Model, State, TrainScopeMask};
22pub use profiling::{LayerProfiler, LayerTiming, NullProfiler, ProfilerSink};
23pub use tensor::{Tensor1D, Tensor2D, TensorView1D, TensorView2D};
24pub use weights::Weights;