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/// Compute channel module.
18pub mod channel;
19/// Compute client module.
20pub mod client;
21
22/// Autotune module
23pub mod tune;
24
25/// Memory management module.
26pub mod memory_management;
27/// Compute server module.
28pub mod server;
29/// Compute Storage module.
30pub mod storage;
31
32/// CubeCL config module.
33pub mod config;
34
35mod feature_set;
36
37mod base;
38pub use base::*;
39pub use cubecl_common::benchmark;
40
41pub use feature_set::*;
42/// Logging utilities to be used by a compute server.
43pub mod logging;
44
45/// TMA-related runtime types
46pub mod tma;
47
48/// Simple system profiling using timestamps.
49pub mod timestamp_profiler;