Crate cubecl_std

Cubecl standard library.

Modules

div_ceil

size_of

tensor

Structs

ReinterpretSlice

This struct allows to take a slice of Line<S> and reinterpret it as a slice of T. Semantically, this is equivalent to reinterpreting the slice of Line<S> to a slice of T. When indexing, the index is valid in the casted list.

ReinterpretSliceExpand

ReinterpretSliceMut

This struct allows to take a mutable slice of Line<S> and reinterpret it as a mutable slice of T. Semantically, this is equivalent to reinterpreting the slice of Line<S> to a mutable slice of T. When indexing, the index is valid in the casted list.

ReinterpretSliceMutExpand

SymQ8

Represent the quantization of a f32 into an i8 using the symmetric scheme.

Enums

CubeOption

CubeOptionArgs

CubeOptionCompilationArg

CubeOptionExpand

FastDivmod

Create a fast-divmod object if supported, or a regular fallback if not. This precalculates certain values on the host, in exchange for making division and modulo operations on the GPU much faster. Only supports u32 right now to allow for a simpler algorithm. It’s mostly used for indices regardless.

FastDivmodArgs

FastDivmodCompilationArg

FastDivmodExpand

Functions

div_ceil

size_of