cubecl_core/
prelude.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
pub use crate::{cube, CubeLaunch, CubeType, Kernel, RuntimeArg};

pub use crate::codegen::{KernelExpansion, KernelIntegrator, KernelSettings};
pub use crate::compute::{CompiledKernel, CubeTask, KernelBuilder, KernelLauncher, KernelTask};
pub use crate::frontend::cmma;
pub use crate::frontend::{branch::*, synchronization::*, vectorization_of};
pub use crate::ir::{CubeDim, KernelDefinition};
pub use crate::runtime::Runtime;

/// Elements
pub use crate::frontend::{
    Array, ArrayHandleRef, AtomicI32, AtomicI64, AtomicU32, Float, LaunchArg, Slice, SliceMut,
    Tensor, TensorArg,
};
pub use crate::pod::CubeElement;

/// Topology
pub use crate::frontend::{
    ABSOLUTE_POS, ABSOLUTE_POS_X, ABSOLUTE_POS_Y, ABSOLUTE_POS_Z, CUBE_COUNT, CUBE_COUNT_X,
    CUBE_COUNT_Y, CUBE_COUNT_Z, CUBE_DIM, CUBE_DIM_X, CUBE_DIM_Y, CUBE_DIM_Z, CUBE_POS, CUBE_POS_X,
    CUBE_POS_Y, CUBE_POS_Z, SUBCUBE_DIM, UNIT_POS, UNIT_POS_X, UNIT_POS_Y, UNIT_POS_Z,
};

/// Export subcube operations.
pub use crate::frontend::{subcube_all, subcube_max, subcube_min, subcube_prod, subcube_sum};
pub use cubecl_runtime::client::ComputeClient;
pub use cubecl_runtime::server::CubeCount;

pub use crate::comptime;
pub use crate::frontend::*;