Mojo struct
GroupedWorkInfo
@register_passable(trivial)
struct GroupedWorkInfo
Work info for grouped GEMM with group-specific metadata.
Extends the base WorkInfo with:
- group_idx: Current group index
- k_tile_count: Number of K tiles for this group
- group_changed: True if group changed since last tile (triggers tensormap update)
Fields
- m (
UInt32): M-coordinate of tile within current group. - n (
UInt32): N-coordinate of tile within current group. - k_start (
UInt32): Starting K index (always 0 for grouped GEMM). - is_valid_tile (
Bool): Whether this work tile is valid (not OOB). - group_idx (
UInt32): Current group index. - k_tile_count (
UInt32): Number of K tiles for this group. - group_changed (
Bool): True if group changed since last tile (triggers tensormap update).
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterType,
Stringable,
TrivialRegisterType,
Writable
comptime members
__copyinit__is_trivial
comptime __copyinit__is_trivial = True
__del__is_trivial
comptime __del__is_trivial = True
__moveinit__is_trivial
comptime __moveinit__is_trivial = True
Methods
__init__
__init__() -> Self
Create an invalid/empty work info.
is_valid
coord
__str__
write_to
write_to(self, mut writer: T)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!