Update docs

Author: github-actions[bot]

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

@@ -132,6 +132,19 @@ Module Contents
 
 
 
+   .. py:method:: reshape(shape, rescale_num=1, rescale_den=1)
+
+      Reshape the input shape of the layout.
+
+      :param shape: The new input shape.
+      :type shape: list[PrimExpr] or list[int]
+      :param rescale_num: Rescale numerator for element size changes.
+      :type rescale_num: int
+      :param rescale_den: Rescale denominator for element size changes.
+      :type rescale_den: int
+
+
+
    .. py:method:: is_equal(other)
 
       Check if the current layout is equal to another layout.

_sources/autoapi/tilelang/layout/swizzle/index.rst.txt

@@ -38,41 +38,38 @@ Module Contents
 
 .. py:function:: make_tcgen05mma_swizzled_layout(buffer, continuity = None, k_major = True)
 
-.. py:function:: make_full_bank_swizzled_layout(*args)
+.. py:function:: make_full_bank_swizzled_layout(buffer)
 
-   :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)
+   :param buffer: BufferLikeType
 
    .. rubric:: Examples
 
    make_full_bank_swizzled_layout(buffer)
-   make_full_bank_swizzled_layout(stride, continuous, element_size)
 
 
-.. py:function:: make_half_bank_swizzled_layout(*args)
+.. py:function:: make_half_bank_swizzled_layout(buffer)
 
-   :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)
+   :param buffer: BufferLikeType
 
    .. rubric:: Examples
 
    make_half_bank_swizzled_layout(buffer)
-   make_half_bank_swizzled_layout(stride, continuous, element_size)
 
 
-.. py:function:: make_quarter_bank_swizzled_layout(*args)
+.. py:function:: make_quarter_bank_swizzled_layout(buffer)
 
-   :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)
+   :param buffer: BufferLikeType
 
    .. rubric:: Examples
 
    make_quarter_bank_swizzled_layout(buffer)
-   make_quarter_bank_swizzled_layout(stride, continuous, element_size)
 
 
 .. py:function:: make_linear_layout(buffer_or_load_or_region)
 
    Create a row-major linear layout for any dimension.
 
-   :param buffer_or_load_or_region: Buffer, BufferLoad, or BufferRegion
+   :param buffer_or_load_or_region: BufferLikeType
 
    :returns: A row-major linear layout
    :rtype: Layout
@@ -107,7 +104,7 @@ Module Contents
    entire buffer. This is useful for index buffers or masks that need to be
    accessed uniformly across all threads.
 
-   :param buffer: Buffer, BufferLoad, or BufferRegion to get shape information
+   :param buffer: BufferLikeType to get shape information
    :param threads: Number of threads (replicate extent)
 
    :returns: A fully replicated layout where each thread has a complete copy

autoapi/tilelang/layout/layout/index.html

@@ -724,6 +724,24 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 </dl>
 </dd></dl>
 
+<dl class="py method">
+<dt class="sig sig-object py" id="tilelang.layout.layout.Layout.reshape">
+<span class="sig-name descname"><span class="pre">reshape</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rescale_num</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rescale_den</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">1</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.layout.Layout.reshape" title="Link to this definition">¶</a></dt>
+<dd><p>Reshape the input shape of the layout.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters<span class="colon">:</span></dt>
+<dd class="field-odd"><ul class="simple">
+<li><p><strong>shape</strong> (<em>list</em><em>[</em><em>PrimExpr</em><em>] or </em><em>list</em><em>[</em><em>int</em><em>]</em>) – The new input shape.</p></li>
+<li><p><strong>rescale_num</strong> (<em>int</em>) – Rescale numerator for element size changes.</p></li>
+<li><p><strong>rescale_den</strong> (<em>int</em>) – Rescale denominator for element size changes.</p></li>
+</ul>
+</dd>
+<dt class="field-even">Return type<span class="colon">:</span></dt>
+<dd class="field-even"><p><a class="reference internal" href="#tilelang.layout.layout.Layout" title="tilelang.layout.layout.Layout">Layout</a></p>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.layout.layout.Layout.is_equal">
 <span class="sig-name descname"><span class="pre">is_equal</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">other</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.layout.Layout.is_equal" title="Link to this definition">¶</a></dt>
@@ -829,6 +847,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.repeat"><code class="docutils literal notranslate"><span class="pre">Layout.repeat()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.expand"><code class="docutils literal notranslate"><span class="pre">Layout.expand()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.inverse"><code class="docutils literal notranslate"><span class="pre">Layout.inverse()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.layout.layout.Layout.reshape"><code class="docutils literal notranslate"><span class="pre">Layout.reshape()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.is_equal"><code class="docutils literal notranslate"><span class="pre">Layout.is_equal()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.__call__"><code class="docutils literal notranslate"><span class="pre">Layout.__call__()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.layout.layout.Layout.__repr__"><code class="docutils literal notranslate"><span class="pre">Layout.__repr__()</span></code></a></li>

autoapi/tilelang/layout/swizzle/index.html

@@ -575,13 +575,13 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_tcgen05mma_swizzled_layout" title="tilelang.layout.swizzle.make_tcgen05mma_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_tcgen05mma_swizzled_layout</span></code></a>(buffer[, continuity, ...])</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_full_bank_swizzled_layout" title="tilelang.layout.swizzle.make_full_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_full_bank_swizzled_layout</span></code></a>(*args)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_full_bank_swizzled_layout" title="tilelang.layout.swizzle.make_full_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_full_bank_swizzled_layout</span></code></a>(buffer)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_half_bank_swizzled_layout" title="tilelang.layout.swizzle.make_half_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_half_bank_swizzled_layout</span></code></a>(*args)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_half_bank_swizzled_layout" title="tilelang.layout.swizzle.make_half_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_half_bank_swizzled_layout</span></code></a>(buffer)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_quarter_bank_swizzled_layout" title="tilelang.layout.swizzle.make_quarter_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_quarter_bank_swizzled_layout</span></code></a>(*args)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_quarter_bank_swizzled_layout" title="tilelang.layout.swizzle.make_quarter_bank_swizzled_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_quarter_bank_swizzled_layout</span></code></a>(buffer)</p></td>
 <td><p></p></td>
 </tr>
 <tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.swizzle.make_linear_layout" title="tilelang.layout.swizzle.make_linear_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_linear_layout</span></code></a>(buffer_or_load_or_region)</p></td>
