tvm_tirx.backend.cuda.tile.exec_context.ExecContext

tvm_tirx.backend.cuda.tile.exec_context.ExecContext#

class tvm_tirx.backend.cuda.tile.exec_context.ExecContext(A, scope_kind, inter, intra)#

Bases: object

Per-program-point compiler state: active set + scope kind + split.

Parameters:
A: ActiveSet#
scope_kind: str#
inter: dict[str, AxisRange]#
intra: dict[str, AxisRange]#
static at_kernel_entry(*, lane_ext=32, warp_ext, cta_ext=1)#
Parameters:

  • lane_ext (int)

  • warp_ext (int)

  • cta_ext (int)

Return type:

ExecContext

with_filter(binding, lo, hi)#
Parameters:
Return type:

ExecContext

with_cta_axis_modulo(axis, modulus, residue)#
Parameters:

  • axis (str)

  • modulus (int)

  • residue (int)

Return type:

ExecContext

with_scope_switch(scope_kind)#
Parameters:

scope_kind (str)

Return type:

ExecContext