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
Tags each warp with its specialized role in the warp-specialized MLA prefill kernel.
Implemented traits
AnyType,
Copyable,
Deinitable,
Equatable,
ImplicitlyCopyable,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
Correction
comptime Correction = WarpRole(Int32(2))
CVTToBF16
comptime CVTToBF16 = WarpRole(Int32(5))
Empty
comptime Empty = WarpRole(Int32(6))
Load
comptime Load = WarpRole(Int32(4))
MMA
comptime MMA = WarpRole(Int32(3))
Softmax0
comptime Softmax0 = WarpRole(Int32(0))
Softmax1
comptime Softmax1 = WarpRole(Int32(1))