tilelang.language.warpgroup
===========================

.. py:module:: tilelang.language.warpgroup

.. autoapi-nested-parse::

   The language interface for tl programs.



Attributes
----------

.. autoapisummary::

   tilelang.language.warpgroup.ws


Classes
-------

.. autoapisummary::

   tilelang.language.warpgroup.WarpSpecializeFrame


Functions
---------

.. autoapisummary::

   tilelang.language.warpgroup.WarpSpecialize


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

.. py:class:: WarpSpecializeFrame

   Bases: :py:obj:`tvm.script.ir_builder.tir.frame.TIRFrame`


   WarpSpecializeFrame is a custom TIRFrame that manages warp group indices
   and handles the entry and exit of the kernel launch scope.


.. py:function:: WarpSpecialize(*warp_group_idx)

   Tools to construct a warp group frame.

   :param warp_group_idx: A integer representing warp group index
                          Or a list of integers representing blockDim.(x|y|z)
                          if the value is -1, we skip the threadIdx.x binding.
   :type warp_group_idx: int

   :returns: * **res** (*Tuple[frame.LaunchThreadFrame]*) -- The result LaunchThreadFrame.
             * *Examples* -- >>> T.ws(0) -> if tx < 128
               >>> T.ws(1) -> if tx >= 128 and tx < 256
               >>> T.ws(0, 1) -> if tx < 128 or (tx >= 128 and tx < 256)


.. py:data:: ws