Mojo struct
TileConsumer
@register_passable(trivial)
struct TileConsumer[origin: MutOrigin, Payload: TilePayload, num_group_stages: Int, k_group_size: Int]
Consumer view for MMA warp. Use acquire() to get stages.
Fields
- pipeline_ptr (
Pointer[TileConsumer[origin, Payload, num_group_stages, k_group_size].PipelineType, origin]):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable
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
PipelineType
comptime PipelineType = InputTilePipeline[Payload, num_group_stages, k_group_size]
Methods
__enter__
__enter__(mut self) -> Self
__exit__
__exit__(mut self)
acquire
acquire(mut self) -> InputConsumerStage[origin, Payload, num_group_stages, k_group_size]
Acquire next stage, waiting for tiles to be ready.
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!