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.
rasterization2DRowWithCluster(panel_width, cluster_dim_x)	Row-major swizzle at cluster granularity while preserving intra-cluster x rank.
rasterization2DColumnWithCluster(panel_width, ...)	Column-major swizzle at cluster granularity while preserving intra-cluster x rank.

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

tilelang.contrib.cutedsl.threadblock_swizzle.rasterization2DRowWithCluster(panel_width, cluster_dim_x)

Row-major swizzle at cluster granularity while preserving intra-cluster x rank.

Parameters:

• panel_width (cutlass.Constexpr[int])

• cluster_dim_x (cutlass.Constexpr[int])

Return type:

dim3

tilelang.contrib.cutedsl.threadblock_swizzle.rasterization2DColumnWithCluster(panel_width, cluster_dim_x)

Column-major swizzle at cluster granularity while preserving intra-cluster x rank.

Parameters:

• panel_width (cutlass.Constexpr[int])

• cluster_dim_x (cutlass.Constexpr[int])

Return type:

dim3