tilelang.contrib.cutedsl.threadblock_swizzle

Classes

dim3	Three-dimensional CUDA index tuple.

Functions

ThreadIdx()	Return the current CUDA thread index.
BlockIdx()	Return the current CUDA block index.
GridDim()	Return the CUDA grid dimensions.
rasterization2DRow(panel_width)	Map block indices to row-major swizzled rasterization coordinates.
rasterization2DColumn(panel_width)	Map block indices to column-major swizzled rasterization coordinates.

Module Contents

class tilelang.contrib.cutedsl.threadblock_swizzle.dim3

Three-dimensional CUDA index tuple.

x: int

y: int

z: int

tilelang.contrib.cutedsl.threadblock_swizzle.ThreadIdx()

Return the current CUDA thread index.

Return type:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.BlockIdx()

Return the current CUDA block index.

Return type:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.GridDim()

Return the CUDA grid dimensions.

Return type:

dim3

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

Map block indices to row-major swizzled rasterization coordinates.

Parameters:

panel_width (cutlass.Constexpr[int])

Return type:

dim3

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

Map block indices to column-major swizzled rasterization coordinates.

Parameters:

panel_width (cutlass.Constexpr[int])

Return type:

dim3