Crate cubecl_core Copy item path Source pub use frontend::cmma ;pub use cubecl_ir as ir;pub use codegen ::*;benchmark Module for benchmarking any executable part channel Compute channel module. client Compute client module. codegen compute frontend Cube Frontend Types. future Some future utilities that work across environments.
Future utils with a compatible API for native, non-std and wasm environments. io Input Output utilities. prelude server Compute server module. tune Autotune module comment Insert a literal comment into the kernel source code. comptime Mark the contents of this macro as compile time values, turning off all expansion for this code
and using it verbatim comptime_type Makes the function return a compile time value
Useful in a cube trait to have a part of the trait return comptime values debug_print Print a formatted message using the target’s debug print facilities. The format string is target
specific, but Vulkan and CUDA both use the C++ conventions. WGSL isn’t currently supported. debug_print_expand Print a formatted message using the target’s debug print facilities. The format string is target
specific, but Vulkan and CUDA both use the C++ conventions. WGSL isn’t currently supported. terminate Terminate the execution of the kernel for the current unit. unexpanded CubeDim CubeTuneId ID used to identify a Just-in-Time environment. DeviceId The device id. KernelId Kernel unique identifier. MemoryUsage Amount of memory in use by this allocator
and statistics on how much memory is reserved and
wasted in total. flex32 A floating point type with relaxed precision, minimum f16f32]. tf32 A 19-bit floating point type implementing the tfloat32 AtomicFeature Atomic features that may be supported by a cube runtime . CubeCount Specifieds the number of cubes to be dispatched for a kernel. ExecutionMode The kind of execution to be performed. Feature Every feature that can be supported by a cube runtime . MemoryConfiguration High level configuration of memory management. TmaFeature Atomic features that may be supported by a cube runtime . PLANE_DIM_APPROX Some future utilities that work across environments.
An approximation of the plane dimension. CubeElement The base element trait for the jit backend. Kernel Implement this trait to create a kernel definition . Runtime Runtime for the CubeCL. calculate_cube_count_elemwise Calculate the number of cubes required to execute an operation where one cube unit is
assigned to one element. tensor_line_size tensor_line_size_parallel Find the maximum line size usable for parallel vectorization along the given axis
from the supported line sizes or return 1 if vectorization is impossible. tensor_line_size_perpendicular Find the maximum line size usable for perpendicular vectorization along the given axis
from the supported line sizes or return 1 if vectorization is impossible. tensor_vectorization_factor RuntimeArg Runtime arguments to launch a kernel. cube Mark a cube function, trait or implementation for expansion. derive_cube_comptime Attribute macro to define a type that can be used as a kernel comptime argument
This derive Debug, Hash, PartialEq, Eq, Clone, Copy AutotuneKey Implements display and initialization for autotune keys. CubeLaunch Derive macro to define a cube type that is launched with a kernel CubeType Derive macro to define a cube type that is not launched