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 conv_bwd;
32pub mod conv_fwd;
33pub mod cost;
34pub mod dequant_cache;
35pub mod dispatch;
36pub mod executor;
37pub mod gdn;
38pub mod gguf_matmul;
39pub mod im2col;
40pub mod intrinsics;
41pub mod kernel_config;
42pub mod kernels;
43pub mod llada2_gate;
44pub mod lm_head;
45pub mod moe_residency;
46pub mod moe_topk_capture;
47pub mod naive;
48pub mod onnx_ref;
49pub mod op_registry;
50pub mod pool;
51pub mod splat;
52pub mod thunk;
53pub mod tile;
54pub mod training_bwd;
55pub mod umap_knn;