tilelang.language.warpgroup module

The language interface for tl programs.

tilelang.language.warpgroup.WarpSpecialize(warp_group_idx: int)

Tools to construct a warp group frame.

Parameters:

warp_group_idx (int) – A integer representing warp group index Or a list of integers representing blockDim.(x|y|z) if the value is -1, we skip the threadIdx.x binding.

Returns:

res – The result LaunchThreadFrame.

Return type:

Tuple[frame.LaunchThreadFrame]

class tilelang.language.warpgroup.WarpSpecializeFrame

Bases: TIRFrame

WarpSpecializeFrame is a custom TIRFrame that manages warp group indices and handles the entry and exit of the kernel launch scope.

tilelang.language.warpgroup.ws(warp_group_idx: int)

Tools to construct a warp group frame.

Parameters:

warp_group_idx (int) – A integer representing warp group index Or a list of integers representing blockDim.(x|y|z) if the value is -1, we skip the threadIdx.x binding.

Returns:

res – The result LaunchThreadFrame.

Return type:

Tuple[frame.LaunchThreadFrame]