Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/language/index.rst.txt

@@ -28,11 +28,9 @@ Submodules
    /autoapi/tilelang/language/gemm/index
    /autoapi/tilelang/language/kernel/index
    /autoapi/tilelang/language/logical/index
+   /autoapi/tilelang/language/loop/index
    /autoapi/tilelang/language/math_intrinsics/index
    /autoapi/tilelang/language/overrides/index
-   /autoapi/tilelang/language/parallel/index
-   /autoapi/tilelang/language/persistent/index
-   /autoapi/tilelang/language/pipeline/index
    /autoapi/tilelang/language/print/index
    /autoapi/tilelang/language/proxy/index
    /autoapi/tilelang/language/reduce/index

_sources/autoapi/tilelang/language/loop/index.rst.txt

@@ -0,0 +1,71 @@
+tilelang.language.loop
+======================
+
+.. py:module:: tilelang.language.loop
+
+.. autoapi-nested-parse::
+
+   The language interface for tl programs.
+
+
+
+Functions
+---------
+
+.. autoapisummary::
+
+   tilelang.language.loop.Parallel
+   tilelang.language.loop.Persistent
+   tilelang.language.loop.Pipelined
+   tilelang.language.loop.serial
+
+
+Module Contents
+---------------
+
+.. py:function:: Parallel(*extents, coalesced_width = None)
+
+   Tools to construct nested parallel for loop.
+      This can be used to create element-wise tensor expression.
+
+   :param extents: The extents of the iteration.
+   :type extents: PrimExpr
+   :param coalesced_width: The coalesced width of the parallel loop.
+   :type coalesced_width: Optional[int]
+
+   :returns: **res** -- The ForFrame.
+   :rtype: frame.ForFrame
+
+
+.. py:function:: Persistent(domain, wave_size, index, group_size = 8)
+
+   Tools to construct persistent for loop.
+
+   :param domain: The list of dominators.
+   :type domain: List[tir.PrimExpr]
+   :param wave_size: The wave size.
+   :type wave_size: int
+   :param index: The tile index in one wave.
+   :type index: int
+   :param group_size: The group size.
+   :type group_size: tir.PrimExpr
+
+
+.. py:function:: Pipelined(start, stop = None, num_stages = 0, order = None, stage = None, sync = None, group = None)
+
+   Tools to construct pipelined for loop.
+
+   :param start: The minimum value of iteration.
+   :type start: PrimExpr
+   :param stop: The maximum value of iteration.
+   :type stop: PrimExpr
+   :param num_stages: The max number of buffer used between pipeline producers and consumers.
+                      if num_stages is 0, pipeline will not be enabled.
+   :type num_stages: int
+
+   :returns: **res** -- The ForFrame.
+   :rtype: frame.ForFrame
+
+
+.. py:function:: serial(start, stop = None, step = None, *, annotations = None)
+

_sources/autoapi/tilelang/language/parallel/index.rst.txt

@@ -29,6 +29,7 @@ Classes
    tilelang.language.v2.builder.BlockFrame
    tilelang.language.v2.builder.ContinueFrame
    tilelang.language.v2.builder.BreakFrame
+   tilelang.language.v2.builder.SerialForWithStep
    tilelang.language.v2.builder.Builder
    tilelang.language.v2.builder.PrimFunc
    tilelang.language.v2.builder.Macro
@@ -130,6 +131,26 @@ Module Contents
    They do not have any runtime representation in the generated TIR.
 
 
+.. py:class:: SerialForWithStep
+
+   .. py:attribute:: start
+      :type:  tvm.tir.expr.PrimExpr
+
+
+   .. py:attribute:: stop
+      :type:  tvm.tir.expr.PrimExpr
+
+
+   .. py:attribute:: step
+      :type:  tvm.tir.expr.PrimExpr
+
+
+   .. py:attribute:: annotations
+      :type:  dict[str, Any] | None
+      :value: None
+
+
+
 .. py:data:: ContinueOrBreak
 
 .. py:data:: AnyFrame

_sources/autoapi/tilelang/language/persistent/index.rst.txt

