cubecl-core 0.4.0

CubeCL core create
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

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::*};
pub use crate::ir::{CubeDim, KernelDefinition};
pub use crate::runtime::Runtime;

/// Elements
pub use crate::frontend::{
    Array, ArrayHandleRef, Atomic, Float, FloatExpand, LaunchArg, NumericExpand, Slice, SliceMut,
    Tensor, TensorArg,
};
pub use crate::pod::CubeElement;

/// Export plane operations.
pub use crate::frontend::{plane_all, plane_max, plane_min, plane_prod, plane_sum};
pub use cubecl_runtime::client::ComputeClient;
pub use cubecl_runtime::server::CubeCount;

pub use crate::frontend::*;
pub use crate::{comment, comptime};