Update docs

Author: LeiWang1999

api/tilelang.autotuner.html

@@ -443,18 +443,20 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.atol"><code class="docutils literal notranslate"><span class="pre">CompileArgs.atol</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.max_mismatched_ratio"><code class="docutils literal notranslate"><span class="pre">CompileArgs.max_mismatched_ratio</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.skip_check"><code class="docutils literal notranslate"><span class="pre">CompileArgs.skip_check</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.manual_check_prog"><code class="docutils literal notranslate"><span class="pre">CompileArgs.manual_check_prog</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.cache_input_tensors"><code class="docutils literal notranslate"><span class="pre">CompileArgs.cache_input_tensors</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.CompileArgs.target"><code class="docutils literal notranslate"><span class="pre">CompileArgs.target</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id10"><code class="docutils literal notranslate"><span class="pre">CompileArgs.atol</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id11"><code class="docutils literal notranslate"><span class="pre">CompileArgs.cache_input_tensors</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id12"><code class="docutils literal notranslate"><span class="pre">CompileArgs.max_mismatched_ratio</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id13"><code class="docutils literal notranslate"><span class="pre">CompileArgs.out_idx</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id14"><code class="docutils literal notranslate"><span class="pre">CompileArgs.ref_prog</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id15"><code class="docutils literal notranslate"><span class="pre">CompileArgs.rtol</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id16"><code class="docutils literal notranslate"><span class="pre">CompileArgs.skip_check</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id17"><code class="docutils literal notranslate"><span class="pre">CompileArgs.supply_prog</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id18"><code class="docutils literal notranslate"><span class="pre">CompileArgs.supply_type</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id19"><code class="docutils literal notranslate"><span class="pre">CompileArgs.target</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id12"><code class="docutils literal notranslate"><span class="pre">CompileArgs.manual_check_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id13"><code class="docutils literal notranslate"><span class="pre">CompileArgs.max_mismatched_ratio</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id14"><code class="docutils literal notranslate"><span class="pre">CompileArgs.out_idx</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id15"><code class="docutils literal notranslate"><span class="pre">CompileArgs.ref_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id16"><code class="docutils literal notranslate"><span class="pre">CompileArgs.rtol</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id17"><code class="docutils literal notranslate"><span class="pre">CompileArgs.skip_check</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id18"><code class="docutils literal notranslate"><span class="pre">CompileArgs.supply_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id19"><code class="docutils literal notranslate"><span class="pre">CompileArgs.supply_type</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id20"><code class="docutils literal notranslate"><span class="pre">CompileArgs.target</span></code></a></li>
 </ul>
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.JITContext"><code class="docutils literal notranslate"><span class="pre">JITContext</span></code></a><ul>
@@ -469,17 +471,18 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.JITContext.kernel"><code class="docutils literal notranslate"><span class="pre">JITContext.kernel</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.JITContext.supply_type"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_type</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.JITContext.target"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id20"><code class="docutils literal notranslate"><span class="pre">JITContext.atol</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id21"><code class="docutils literal notranslate"><span class="pre">JITContext.cache_input_tensors</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id22"><code class="docutils literal notranslate"><span class="pre">JITContext.kernel</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id23"><code class="docutils literal notranslate"><span class="pre">JITContext.max_mismatched_ratio</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id24"><code class="docutils literal notranslate"><span class="pre">JITContext.out_idx</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id25"><code class="docutils literal notranslate"><span class="pre">JITContext.ref_prog</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id26"><code class="docutils literal notranslate"><span class="pre">JITContext.rtol</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id27"><code class="docutils literal notranslate"><span class="pre">JITContext.skip_check</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id28"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_prog</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id29"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_type</span></code></a></li>
-<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id30"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id21"><code class="docutils literal notranslate"><span class="pre">JITContext.atol</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id22"><code class="docutils literal notranslate"><span class="pre">JITContext.cache_input_tensors</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id23"><code class="docutils literal notranslate"><span class="pre">JITContext.kernel</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.JITContext.manual_check_prog"><code class="docutils literal notranslate"><span class="pre">JITContext.manual_check_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id24"><code class="docutils literal notranslate"><span class="pre">JITContext.max_mismatched_ratio</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id25"><code class="docutils literal notranslate"><span class="pre">JITContext.out_idx</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id26"><code class="docutils literal notranslate"><span class="pre">JITContext.ref_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id27"><code class="docutils literal notranslate"><span class="pre">JITContext.rtol</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id28"><code class="docutils literal notranslate"><span class="pre">JITContext.skip_check</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id29"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_prog</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id30"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_type</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.autotuner.html#id31"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
 </ul>
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.autotune"><code class="docutils literal notranslate"><span class="pre">autotune()</span></code></a></li>
@@ -995,6 +998,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler.determine_profiler"><code class="docutils literal notranslate"><span class="pre">Profiler.determine_profiler()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler.do_bench"><code class="docutils literal notranslate"><span class="pre">Profiler.do_bench()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler.func"><code class="docutils literal notranslate"><span class="pre">Profiler.func</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler.manual_assert_close"><code class="docutils literal notranslate"><span class="pre">Profiler.manual_assert_close()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#id1"><code class="docutils literal notranslate"><span class="pre">Profiler.params</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#id2"><code class="docutils literal notranslate"><span class="pre">Profiler.result_idx</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler.run_once"><code class="docutils literal notranslate"><span class="pre">Profiler.run_once()</span></code></a></li>

api/tilelang.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">0x7fa60b69ca60&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">0x7f7cc09c9090&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">0x7fa60b69ca60&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">0x7f7cc09c9090&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.logical.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">0x7fa60b69ca90&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">0x7f7cc09c90c0&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>