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

Enumerates the role each warp plays in the warp-specialized kernel.

The main-load, MMA, and epilogue warps select their code path by comparing the runtime warp id against these roles.

Implemented traits

AnyType, Copyable, Deinitable, ImplicitlyCopyable, Movable, RegisterPassable, TrivialRegisterPassable

comptime members

Epilogue

comptime Epilogue = WarpRole(Int32(3))

MainLoad

comptime MainLoad = WarpRole(Int32(4))

Mma

comptime Mma = WarpRole(Int32(5))

Methods

__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_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