@@ -32,6 +32,7 @@ Functions
    tilelang.transform.LowerTileOp
    tilelang.transform.InjectSoftwarePipeline
    tilelang.transform.FrontendLegalize
+   tilelang.transform.LegalizeNegativeIndex
    tilelang.transform.InjectAssumes
    tilelang.transform.LowerHopperIntrin
    tilelang.transform.WarpSpecializedPipeline
@@ -127,6 +128,14 @@ Package Contents
    :rtype: tvm.transform.Pass
 
 
+.. py:function:: LegalizeNegativeIndex()
+
+   Legalize negative indices in buffer loads.
+
+   :returns: **fpass** -- The result pass
+   :rtype: tvm.transform.Pass
+
+
 .. py:function:: InjectAssumes()
 
    Inject Assumes

_sources/autoapi/tilelang/language/pipeline/index.rst.txt

@@ -362,11 +362,9 @@
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/gemm/index.html">tilelang.language.gemm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/kernel/index.html">tilelang.language.kernel</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/logical/index.html">tilelang.language.logical</a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang/language/loop/index.html">tilelang.language.loop</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/math_intrinsics/index.html">tilelang.language.math_intrinsics</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/overrides/index.html">tilelang.language.overrides</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/parallel/index.html">tilelang.language.parallel</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/persistent/index.html">tilelang.language.persistent</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/pipeline/index.html">tilelang.language.pipeline</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/print/index.html">tilelang.language.print</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/proxy/index.html">tilelang.language.proxy</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/reduce/index.html">tilelang.language.reduce</a></li>
@@ -636,11 +634,9 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Link to this
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/gemm/index.html">tilelang.language.gemm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/kernel/index.html">tilelang.language.kernel</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/logical/index.html">tilelang.language.logical</a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang/language/loop/index.html">tilelang.language.loop</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/math_intrinsics/index.html">tilelang.language.math_intrinsics</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/overrides/index.html">tilelang.language.overrides</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/parallel/index.html">tilelang.language.parallel</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/persistent/index.html">tilelang.language.persistent</a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang/language/pipeline/index.html">tilelang.language.pipeline</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/print/index.html">tilelang.language.print</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/proxy/index.html">tilelang.language.proxy</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/language/reduce/index.html">tilelang.language.reduce</a></li>

_sources/autoapi/tilelang/language/v2/builder/index.rst.txt

@@ -362,11 +362,9 @@
 <li class="toctree-l3"><a class="reference internal" href="../../language/gemm/index.html">tilelang.language.gemm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/kernel/index.html">tilelang.language.kernel</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/logical/index.html">tilelang.language.logical</a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../language/loop/index.html">tilelang.language.loop</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/math_intrinsics/index.html">tilelang.language.math_intrinsics</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/overrides/index.html">tilelang.language.overrides</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../../language/parallel/index.html">tilelang.language.parallel</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../../language/persistent/index.html">tilelang.language.persistent</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../../language/pipeline/index.html">tilelang.language.pipeline</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/print/index.html">tilelang.language.print</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/proxy/index.html">tilelang.language.proxy</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../language/reduce/index.html">tilelang.language.reduce</a></li>

_sources/autoapi/tilelang/transform/index.rst.txt

@@ -362,11 +362,9 @@
 <li class="toctree-l3"><a class="reference internal" href="../language/gemm/index.html">tilelang.language.gemm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/kernel/index.html">tilelang.language.kernel</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/logical/index.html">tilelang.language.logical</a></li>
+<li class="toctree-l3"><a class="reference internal" href="../language/loop/index.html">tilelang.language.loop</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/math_intrinsics/index.html">tilelang.language.math_intrinsics</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/overrides/index.html">tilelang.language.overrides</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../language/parallel/index.html">tilelang.language.parallel</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../language/persistent/index.html">tilelang.language.persistent</a></li>
-<li class="toctree-l3"><a class="reference internal" href="../language/pipeline/index.html">tilelang.language.pipeline</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/print/index.html">tilelang.language.print</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/proxy/index.html">tilelang.language.proxy</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../language/reduce/index.html">tilelang.language.reduce</a></li>