tilelang.jit.adapter.pto.wrapper
================================

.. py:module:: tilelang.jit.adapter.pto.wrapper

.. autoapi-nested-parse::

   Host launch wrappers and ABI validation for PTODSL kernels.



Classes
-------

.. autoapisummary::

   tilelang.jit.adapter.pto.wrapper.TLPTOSourceWrapper
   tilelang.jit.adapter.pto.wrapper.TLPTOWrapper


Module Contents
---------------

.. py:class:: 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.


   .. py:attribute:: pto_kernel_source


   .. py:attribute:: pto_kernel_names
      :value: []



   .. py:attribute:: lib_code
      :value: ''



   .. py:method:: get_dynamic_symbolic_set(prim_func)


.. py:class:: TLPTOWrapper(target)

   Bases: :py:obj:`tilelang.jit.adapter.wrapper.TLWrapper`


   A wrapper class for the TileLang backend.


   .. py:method:: wrap(source)