Skip to main content

Mojo trait

AsyncTensorAccumulatorTS

Implemented traits

AnyType, UnknownDestructibility

Aliases

a_t

alias a_t

ab_t

alias ab_t

accum_t

alias accum_t

b_t

alias b_t

c_t

alias c_t

operand_t

alias operand_t

Methods

__init__

__init__(smem: UnsafePointer[SharedMemBarrier, address_space=AddressSpace(3), alignment=8]) -> _Self

mma

mma(self: _Self, a: get_vtable_entry(:trait<_nn::_mha_sm100::_AsyncTensorAccumulatorTS> _Self, "a_t"), b: get_vtable_entry(:trait<_nn::_mha_sm100::_AsyncTensorAccumulatorTS> _Self, "b_t"), c: get_vtable_entry(:trait<_nn::_mha_sm100::_AsyncTensorAccumulatorTS> _Self, "c_t"), c_scale: SIMD[uint32, 1], wg_idx: SIMD[uint32, 1] = __init__[__mlir_type.!pop.int_literal](0))

wait_group

wait_group[wgmma_left_in_flight: Int = 0](mut self: _Self)

Was this page helpful?