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

WarpRole

struct WarpRole

Encodes the warp role assignments used by the warp-specialized kernel.

Each warp in the block is assigned one role (MainLoad, SfbLoad, SfbReady, Mma, Scheduler, Epilogue) and the is_* helpers test the current warp's role against get_warp_id().

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDeletable, Movable, RegisterPassable, TrivialRegisterPassable

`comptime` members

`Epilogue`

comptime Epilogue = WarpRole(Int32(3))

`MainLoad`

comptime MainLoad = WarpRole(Int32(5))

`Mma`

comptime Mma = WarpRole(Int32(6))

`Scheduler`

comptime Scheduler = WarpRole(Int32(4))

`SfbLoad`

comptime SfbLoad = WarpRole(Int32(7))

`SfbReady`

comptime SfbReady = WarpRole(Int32(8))

Methods

`le`

def __le__(self, other: Int) -> Bool

Returns:

Bool

`eq`

def __eq__(self, other: Int) -> Bool

Returns:

Bool

def __eq__(self, other: Self) -> Bool

Returns:

Bool

`ne`

def __ne__(self, other: Self) -> Bool

Returns:

Bool

`ge`

def __ge__(self, other: Int) -> Bool

Returns:

Bool

`is_sfb_ready`

static def is_sfb_ready() -> Bool

Returns:

Bool

`is_sfb_load`

static def is_sfb_load() -> Bool

Returns:

Bool

`is_main_load`

static def is_main_load() -> Bool

Returns:

Bool

`is_mma`

static def is_mma() -> Bool

Returns:

Bool

`is_epilogue`

static def is_epilogue() -> Bool

Returns:

Bool

`is_scheduler`

static def is_scheduler() -> Bool

Returns:

Bool

Implemented traits​

comptime members​

Epilogue​

MainLoad​

Mma​

Scheduler​

SfbLoad​

SfbReady​

Methods​

__le__​

__eq__​

__ne__​

__ge__​

is_sfb_ready​

is_sfb_load​

is_main_load​

is_mma​

is_epilogue​

is_scheduler​

Implemented traits

`comptime` members

`Epilogue`

`MainLoad`

`Mma`

`Scheduler`

`SfbLoad`

`SfbReady`

Methods

`le`

`eq`

`ne`

`ge`

`is_sfb_ready`

`is_sfb_load`

`is_main_load`

`is_mma`

`is_epilogue`

`is_scheduler`