tilelang.analysis.fragment_loop_checker
=======================================

.. py:module:: tilelang.analysis.fragment_loop_checker


Functions
---------

.. autoapisummary::

   tilelang.analysis.fragment_loop_checker.collect_local_buffer_accesses
   tilelang.analysis.fragment_loop_checker.FragmentLoopChecker


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

.. py:function:: collect_local_buffer_accesses(statement)

   Collect local buffer accesses in the loop body.

   :param statement: The TIR statement to analyze

   :returns: Tuple of buffer accesses in the loop body.


.. py:function:: FragmentLoopChecker()

   When using T.Parallel over a local/fragment buffer, there are several restrictions:
   to ensure that the parallelization is valid.

   1. The range of loop can not be symbolic.

   :returns: A prim_func_pass that applies the transformation