Mojo struct
OutputPipelineStorage
struct OutputPipelineStorage[num_stages: Int]
Unified storage for output/accumulator pipeline.
For MMA → Epilogue synchronization. TMEM stages are allocated dynamically, so this only stores barriers.
Parameters
- num_stages (
Int): Number of accumulator pipeline stages.
Fields
- barriers (
BarrierPair[num_stages]):
Implemented traits
AnyType,
ImplicitlyDestructible
comptime members
__del__is_trivial
comptime __del__is_trivial = True
BarrierArray
comptime BarrierArray = SMemArray[SharedMemBarrier, (num_stages * 2)]
Methods
create_pipeline
create_pipeline(ref[AddressSpace._value._mlir_value] self) -> ProducerConsumerPipeline[num_stages]
Create runtime pipeline from this storage.
Returns:
barrier_ptr
barrier_ptr(ref[AddressSpace._value._mlir_value] self) -> MbarPtr
Escape hatch: Get raw barrier pointer.
Returns:
MbarPtr
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!