tilelang.language.warpgroup¶

The language interface for tl programs.

Attributes¶

ws

Classes¶

WarpSpecializeFrame

WarpSpecializeFrame is a custom TIRFrame that manages warp group indices

Functions¶

WarpSpecialize(*warp_group_idx)

Tools to construct a warp group frame.

Module Contents¶

class tilelang.language.warpgroup.WarpSpecializeFrame¶

Bases: tvm.script.ir_builder.tir.frame.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.WarpSpecialize(*warp_group_idx)¶

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 (Tuple[frame.LaunchThreadFrame]) – The result LaunchThreadFrame.

  • Examples – >>> T.ws(0) -> if tx < 128 >>> T.ws(1) -> if tx >= 128 and tx < 256 >>> T.ws(0, 1) -> if tx < 128 or (tx >= 128 and tx < 256)

tilelang.language.warpgroup.ws¶