tilelang.language.pipeline¶

The language interface for tl programs.

Functions¶

Pipelined(start[, stop, num_stages, order, stage, ...])

Tools to construct pipelined for loop.

Module Contents¶

tilelang.language.pipeline.Pipelined(start, stop=None, num_stages=0, order=None, stage=None, sync=None, group=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.

  • order (Optional[List[int]])

  • stage (Optional[List[int]])

  • sync (Optional[List[List[int]]])

  • group (Optional[List[List[int]]])

Returns:

res – The ForFrame.

Return type:

frame.ForFrame