Crate cubecl_core

Re-exports

• pub use frontend::cmma;
• pub use prelude::CubeCount;
• pub use prelude::CubeDim;
• pub use codegen::*;

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

• FeatureSet
  The set of features supported by a runtime.
• KernelId
  Kernel unique identifier.

Enums

• ExecutionMode
  The kind of execution to be performed.
• Feature
  Every feature that can be supported by a cube runtime.

Constants

• SUBCUBE_DIM_APPROX
  An approximation of the subcube dimension.

Traits

• CubeElement
  The base element trait for the jit backend.
• Kernel
  Implement this trait to create a kernel definition.
• Runtime
  Runtime for the CubeCL.

Functions

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

Type Aliases

• RuntimeArg
  Runtime arguments to launch a kernel.

Attribute Macros

• cube
  Derive macro for the module.

Derive Macros

• CubeLaunch
• CubeType