For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
SmemPipelineBundleNoClc
struct SmemPipelineBundleNoClc[num_group_pipeline_stages: Int, num_accum_pipeline_stages: Int, Payload: TilePayload]
Composed pipeline storage without CLC scheduler.
Used by kernels with 3-warp specialization (Load, MMA, Epilogue) that don't use a scheduler warp (e.g. Grouped1D1DSmem).
Parameters
- num_group_pipeline_stages (
Int): Number of grouped pipeline stages for input. - num_accum_pipeline_stages (
Int): Number of accumulator pipeline stages. - Payload (
TilePayload): Tile payload type.
Fields
- input_pipeline (
SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].InputPipeline): - output_pipeline (
SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].OutputPipeline): - tmem_dealloc_pipeline (
SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].TmemDeallocPipeline):
Implemented traits
comptime members
AccumBarriers
comptime AccumBarriers = SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].OutputPipeline.BarrierArray
InputBarriers
comptime InputBarriers = SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].InputPipeline.BarrierArray
InputPipeline
comptime InputPipeline = InputPipelineStorage[num_group_pipeline_stages, Payload]
OutputPipeline
comptime OutputPipeline = OutputPipelineStorage[num_accum_pipeline_stages]
TmemAddr
comptime TmemAddr = SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].TmemDeallocPipeline.AddrArray
TmemDealloc
comptime TmemDealloc = SmemPipelineBundleNoClc[num_group_pipeline_stages, num_accum_pipeline_stages, Payload].TmemDeallocPipeline.BarrierArray
TmemDeallocPipeline
comptime TmemDeallocPipeline = TmemDeallocStorage
Methods
input_barriers
def input_barriers(ref[AddressSpace._value] self) -> Self.InputBarriers
Returns input tile pipeline barriers.
Returns:
Self.InputBarriers
accum_barriers
def accum_barriers(ref[AddressSpace._value] self) -> Self.AccumBarriers
Returns accumulator pipeline barriers.
Returns:
Self.AccumBarriers
tmem_dealloc
def tmem_dealloc(ref[AddressSpace._value] self) -> Self.TmemDealloc
Returns TMEM deallocation barrier.
Returns:
Self.TmemDealloc
tmem_addr
def tmem_addr(ref[AddressSpace._value] self) -> Self.TmemAddr
Returns TMEM address storage.
Returns:
Self.TmemAddr