tilelang.jit.adapter.pto.wrapper¶

Host launch wrappers and ABI validation for PTODSL kernels.

Classes¶

TLPTOSourceWrapper

Wrapper for PTO JIT source.

TLPTOWrapper

A wrapper class for the TileLang backend.

Module Contents¶

class tilelang.jit.adapter.pto.wrapper.TLPTOSourceWrapper(scheduled_ir_module, source, target, device_mod=None, host_mod=None, pass_configs=None)¶

Wrapper for PTO JIT source.

PTO codegen emits PTODSL source. This wrapper generates the host launch stub directly from PTODSL/TIR metadata and keeps the PTODSL source for device compilation in libgen.

Parameters:

  • scheduled_ir_module (tvm.IRModule)

  • source (str)

  • target (tvm.target.Target)

  • device_mod (tvm.IRModule | None)

  • host_mod (tvm.IRModule | None)

  • pass_configs (dict[str, Any] | None)

pto_kernel_source¶
pto_kernel_names = []¶
lib_code = ''¶
get_dynamic_symbolic_set(prim_func)¶
class tilelang.jit.adapter.pto.wrapper.TLPTOWrapper(target)¶

Bases: tilelang.jit.adapter.wrapper.TLWrapper

A wrapper class for the TileLang backend.

Parameters:

target (tvm.target.Target)

wrap(source)¶
Parameters:

source (str)