atlas_core/
lib.rs

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