tilelang.language.persistent

The language interface for tl programs.

Functions

Persistent(domain, wave_size, index[, group_size])

Tools to construct persistent for loop.

Module Contents

tilelang.language.persistent.Persistent(domain, wave_size, index, group_size=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.