Skip to main content

Module cubecl

Module cubecl 

Source
Expand description

Public tenferro-wide CubeCL session (issue #1597).

Exposes a narrow prelude of the CubeCL types needed to write and launch #[cube] kernels against tenferro’s GPU runtime. This module does not re-export the whole of cubecl; downstream crates declare the framework t4a-cubecl package explicitly.

Structs§

CubeDim
The number of units across all 3 axis totalling to the number of working units in a cube.
Session
Public CubeCL extension session.
TensorBinding
Tensor representation with a reference to the server handle, the strides and the shape.

Enums§

ArrayArg
CubeCount
Specifieds the number of cubes to be dispatched for a kernel.