tilelang.contrib.cutedsl.threadblock_swizzle

Classes

dim3

Functions

ThreadIdx()
BlockIdx()
GridDim()
rasterization2DRow(panel_width)
rasterization2DColumn(panel_width)

Module Contents

class tilelang.contrib.cutedsl.threadblock_swizzle.dim3

x: int

y: int

z: int

tilelang.contrib.cutedsl.threadblock_swizzle.ThreadIdx()

返回类型:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.BlockIdx()

返回类型:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.GridDim()

返回类型:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.rasterization2DRow(panel_width)

参数:

panel_width (cutlass.cute.typing.Constexpr[int])

返回类型:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.rasterization2DColumn(panel_width)

参数:

panel_width (cutlass.cute.typing.Constexpr[int])

返回类型:

dim3