Mojo struct
EpilogueConfig
struct EpilogueConfig
Computed epilogue parameters based on MMA and CTA configuration.
Bundles the 7 input parameters shared by all epilogue component structs (TMAStoreCoords, TMAStoreExecutor, TMEMToSMemWriter, SMemEpilogueWriter) plus 3 derived fields (is_lower_frag_required, num_stages, fragment_size).
Constructed once per TileWriter/BlockwiseFP8TileWriter and propagated to all epilogue component types.
Fields
- MMA_M (
Int): - MMA_N (
Int): - stageN (
Int): - cta_group (
Int): - transpose_c (
Bool): - BM (
Int): - BN (
Int): - is_lower_frag_required (
Bool): - num_stages (
Int): - fragment_size (
Int):
Implemented traits
AnyType,
Copyable,
Equatable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable
Methods
create
static create(*, MMA_M: Int, MMA_N: Int, stageN: Int, cta_group: Int, transpose_c: Bool, BM: Int, BN: Int) -> Self
Construct EpilogueConfig with derived fields computed automatically.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!