Module memory

cubecl_matmul::components::global

Module memory 

Source

Modules§

load_window_in_stage
load_window_in_tile

Structs§

BatchedGlobalLayout
Global layout that uses the last two dimensions and ignores all others.
BatchedGlobalLayoutCompilationArg
BatchedGlobalLayoutExpand
BatchedGlobalLayoutLaunch
BlockScaledLayout
Workaround for enums not supporting comptime, should fix that in the future
BlockScaledLayoutCompilationArg
BlockScaledLayoutExpand
BlockScaledLayoutLaunch
GlobalIterator
An iterator over global memory, advancing along k.
GlobalLayoutConfig
GlobalMemoryConfig
SimpleTmaGlobalLayout
Global layout that uses the last two dimensions and ignores all others.
SimpleTmaGlobalLayoutCompilationArg
SimpleTmaGlobalLayoutExpand
SimpleTmaGlobalLayoutLaunch

Enums§

BatchedGlobalScaleLayout
BatchedGlobalScaleLayoutArgs
BatchedGlobalScaleLayoutCompilationArg
BatchedGlobalScaleLayoutExpand
ViewDirection

Functions§

load_window_in_stage
Reads data from the tensor view as a window, i.e. a slice of global memory
load_window_in_tile
Reads data from the tensor view as a window, i.e. a slice of global memory Also returns the length of the slice