tilelang.language.parallel¶

The language interface for tl programs.

Functions¶

Parallel(*extents[, coalesced_width])

Tools to construct nested parallel for loop.

Module Contents¶

tilelang.language.parallel.Parallel(*extents, coalesced_width=None)¶
Tools to construct nested parallel for loop.

This can be used to create element-wise tensor expression.

Parameters:
  • extents (PrimExpr) – The extents of the iteration.

  • coalesced_width (Optional[int]) – The coalesced width of the parallel loop.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame