tilelang.engine.callback
========================

.. py:module:: tilelang.engine.callback


Functions
---------

.. autoapisummary::

   tilelang.engine.callback.register_cuda_postproc
   tilelang.engine.callback.register_hip_postproc
   tilelang.engine.callback.register_cuda_postproc_callback
   tilelang.engine.callback.register_hip_postproc_callback


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

.. py:function:: register_cuda_postproc(func, override = True)

   Register a post-processing function for CUDA code generation.

   :param func: A callable that takes generated code (str) and target (Target) as input,
                and returns the processed code (str).
   :param override: Whether to override existing registered function. Defaults to True.


.. py:function:: register_hip_postproc(func, override = True)

   Register a post-processing function for HIP code generation.

   :param func: A callable that takes generated code (str) and target (Target) as input,
                and returns the processed code (str).
   :param override: Whether to override existing registered function. Defaults to True.


.. py:function:: register_cuda_postproc_callback(func = None, override = True)

   Decorator for registering CUDA post-processing callback function.

   Can be used with or without parentheses:
       @register_cuda_postproc_callback
       def func(code, target): ...

       @register_cuda_postproc_callback()
       def func(code, target): ...

       @register_cuda_postproc_callback(override=False)
       def func(code, target): ...

   :param func: The function to be decorated or a boolean override flag
   :param override: Whether to override existing registered function. Defaults to True.


.. py:function:: register_hip_postproc_callback(func = None, override = True)

   Decorator for registering HIP post-processing callback function.

   Can be used with or without parentheses:
       @register_hip_postproc_callback
       def func(code, target): ...

       @register_hip_postproc_callback()
       def func(code, target): ...

       @register_hip_postproc_callback(override=False)
       def func(code, target): ...

   :param func: The function to be decorated or a boolean override flag
   :param override: Whether to override existing registered function. Defaults to True.