Update docs

Author: github-actions[bot]

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

@@ -23,5 +23,5 @@ Module Contents
 
 .. py:function:: decompose_col_major(index_1d, basis)
 
-.. py:function:: make_metadata_layout(buffer, mma_dtype = 'float16', arch = 'sm90', backend = 'cutlass', **extra_args)
+.. py:function:: make_metadata_layout(buffer, mma_dtype = 'float16', backend = 'cutlass', arch = None, **extra_args)
 

_sources/autoapi/tilelang/utils/sparse/index.rst.txt

@@ -18,6 +18,8 @@ Functions
 .. autoapisummary::
 
    tilelang.utils.sparse.compress_sm90
+   tilelang.utils.sparse.compress_sm80
+   tilelang.utils.sparse.compress
 
 
 Module Contents
@@ -27,3 +29,10 @@ Module Contents
 
 .. py:function:: compress_sm90(A, block_k, transposed)
 
+.. py:function:: compress_sm80(A, transposed)
+
+.. py:function:: compress(A, transposed, arch = None, **kwargs)
+
+   Compress a tensor using the appropriate method based on the CUDA architecture.
+
+

autoapi/tilelang/layout/gemm_sp/index.html

@@ -479,7 +479,7 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.layout.gemm_sp.decompose_col_major" title="tilelang.layout.gemm_sp.decompose_col_major"><code class="xref py py-obj docutils literal notranslate"><span class="pre">decompose_col_major</span></code></a>(index_1d, basis)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.gemm_sp.make_metadata_layout" title="tilelang.layout.gemm_sp.make_metadata_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_metadata_layout</span></code></a>(buffer[, mma_dtype, arch, backend])</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.layout.gemm_sp.make_metadata_layout" title="tilelang.layout.gemm_sp.make_metadata_layout"><code class="xref py py-obj docutils literal notranslate"><span class="pre">make_metadata_layout</span></code></a>(buffer[, mma_dtype, backend, arch])</p></td>
 <td><p></p></td>
 </tr>
 </tbody>
@@ -506,14 +506,14 @@ <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.gemm_sp.make_metadata_layout">
-<span class="sig-prename descclassname"><span class="pre">tilelang.layout.gemm_sp.</span></span><span class="sig-name descname"><span class="pre">make_metadata_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mma_dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'float16'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">arch</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'sm90'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">backend</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cutlass'</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">extra_args</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.gemm_sp.make_metadata_layout" title="Link to this definition">¶</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.layout.gemm_sp.</span></span><span class="sig-name descname"><span class="pre">make_metadata_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mma_dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'float16'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">backend</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'cutlass'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">arch</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">extra_args</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.layout.gemm_sp.make_metadata_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"><ul class="simple">
 <li><p><strong>buffer</strong> (<em>tvm.tir.Buffer</em>)</p></li>
 <li><p><strong>mma_dtype</strong> (<em>str</em>)</p></li>
-<li><p><strong>arch</strong> (<em>str</em>)</p></li>
 <li><p><strong>backend</strong> (<em>str</em>)</p></li>
+<li><p><strong>arch</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>)</p></li>
 </ul>
 </dd>
 </dl>

autoapi/tilelang/utils/sparse/index.html

@@ -490,6 +490,12 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.sparse.compress_sm90" title="tilelang.utils.sparse.compress_sm90"><code class="xref py py-obj docutils literal notranslate"><span class="pre">compress_sm90</span></code></a>(A, block_k, transposed)</p></td>
 <td><p></p></td>
 </tr>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.sparse.compress_sm80" title="tilelang.utils.sparse.compress_sm80"><code class="xref py py-obj docutils literal notranslate"><span class="pre">compress_sm80</span></code></a>(A, transposed)</p></td>
+<td><p></p></td>
+</tr>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.sparse.compress" title="tilelang.utils.sparse.compress"><code class="xref py py-obj docutils literal notranslate"><span class="pre">compress</span></code></a>(A, transposed[, arch])</p></td>
+<td><p>Compress a tensor using the appropriate method based on the CUDA architecture.</p></td>
+</tr>
 </tbody>
 </table>
 </div>
