atlas_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 error;
12pub mod fault;
13pub mod kimi_k3;
14pub mod mxfp4_e8m0;
15pub mod numeric;
16pub mod safetensors;
17pub mod scope;
18pub mod target;
19pub mod tensor;
20
21// `device` always compiles — its `sm121` submodule is pure constants
22// that spark-model's launch heuristics consume on every backend.
23// The `AtlasDevice` cudarc wrapper inside `device` is itself gated
24// behind the `cuda` feature.
25pub mod device;
26
27// CUDA-only modules: rely on `cudarc` and the NVIDIA driver. Gated so the
28// crate compiles on hosts without a CUDA toolchain (e.g. Apple Silicon).
29#[cfg(feature = "cuda")]
30pub mod cuda_host;
31#[cfg(feature = "cuda")]
32pub mod kernel;
33#[cfg(feature = "cuda")]
34pub mod registry;
35#[cfg(feature = "cuda")]
36pub mod stream;