cubecl/
lib.rs

1pub use cubecl_core::*;
2
3pub use cubecl_runtime::config;
4pub use cubecl_runtime::features;
5pub use cubecl_runtime::memory_management::MemoryAllocationMode;
6
7#[cfg(feature = "wgpu")]
8pub use cubecl_wgpu as wgpu;
9
10#[cfg(feature = "cuda")]
11pub use cubecl_cuda as cuda;
12
13#[cfg(feature = "hip")]
14pub use cubecl_hip as hip;
15
16#[cfg(feature = "attention")]
17pub use cubecl_attention as attention;
18
19#[cfg(feature = "matmul")]
20pub use cubecl_matmul as matmul;
21
22#[cfg(feature = "convolution")]
23pub use cubecl_convolution as convolution;
24
25#[cfg(feature = "stdlib")]
26pub use cubecl_std as std;
27
28#[cfg(feature = "reduce")]
29pub use cubecl_reduce as reduce;
30
31#[cfg(feature = "random")]
32pub use cubecl_random as random;
33
34#[cfg(feature = "cpu")]
35pub use cubecl_cpu as cpu;