@@ -518,6 +524,40 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.utils.sparse.compress_sm80">
+<span class="sig-prename descclassname"><span class="pre">tilelang.utils.sparse.</span></span><span class="sig-name descname"><span class="pre">compress_sm80</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">A</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">transposed</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.utils.sparse.compress_sm80" 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"><ul class="simple">
+<li><p><strong>A</strong> (<em>torch.Tensor</em>)</p></li>
+<li><p><strong>transposed</strong> (<em>bool</em>)</p></li>
+</ul>
+</dd>
+<dt class="field-even">Return type<span class="colon">:</span></dt>
+<dd class="field-even"><p>tuple[torch.Tensor, torch.Tensor]</p>
+</dd>
+</dl>
+</dd></dl>
+
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.utils.sparse.compress">
+<span class="sig-prename descclassname"><span class="pre">tilelang.utils.sparse.</span></span><span class="sig-name descname"><span class="pre">compress</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">A</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">transposed</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">arch</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.utils.sparse.compress" title="Link to this definition">¶</a></dt>
+<dd><p>Compress a tensor using the appropriate method based on the CUDA architecture.</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>A</strong> (<em>torch.Tensor</em>)</p></li>
+<li><p><strong>transposed</strong> (<em>bool</em>)</p></li>
+<li><p><strong>arch</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>)</p></li>
+</ul>
+</dd>
+<dt class="field-even">Return type<span class="colon">:</span></dt>
+<dd class="field-even"><p>tuple[torch.Tensor, torch.Tensor]</p>
+</dd>
+</dl>
+</dd></dl>
+
 </section>
 </section>
 
@@ -582,6 +622,8 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#module-contents">Module Contents</a><ul>
 <li><a class="reference internal" href="#tilelang.utils.sparse.compress_util"><code class="docutils literal notranslate"><span class="pre">compress_util</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.utils.sparse.compress_sm90"><code class="docutils literal notranslate"><span class="pre">compress_sm90()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.utils.sparse.compress_sm80"><code class="docutils literal notranslate"><span class="pre">compress_sm80()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.utils.sparse.compress"><code class="docutils literal notranslate"><span class="pre">compress()</span></code></a></li>
 </ul>
 </li>
 </ul>

genindex.html

@@ -1245,14 +1245,14 @@ <h2>C</h2>
           <li><a href="autoapi/tilelang/contrib/nvrtc/index.html#tilelang.contrib.nvrtc.compile_cuda">(in module tilelang.contrib.nvrtc)</a>
 </li>
         </ul></li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/tilelang/jit/adapter/libgen/index.html#tilelang.jit.adapter.libgen.LibraryGenerator.compile_flags">compile_flags (tilelang.jit.adapter.libgen.LibraryGenerator attribute)</a>
 
         <ul>
           <li><a href="autoapi/tilelang/jit/kernel/index.html#tilelang.jit.kernel.JITKernel.compile_flags">(tilelang.jit.kernel.JITKernel attribute)</a>
 </li>
         </ul></li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/tilelang/contrib/hipcc/index.html#tilelang.contrib.hipcc.compile_hip">compile_hip() (in module tilelang.contrib.hipcc)</a>
 </li>
         <li><a href="autoapi/tilelang/jit/adapter/libgen/index.html#tilelang.jit.adapter.libgen.LibraryGenerator.compile_lib">compile_lib() (tilelang.jit.adapter.libgen.LibraryGenerator method)</a>
@@ -1276,6 +1276,10 @@ <h2>C</h2>
 </li>
         </ul></li>
         <li><a href="autoapi/tilelang/env/index.html#tilelang.env.COMPOSABLE_KERNEL_NOT_FOUND_MESSAGE">COMPOSABLE_KERNEL_NOT_FOUND_MESSAGE (in module tilelang.env)</a>
+</li>
+        <li><a href="autoapi/tilelang/utils/sparse/index.html#tilelang.utils.sparse.compress">compress() (in module tilelang.utils.sparse)</a>
+</li>
+        <li><a href="autoapi/tilelang/utils/sparse/index.html#tilelang.utils.sparse.compress_sm80">compress_sm80() (in module tilelang.utils.sparse)</a>
 </li>
         <li><a href="autoapi/tilelang/utils/sparse/index.html#tilelang.utils.sparse.compress_sm90">compress_sm90() (in module tilelang.utils.sparse)</a>
 </li>