cubecl_runtime/
lib.rs

1#![cfg_attr(not(feature = "std"), no_std)]
2#![warn(missing_docs)]
3
4//! CubeCL runtime crate that helps creating high performance async runtimes.
5
6extern crate alloc;
7
8#[macro_use]
9extern crate derive_new;
10
11/// Various identifier types used in CubeCL.
12pub mod id;
13
14/// Kernel related traits.
15pub mod kernel;
16
17/// Stream related utilities.
18#[cfg(feature = "std")]
19pub mod stream;
20
21/// Compute client module.
22pub mod client;
23
24/// Autotune module
25pub mod tune;
26
27/// Memory management module.
28pub mod memory_management;
29/// Compute server module.
30pub mod server;
31/// Compute Storage module.
32pub mod storage;
33
34/// CubeCL config module.
35pub mod config;
36
37mod feature_set;
38/// Runtime features and associated types
39pub mod features;
40
41pub use cubecl_common::benchmark;
42
43pub use feature_set::*;
44pub use features::*;
45/// Logging utilities to be used by a compute server.
46pub mod logging;
47
48/// TMA-related runtime types
49pub mod tma;
50
51/// Simple system profiling using timestamps.
52pub mod timestamp_profiler;