tvm_tirx.backend.cuda.tile.exec_context.initial_A

Contents

tvm_tirx.backend.cuda.tile.exec_context.initial_A#

tvm_tirx.backend.cuda.tile.exec_context.initial_A(*, lane_ext=32, warp_ext, cta_ext=1)#

Build A at Function device entry: all threads active, offsets all zero.

Parameters:

  • lane_ext (int)

  • warp_ext (int)

  • cta_ext (int)

Return type:

ActiveSet

Contents