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