rlx_cpu/lib.rs
1// RLX — versatile ML compiler + runtime.
2// Copyright (C) 2026 Eugene Hauptmann, Nataliya Kosmyna.
3//
4// This program is free software: you can redistribute it and/or modify
5// it under the terms of the GNU General Public License as published by
6// the Free Software Foundation, version 3.
7//
8// This program is distributed in the hope that it will be useful,
9// but WITHOUT ANY WARRANTY; without even the implied warranty of
10// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11// GNU General Public License for more details.
12//
13// You should have received a copy of the GNU General Public License
14// along with this program. If not, see <https://www.gnu.org/licenses/>.
15
16//! RLX CPU backend — executes optimized IR graphs on CPU.
17//!
18//! Takes a fused + memory-planned IR graph and executes it using:
19//! - BLAS (Accelerate/MKL/OpenBLAS) for matmul
20//! - NEON/AVX SIMD kernels for element-wise ops
21//! - Persistent Rayon thread pool for parallelism
22//! - Arena allocator for zero per-call allocation
23
24pub mod arena;
25pub mod asm_check;
26pub mod attention_bwd;
27pub mod autotune;
28pub mod blas;
29pub mod calibrate;
30pub mod config;
31pub mod cost;
32pub mod dequant_cache;
33pub mod dispatch;
34pub mod executor;
35pub mod gdn;
36pub mod gguf_matmul;
37pub mod intrinsics;
38pub mod kernel_config;
39pub mod kernels;
40pub mod llada2_gate;
41pub mod lm_head;
42pub mod moe_residency;
43pub mod moe_topk_capture;
44pub mod naive;
45pub mod op_registry;
46pub mod pool;
47pub mod splat;
48pub mod thunk;
49pub mod tile;
50pub mod training_bwd;