Skip to main content

Module stage

Module stage 

Source

Modules§

as_swizzle_object

Structs§

StageMemoryConfig

Enums§

SwizzleMode
Swizzling mode of the shared memory. Default None. Matches the base TMA functionality, alternative chunk sizes or more complex patterns don’t really apply to matmul.

Functions§

as_swizzle_object