tilelang.tools.cuda.iket.session¶

IKET compilation-session and CUDA callback lifecycle.

Functions¶

`enable_runtime_payloads`()	Enable runtime payload metadata and records for subsequent compiles.
`disable_runtime_payloads`()	Disable payload records while keeping event instrumentation enabled.
`runtime_payloads_enabled`()	Return whether runtime payload records are enabled.
`enable`(*[, override])	Enable IKET CUDA source post-processing for subsequent compiles.
`disable`(*[, restore])	Release one IKET enable scope and restore the prior callback at depth zero.
`is_enabled`()	Return whether an IKET callback scope is active.
`session`(*[, reset_events, override, disable_on_exit, ...])	Configure IKET around kernel construction and compilation.

Module Contents¶

tilelang.tools.cuda.iket.session.enable_runtime_payloads()¶

Enable runtime payload metadata and records for subsequent compiles.

Return type:: None

tilelang.tools.cuda.iket.session.disable_runtime_payloads()¶

Disable payload records while keeping event instrumentation enabled.

Return type:: None

tilelang.tools.cuda.iket.session.runtime_payloads_enabled()¶

Return whether runtime payload records are enabled.

Return type:: bool

tilelang.tools.cuda.iket.session.enable(*, override=True)¶

Enable IKET CUDA source post-processing for subsequent compiles.

Parameters:: override (bool)
Return type:: None

tilelang.tools.cuda.iket.session.disable(*, restore=True)¶

Release one IKET enable scope and restore the prior callback at depth zero.

Parameters:: restore (bool)
Return type:: None

tilelang.tools.cuda.iket.session.is_enabled()¶

Return whether an IKET callback scope is active.

Return type:: bool

tilelang.tools.cuda.iket.session.session(*, reset_events=True, override=True, disable_on_exit=True, output_dir=None, runtime_payloads=None, disable_cache=True)¶

Configure IKET around kernel construction and compilation.

Parameters:

reset_events (bool)
override (bool)
disable_on_exit (bool)
output_dir (str | os.PathLike[str] | None)
runtime_payloads (bool | None)
disable_cache (bool)

Return type:

_Session