tvm_tirx.backend.cuda.script.iket

Contents

tvm_tirx.backend.cuda.script.iket#

CUDA TVMScript namespaces.

Functions#

mark([payload])

Create an NVIDIA IKET marker annotation.

range_start([payload])

Create an NVIDIA IKET token-range start annotation.

range_end([payload])

Create an NVIDIA IKET token-range end annotation.

range_push([payload])

Create an NVIDIA IKET stack-range push annotation.

range_pop(*[, attrs, ty_args, loc, ty])

Construct a call to tirx.cuda.iket_range_pop.

sentinel_token(*[, attrs, ty_args, loc, ty])

Construct a call to tirx.cuda.iket_sentinel_token.

official_event([source_code, payload])

Create an NVIDIA IKET official range-end event.
Contents