tvm_tirx.backend.cuda.tile.exec_context.ActiveSet

Contents

tvm_tirx.backend.cuda.tile.exec_context.ActiveSet#

class tvm_tirx.backend.cuda.tile.exec_context.ActiveSet(layout)#

Bases: object

Active thread set represented by a TileLayout.

Parameters:: layout (TileLayout)

layout: TileLayout#

static from_axes(axes)#

Parameters:: axes (list[tuple[str, AxisRange]])
Return type:: ActiveSet

property size: int#

property axis_names: list[str]#

axis(name)#

Parameters:: name (str)
Return type:: AxisRange

replace_axis(axis, axis_range)#

Parameters:

axis (str)
axis_range (AxisRange)

Return type:

property laneid: AxisRange#

property warpid: AxisRange#

property cta_id: AxisRange#