For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
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
BarrierArray
comptime BarrierArray = SMemArray[SharedMemBarrier, (num_stages * 2)]
Methods
create_pipeline
create_pipeline(ref[AddressSpace._value] self) -> ProducerConsumerPipeline[num_stages]
Create runtime pipeline from this storage.
Returns:
ProducerConsumerPipeline[num_stages]
barrier_ptr
barrier_ptr(ref[AddressSpace._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!