Mojo struct
StandardTileProducer
@register_passable(trivial)
struct StandardTileProducer[origin: MutOrigin, a_type: DType, b_type: DType, a_dim0: Int, a_dim1: Int, b_dim0: Int, b_dim1: Int, num_pipeline_stages: Int, num_group_stages: Int, k_group_size: Int]
Producer view for TMA Load warp (standard tile pipeline).
Fields
- pipeline_ptr (
Pointer[StandardTileProducer[origin, a_type, b_type, a_dim0, a_dim1, b_dim0, b_dim1, num_pipeline_stages, num_group_stages, k_group_size].TilePipelineType, origin]):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable
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
TilePipelineType
comptime TilePipelineType = TilePipeline[a_type, b_type, a_dim0, a_dim1, b_dim0, b_dim1, num_pipeline_stages, num_group_stages, k_group_size]
Methods
__enter__
__enter__(mut self) -> Self
__exit__
__exit__(mut self)
drain
drain(mut self)
Drain pipeline to prevent CTA exit while peer is still working.
acquire
acquire(mut self) -> StandardProducerStage[origin, a_type, b_type, a_dim0, a_dim1, b_dim0, b_dim1, num_pipeline_stages, num_group_stages, k_group_size]
Acquire next stage, waiting for slot availability.
Returns:
StandardProducerStage
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!