tilelang.language.pipeline module#

The language interface for tl programs.

tilelang.language.pipeline.Pipelined(start: PrimExpr, stop: Optional[PrimExpr] = None, num_stages: int = 0, order: Optional[List[int]] = None, stage: Optional[List[int]] = None, sync: Optional[List[List[int]]] = None, group: Optional[List[List[int]]] = None)#

Tools to construct pipelined for loop.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • num_stages (int) – The max number of buffer used between pipeline producers and consumers. if num_stages is 0, pipeline will not be enabled.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame