Update docs

Author: LeiWang1999

api/modules.html

@@ -474,7 +474,7 @@ <h1>Python API<a class="headerlink" href="#python-api" title="Permalink to this
 </ul>
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html">tilelang.testing package</a><ul>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.testing.html#module-tilelang.testing">Module contents</a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.testing.html#module-contents">Module contents</a></li>
 </ul>
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.tools.html">tilelang.tools package</a><ul>

api/tilelang.html

@@ -1007,15 +1007,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 </ul>
 </li>
 <li class="toctree-l1"><a class="reference internal" href="tilelang.testing.html">tilelang.testing package</a><ul>
-<li class="toctree-l2"><a class="reference internal" href="tilelang.testing.html#module-tilelang.testing">Module contents</a><ul>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.main"><code class="docutils literal notranslate"><span class="pre">main()</span></code></a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.requires_cuda_compute_version"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version()</span></code></a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.requires_cuda_compute_version_eq"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_eq()</span></code></a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.requires_cuda_compute_version_ge"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_ge()</span></code></a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.requires_cuda_compute_version_gt"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_gt()</span></code></a></li>
-<li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html#tilelang.testing.set_random_seed"><code class="docutils literal notranslate"><span class="pre">set_random_seed()</span></code></a></li>
-</ul>
-</li>
+<li class="toctree-l2"><a class="reference internal" href="tilelang.testing.html#module-contents">Module contents</a></li>
 </ul>
 </li>
 <li class="toctree-l1"><a class="reference internal" href="tilelang.tools.html">tilelang.tools package</a><ul>

api/tilelang.language.logical.html

@@ -406,7 +406,7 @@
 <p>The language interface for tl programs.</p>
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.all_of">
-<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f6af9148190&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f24064baef0&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
 <dd><p>Check if all elements in the buffer are true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -420,7 +420,7 @@
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.any_of">
-<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f6af9148190&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f24064baef0&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
 <dd><p>Check if any element in the buffer is true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.language.proxy.html

@@ -454,7 +454,7 @@
 <p>Buffer proxy class for constructing tir buffer.</p>
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
-<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f6af9148160&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f24064baec0&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
 <dd><p>Create a buffer from a pointer, shape, and data type.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.testing.html

@@ -403,56 +403,8 @@
         <article role="main">
           <section id="tilelang-testing-package">
 <h1>tilelang.testing package<a class="headerlink" href="#tilelang-testing-package" title="Permalink to this heading">#</a></h1>
-<section id="module-tilelang.testing">
-<span id="module-contents"></span><h2>Module contents<a class="headerlink" href="#module-tilelang.testing" title="Permalink to this heading">#</a></h2>
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.main">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">main</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.testing.main" title="Permalink to this definition">#</a></dt>
-<dd></dd></dl>
-
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.requires_cuda_compute_version">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">requires_cuda_compute_version</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">major_version</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">minor_version</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mode</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'ge'</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.testing.requires_cuda_compute_version" title="Permalink to this definition">#</a></dt>
-<dd><p>Mark a test as requiring at least a compute architecture</p>
-<p>Unit test marked with this decorator will run only if the CUDA
-compute architecture of the GPU is at least <cite>(major_version,
-minor_version)</cite>.</p>
-<p>This also marks the test as requiring a cuda support.</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>major_version</strong> (<em>int</em>) – The major version of the (major,minor) version tuple.</p></li>
-<li><p><strong>minor_version</strong> (<em>int</em>) – The minor version of the (major,minor) version tuple.</p></li>
-<li><p><strong>mode</strong> (<em>str</em>) – The mode of the comparison.
-- “ge”: greater than or equal to
-- “gt”: greater than
-- “le”: less than or equal to
-- “lt”: less than</p></li>
-</ul>
-</dd>
-</dl>
-</dd></dl>
-
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.requires_cuda_compute_version_eq">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">requires_cuda_compute_version_eq</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">major_version</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">minor_version</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.testing.requires_cuda_compute_version_eq" title="Permalink to this definition">#</a></dt>
-<dd></dd></dl>
-
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.requires_cuda_compute_version_ge">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">requires_cuda_compute_version_ge</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">major_version</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">minor_version</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.testing.requires_cuda_compute_version_ge" title="Permalink to this definition">#</a></dt>
-<dd></dd></dl>
-
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.requires_cuda_compute_version_gt">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">requires_cuda_compute_version_gt</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">major_version</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">minor_version</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.testing.requires_cuda_compute_version_gt" title="Permalink to this definition">#</a></dt>
-<dd></dd></dl>
-
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.testing.set_random_seed">
-<span class="sig-prename descclassname"><span class="pre">tilelang.testing.</span></span><span class="sig-name descname"><span class="pre">set_random_seed</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">seed</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">42</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">None</span></span></span><a class="headerlink" href="#tilelang.testing.set_random_seed" title="Permalink to this definition">#</a></dt>
-<dd></dd></dl>
-
+<section id="module-contents">
+<h2>Module contents<a class="headerlink" href="#module-contents" title="Permalink to this heading">#</a></h2>
 </section>
 </section>
 
@@ -512,15 +464,7 @@ <h1>tilelang.testing package<a class="headerlink" href="#tilelang-testing-packag
           <div class="toc-tree">
             <ul>
 <li><a class="reference internal" href="#">tilelang.testing package</a><ul>
-<li><a class="reference internal" href="#module-tilelang.testing">Module contents</a><ul>
-<li><a class="reference internal" href="#tilelang.testing.main"><code class="docutils literal notranslate"><span class="pre">main()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.testing.requires_cuda_compute_version"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.testing.requires_cuda_compute_version_eq"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_eq()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.testing.requires_cuda_compute_version_ge"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_ge()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.testing.requires_cuda_compute_version_gt"><code class="docutils literal notranslate"><span class="pre">requires_cuda_compute_version_gt()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.testing.set_random_seed"><code class="docutils literal notranslate"><span class="pre">set_random_seed()</span></code></a></li>
-</ul>
-</li>
+<li><a class="reference internal" href="#module-contents">Module contents</a></li>
 </ul>
 </li>
 </ul>