Skip to main content

Crate cubek_std

Crate cubek_std 

Source

Modules§

as_cmma_layout
cube_count
from_cmma_layout
launch
Shared utilities for launching kernels.
layout

Macros§

define_3d_size_base
impl_3d_size_from_tuple

Structs§

FormattedConfigError
Error that arises from invalid configurations
MatmulProblemSize
MatmulProblemSizeExpand

Enums§

InputBinding
MatmulDim
Matrix dimension specifier for matmul operations.
MatrixLayout
Layout of a 2D structure such as a tensor, shared memory or slice, used within any matmul kernel level
MatrixLayoutExpand
StageIdent

Functions§

as_cmma_layout
Maps the matmul MatrixLayout to cmma’s MatrixLayout, for use in Cmma API.
from_cmma_layout
Maps the cmma’s MatrixLayout to matmul MatrixLayout.

Type Aliases§

InvalidConfigError
Error that arises from invalid configurations