tilelang.language.persistent module

The language interface for tl programs.

tilelang.language.persistent.Persistent(domain: List[PrimExpr], wave_size: PrimExpr, index: PrimExpr, group_size: Optional[PrimExpr] = 8)

Tools to construct persistent for loop.

Parameters:

• domain (List[tir.PrimExpr]) – The list of dominators.

• wave_size (int) – The wave size.

• index (int) – The tile index in one wave.

• group_size (tir.PrimExpr) – The group size.