cubecl-core 0.7.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
24
25
26
27

pub use crate::{CubeLaunch, CubeType, RuntimeArg, cube};

pub use crate::codegen::{KernelExpansion, KernelIntegrator, KernelSettings};
pub use crate::compute::{
    CompiledKernel, CubeKernel, KernelBuilder, KernelDefinition, KernelLauncher, KernelTask,
};
pub use crate::frontend::cmma;
/// Elements
pub use crate::frontend::{
    Array, ArrayHandleRef, Atomic, Float, FloatExpand, LaunchArg, NumericExpand, Slice, SliceMut,
    Tensor, TensorArg,
};
pub use crate::frontend::{branch::*, synchronization::*};
pub use crate::pod::CubeElement;
pub use crate::runtime::Runtime;

/// 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::id::KernelId;
pub use cubecl_runtime::kernel::KernelMetadata;
pub use cubecl_runtime::server::CubeCount;

pub use crate::frontend::*;
pub use crate::{comment, comptime, comptime_type, derive_cube_comptime, terminate};
pub use cubecl_common::{CubeDim, ExecutionMode, flex32, tf32};
pub use cubecl_ir::Scope;