avarok_core/
lib.rs

1// SPDX-License-Identifier: AGPL-3.0-only
2
3#![deny(warnings)]
4#![deny(clippy::all)]
5
6pub mod arch;
7pub mod capabilities;
8pub mod compute;
9pub mod config;
10pub mod dtype;
11pub mod env_compat;
12pub mod error;
13pub mod fault;
14pub mod kimi_k3;
15pub mod mxfp4_e8m0;
16pub mod numeric;
17pub mod safetensors;
18pub mod scope;
19pub mod target;
20pub mod tensor;
21
22// `device` always compiles — its `sm121` submodule is pure constants
23// that spark-model's launch heuristics consume on every backend.
24// The `AvarokDevice` cudarc wrapper inside `device` is itself gated
25// behind the `cuda` feature.
26pub mod device;
27
28// CUDA-only modules: rely on `cudarc` and the NVIDIA driver. Gated so the
29// crate compiles on hosts without a CUDA toolchain (e.g. Apple Silicon).
30#[cfg(feature = "cuda")]
31pub mod cuda_host;
32#[cfg(feature = "cuda")]
33pub mod kernel;
34#[cfg(feature = "cuda")]
35pub mod registry;
36#[cfg(feature = "cuda")]
37pub mod stream;
avarok_core/lib.rs

avarok_core/
lib.rs
