tvm_tirx.tirx.script.launch_thread

Contents

tvm_tirx.tirx.script.launch_thread#

tvm_tirx.tirx.script.launch_thread(thread_tag, extent, *, attrs=None, body_params=None)#

Launch a hardware or virtual thread with a fresh lexical variable.

The extent determines the variable’s scalar integer type. Tags starting with vthread denote virtual threads. This frame works with any TIR builder parent, including standalone statement construction.

Examples

with T.launch_thread("threadIdx.x", 32) as tx:
    T.evaluate(tx)
Parameters:
  • thread_tag (str)

  • extent (Expr)

Return type:

RegionFrame