Crate cubek_std

Modules§

FormattedConfigError: Error that arises from invalid configurations
GlobalPartitionSize: Number of global matmul blocks computed by a single cube.
MatmulProblemSize
MatmulProblemSizeExpand
PartitionSize
PartitionSizeExpand
PartitionThreshold: Threshold of plane id at which the roles change.
PartitionThresholdExpand
PlaneFlowConfig: Plane-flow configuration carried by CubeDimResource::Specialized. Holds the counts for main-flow vs load-only planes and the partition rule used at runtime.
PlaneFlowCounts: Represents how many planes are used for main computation and for loading-only tasks.
StageSize
StageSizeExpand
SwizzleModes
TileSize
TileSizeExpand

CubeDimResource: Number of compute primitives required by some component, specified as either units, planes, or a specialized plane-flow split.
InputBinding
InputLoadFlow: Determines which types of planes are responsible for loading a tensor.
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
PlaneFlowPartition: Runtime view of PlaneFlowPartitionRule: distinguishes a plane’s role based on its plane id.
PlaneFlowPartitionExpand
PlaneFlowPartitionRule: How planes are partitioned by id between the main flow and load-only roles.
StageIdent

as_cmma_layout: Maps the matmul MatrixLayout to cmma’s MatrixLayout, for use in Cmma API.
find_instruction_size: The instruction shape to contract with, chosen from the problem’s aspect ratio: 32x8x16 for a tall problem, 8x32x16 for a wide one, 16x16x16 for a balanced one, then 8x8x8, then whatever the device supports at all.
from_cmma_layout: Maps the cmma’s MatrixLayout to matmul MatrixLayout.
partition_coordinates: Returns the (row, col) of the current compute primitive within the stage, deriving compute_index from role_rule_config via PlaneFlowPartition and delegating the per-scope math to Partitioner::coordinates.