@@ -608,7 +608,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>)</p></li>
+<li><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>)</p></li>
 <li><p><strong>k_major</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 <li><p><strong>allow_pad</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 </ul>
@@ -622,7 +622,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>)</p></li>
+<li><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>)</p></li>
 <li><p><strong>is_a</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 <li><p><strong>k_inner</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 </ul>
@@ -636,7 +636,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>)</p></li>
+<li><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>)</p></li>
 <li><p><strong>continuity</strong> (<em>int</em>)</p></li>
 <li><p><strong>k_major</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 </ul>
@@ -650,7 +650,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>)</p></li>
+<li><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>)</p></li>
 <li><p><strong>continuity</strong> (<em>int</em>)</p></li>
 <li><p><strong>k_major</strong> (<a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
 </ul>
@@ -660,41 +660,38 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.layout.swizzle.make_full_bank_swizzled_layout">
-<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_full_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_full_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_full_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_full_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
-<dd class="field-odd"><p><strong>args</strong> – buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)</p>
+<dd class="field-odd"><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>) – BufferLikeType</p>
 </dd>
 </dl>
 <p class="rubric">Examples</p>
-<p>make_full_bank_swizzled_layout(buffer)
-make_full_bank_swizzled_layout(stride, continuous, element_size)</p>
+<p>make_full_bank_swizzled_layout(buffer)</p>
 </dd></dl>
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.layout.swizzle.make_half_bank_swizzled_layout">
-<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_half_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_half_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_half_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_half_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
-<dd class="field-odd"><p><strong>args</strong> – buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)</p>
+<dd class="field-odd"><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>) – BufferLikeType</p>
 </dd>
 </dl>
 <p class="rubric">Examples</p>
-<p>make_half_bank_swizzled_layout(buffer)
-make_half_bank_swizzled_layout(stride, continuous, element_size)</p>
+<p>make_half_bank_swizzled_layout(buffer)</p>
 </dd></dl>
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.layout.swizzle.make_quarter_bank_swizzled_layout">
-<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_quarter_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_quarter_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.layout.swizzle.</span></span><span class="sig-name descname"><span class="pre">make_quarter_bank_swizzled_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.swizzle.make_quarter_bank_swizzled_layout" title="Link to this definition">¶</a></dt>
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
-<dd class="field-odd"><p><strong>args</strong> – buffer/BufferLoad/BufferRegion or (stride, continuous, element_size)</p>
+<dd class="field-odd"><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>) – BufferLikeType</p>
 </dd>
 </dl>
 <p class="rubric">Examples</p>
-<p>make_quarter_bank_swizzled_layout(buffer)
-make_quarter_bank_swizzled_layout(stride, continuous, element_size)</p>
+<p>make_quarter_bank_swizzled_layout(buffer)</p>
 </dd></dl>
 
 <dl class="py function">
@@ -703,7 +700,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd><p>Create a row-major linear layout for any dimension.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
-<dd class="field-odd"><p><strong>buffer_or_load_or_region</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>) – Buffer, BufferLoad, or BufferRegion</p>
+<dd class="field-odd"><p><strong>buffer_or_load_or_region</strong> (<em>tilelang._typing.BufferLikeType</em>) – BufferLikeType</p>
 </dd>
 <dt class="field-even">Returns<span class="colon">:</span></dt>
 <dd class="field-even"><p>A row-major linear layout</p>
@@ -755,7 +752,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em><em> | </em><em>tvm.tir.BufferLoad</em><em> | </em><em>tvm.tir.BufferRegion</em>) – Buffer, BufferLoad, or BufferRegion to get shape information</p></li>
+<li><p><strong>buffer</strong> (<em>tilelang._typing.BufferLikeType</em>) – BufferLikeType to get shape information</p></li>
 <li><p><strong>threads</strong> (<em>int</em>) – Number of threads (replicate extent)</p></li>
 </ul>
 </dd>

genindex.html

@@ -6258,7 +6258,11 @@ <h2>R</h2>
         <li><a href="autoapi/tilelang/autodd/index.html#tilelang.autodd.ParTaskManager.reset">reset() (tilelang.autodd.ParTaskManager method)</a>
 </li>
         <li><a href="autoapi/tilelang/language/customize/index.html#tilelang.language.customize.reshape">reshape() (in module tilelang.language.customize)</a>
+
+        <ul>
+          <li><a href="autoapi/tilelang/layout/layout/index.html#tilelang.layout.layout.Layout.reshape">(tilelang.layout.layout.Layout method)</a>
 </li>
+        </ul></li>
         <li><a href="autoapi/tilelang/jit/execution_backend/index.html#tilelang.jit.execution_backend.resolve_execution_backend">resolve_execution_backend() (in module tilelang.jit.execution_backend)</a>
 </li>
         <li><a href="autoapi/tilelang/transform/decouple_type_cast/index.html#tilelang.transform.decouple_type_cast.GlobalSharedBufferLoadCollector.result">result (tilelang.transform.decouple_type_cast.GlobalSharedBufferLoadCollector attribute)</a>