Crate cubecl

Re-exports

• pub use cubecl_linalg as linalg;

Modules

• benchmark
  Module for benchmarking any executable part
• channel
  Compute channel module.
• client
  Compute client module.
• codegen
• compute
• frontend
  Cube Frontend Types.
• future
  Future utils with a compatible API for native, non-std and wasm environments.
• ir
  Cube Language Internal Representation.
• prelude
• server
  Compute server module.
• tune
  Autotune module

Macros

• comptime
  Mark the contents of this macro as compile time values, turning off all expansion for this code and using it verbatim
• cpa
  Cube Pseudo Assembly.
• unexpanded

Structs

• CubeDim
• Execution
• InplaceMapping
  Simply indicate the output that can be replaced by the input.
• KernelExpansion
  The information necessary to compile a kernel definition.
• KernelId
  Kernel unique identifier.
• KernelIntegrator
  The kernel integrator allows you to create a kernel definition based on kernel expansion and kernel settings.
• KernelSettings

Enums

• CubeCount
  Specifieds the number of cubes to be dispatched for a kernel.
• CubeCountSettings
  The position of the input or output to calculate the number of cubes to launch.
• ExecutionMode
  The kind of execution to be performed.
• Feature
  Every feature that can be supported by a cube runtime.
• InputInfo
  Information related to an input.
• MemoryConfiguration
  High level configuration of memory management.
• OutputInfo
  Information related to an output.

Constants

• SUBCUBE_DIM_APPROX
  An approximation of the subcube dimension.

Traits

• Compiler
  Compiles the representation into its own representation that can be formatted into tokens.
• CompilerRepresentation
  Trait for compiled code representation
• CubeElement
  The base element trait for the jit backend.
• Kernel
  Implement this trait to create a kernel definition.
• Runtime
  Runtime for the CubeCL.

Functions

• bool_elem
• calculate_cube_count_elemwise
  Calculate the number of cubes required to execute an operation where one cube unit is assigned to one element.
• calculate_num_elems_dyn_rank
• tensor_line_size
• tensor_vectorization_factor

Type Aliases

• RuntimeArg
  Runtime arguments to launch a kernel.

Attribute Macros

• cube
  Mark a cube function, trait or implementation for expansion.
• tune
  Crates a tuning set with a specific signature. Should return a tuple of benchmark inputs.

Derive Macros

• 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