tilelang.language.parallel module

The language interface for tl programs.

tilelang.language.parallel.Parallel(*extents: PrimExpr, coalesced_width: Optional[int] = 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