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.
• ir
  Cube Language Internal Representation.
• prelude
• server
  Compute server module.
• tune
  Autotune module

Macros

• cpa
  Cube Pseudo Assembly.
• unexpanded

Structs

• CubeDim
• Execution
• FeatureSet
  The set of features supported by a runtime.
• InplaceMapping
  Simply indicate the output that can be replaced by the input.
• KernelExpansion
  The information necessary to compile a kernel definition.
• KernelIntegrator
  The kernel integrator allows you to create a kernel definition based on kernel expansion and kernel settings.
• KernelSettings

Enums

• CubeCount
  Provides launch information specifying the number of work groups to be used by a compute shader.
• CubeCountSettings
  The position of the input or output to calculate the number of cubes to launch.
• Feature
  Every feature that can be supported by a cube runtime.
• InputInfo
  Information related to an input.
• 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_vectorization_factor

Type Aliases

• RuntimeArg
  Runtime arguments to launch a kernel.

Attribute Macros

• cube
  Derive macro for the module.

Derive Macros

• CubeLaunch
• CubeType