Mojo struct
UMMADescriptorTS
@register_passable(trivial)
struct UMMADescriptorTS[operand_type: DType, num_m_mmas: Int, num_n_mmas: Int, *, MMA_M: Int, MMA_N: Int, MMA_K: Int, consumer_group_size: Int]
Fields
- a (
TMemOperand[operand_type, num_m_mmas, num_n_mmas, MMA_M, MMA_N, MMA_K, consumer_group_size]): - b (
MMASmemDescriptor):
Implemented traits
AnyType,
Copyable,
DescriptorPairTS,
Movable,
UnknownDestructibility
Aliases
a_t
alias a_t = TMemOperand[operand_type, num_m_mmas, num_n_mmas, MMA_M, MMA_N, MMA_K, consumer_group_size]
b_t
alias b_t = MMASmemDescriptor
operand_t
alias operand_t = operand_type
Methods
__init__
__init__(a: TMemOperand[operand_type, num_m_mmas, num_n_mmas, MMA_M, MMA_N, MMA_K, consumer_group_size], b: MMASmemDescriptor) -> Self
get_a
get_a(self) -> TMemOperand[operand_type, num_m_mmas, num_n_mmas, MMA_M, MMA_N, MMA_K, consumer_group_size]
get_b
get_b(self) -> MMASmemDescriptor
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!