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

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, Deinitable, ImplicitlyCopyable, 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