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
UMMADescriptorSS
struct UMMADescriptorSS[operand_type: DType]
Holds two shared-memory descriptors for the A and B operands of a shared-shared (SS) UMMA.
Parameters
- operand_type (
DType): The element type of the A and B operands.
Fields
- a (
UMMADescriptorSS[operand_type].a_t): - b (
UMMADescriptorSS[operand_type].b_t):
Implemented traits
AnyType,
Copyable,
Deinitable,
DescriptorPair,
ImplicitlyCopyable,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
a_t
comptime a_t = MMASmemDescriptor
b_t
comptime b_t = MMASmemDescriptor
operand_t
comptime operand_t = operand_type
Methods
__init__
def __init__(a: MMASmemDescriptor, b: MMASmemDescriptor) -> Self
get_a
def get_a(self) -> Self.a_t
Returns:
Self.a_t
get_b
def get_b(self) -> Self.b_t
Returns:
Self.b_t