tilelang.language.parallel¶
The language interface for tl programs.
Functions¶
|
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