tilelang.tools.cuda.iket.session ================================ .. py:module:: tilelang.tools.cuda.iket.session .. autoapi-nested-parse:: IKET compilation-session and CUDA callback lifecycle. Functions --------- .. autoapisummary:: tilelang.tools.cuda.iket.session.enable_runtime_payloads tilelang.tools.cuda.iket.session.disable_runtime_payloads tilelang.tools.cuda.iket.session.runtime_payloads_enabled tilelang.tools.cuda.iket.session.enable tilelang.tools.cuda.iket.session.disable tilelang.tools.cuda.iket.session.is_enabled tilelang.tools.cuda.iket.session.session Module Contents --------------- .. py:function:: enable_runtime_payloads() Enable runtime payload metadata and records for subsequent compiles. .. py:function:: disable_runtime_payloads() Disable payload records while keeping event instrumentation enabled. .. py:function:: runtime_payloads_enabled() Return whether runtime payload records are enabled. .. py:function:: enable(*, override = True) Enable IKET CUDA source post-processing for subsequent compiles. .. py:function:: disable(*, restore = True) Release one IKET enable scope and restore the prior callback at depth zero. .. py:function:: is_enabled() Return whether an IKET callback scope is active. .. py:function:: 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.