Mojo struct
AccumBarrier
@register_passable(trivial)
struct AccumBarrier[cta_group: Int]
Helper for accumulator pipeline barrier operations.
Handles the different arrival patterns for single-CTA vs 2-CTA groups.
Template Parameters: cta_group: Number of CTAs cooperating (1 or 2).
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
Movable,
UnknownDestructibility
comptime members
__copyinit__is_trivial
comptime __copyinit__is_trivial = True
__del__is_trivial
comptime __del__is_trivial = True
__moveinit__is_trivial
comptime __moveinit__is_trivial = True
Methods
arrive
static arrive(pipeline: ProducerConsumerPipeline[num_stages], stage: UInt32)
Signal accumulator arrival.
Args:
- pipeline (
ProducerConsumerPipeline): The MMA output pipeline. - stage (
UInt32): Current pipeline stage.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!