For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo struct
WorkInfo
struct WorkInfo
Descriptor for one output tile assigned to a thread block.
Carries the (m, n) tile coordinates in the output matrix, the starting K-tile index for split-K kernels, the number of K tiles to process, and a validity flag indicating whether the tile falls within the problem bounds.
Fields
- m (
UInt32): - n (
UInt32): - k_start (
UInt32): - num_k_tiles (
UInt32): - is_valid_tile (
Bool):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
RegisterPassable,
TrivialRegisterPassable,
Writable
comptime members
INVALID_WORK_INFO
comptime INVALID_WORK_INFO = WorkInfo(UInt32(0), UInt32(0), UInt32(0), UInt32(0), False)
Methods
__init__
def __init__() -> Self
is_valid
is_final_split
get_k_start
write_to
def 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!