tvm_tirx.backend.cuda.iket.run

Contents

tvm_tirx.backend.cuda.iket.run#

tvm_tirx.backend.cuda.iket.run(main, *, output_dir, postprocess='all', clobber=False, cwd=None, env=None, max_ts_cnt_per_warp=None, keep=False, timeout=600.0)#

Replay the active script under run-iket and execute main in its passes.

Module-level code executes once in the parent, tracker, and capture processes. main executes only in tracker and capture. Code following run executes only in the parent after a successful profile.

Parameters:

  • output_dir (str | PathLike[str])

  • postprocess (str)

  • clobber (bool)

  • cwd (str | PathLike[str] | None)

  • env (Mapping[str, str | PathLike[str]] | None)

  • max_ts_cnt_per_warp (int | None)

  • keep (bool)

  • timeout (float | None)

Return type:

IketProfileResult

Contents