tilelang.tools.cuda.iket.codegen¶

CUDA source instrumentation for IKET events.

Functions¶

inject_iket_cuda(code, target, *, runtime_payloads)

Inject IKET declarations and event macros into generated CUDA source.

Module Contents¶

tilelang.tools.cuda.iket.codegen.inject_iket_cuda(code, target, *, runtime_payloads)¶

Inject IKET declarations and event macros into generated CUDA source.

Parameters:

  • code (str)

  • target (Any)

  • runtime_payloads (bool)

Return type:

str