For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /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, Deinitable, ImplicitlyCopyable, 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

def is_valid(self) -> Bool

Returns:

Bool

is_final_split

def is_final_split(self, k_tiles_per_output_tile: UInt32) -> Bool

Returns:

Bool

get_k_start

def get_k_start(self) -> UInt32

Returns:

UInt32

write_to

def write_to(self, mut writer: T)