tvm_tirx.backend.cuda.iket.IketProfiler

class tvm_tirx.backend.cuda.iket.IketProfiler(*args, **kwargs)

Bases: object

TIRx annotations compiled for NVIDIA’s official IKET runtime.

mark(name, payload=None)

Parameters:

name (str)

range_start(name, payload=None)

Parameters:

name (str)

range_end(token, payload=None)

Parameters:

token (Expr)

range_push(name, payload=None)

Parameters:

name (str)

range_pop()

sentinel_token(name)

Parameters:

name (str)

compile(mod, target=None, *, tir_pipeline='tirx')

Compile official IKET metadata and NativeDump placeholders.