tvm_tirx.backend.cuda.tile.exec_context.AxisRange

class tvm_tirx.backend.cuda.tile.exec_context.AxisRange(extent, offset=0, stride=1)

Bases: object

An active slice offset + stride * [0, extent) on one TileLayout axis.

Parameters:

• extent (int)

• offset (int)

• stride (int)

extent: int

offset: int = 0

stride: int = 1

intersect(lo, hi)

Parameters:

• lo (int)

• hi (int)

Return type:

AxisRange

modulo(modulus, residue)

Parameters:

• modulus (int)

• residue (int)

Return type:

AxisRange