Update docs

Author: LeiWang1999

api/tilelang.autotuner.html

@@ -595,7 +595,7 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 
 <dl class="py class">
 <dt class="sig sig-object py" id="tilelang.autotuner.JITContext">
-<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">JITContext</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">out_idx</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">ref_prog</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Callable</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">supply_prog</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Callable</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rtol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">atol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_mismatched_ratio</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">skip_check</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">cache_input_tensors</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">profiler</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler" title="tilelang.profiler.Profiler"><span class="pre">Profiler</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Literal</span><span class="p"><span class="pre">[</span></span><span class="s"><span class="pre">'cuda'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'hip'</span></span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.autotuner.JITContext" title="Permalink to this definition">#</a></dt>
+<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">JITContext</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">out_idx</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">ref_prog</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Callable</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">supply_prog</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Callable</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rtol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">atol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_mismatched_ratio</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">skip_check</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">cache_input_tensors</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">kernel</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel" title="tilelang.jit.kernel.JITKernel"><span class="pre">JITKernel</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">supply_type</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="tilelang.utils.tensor.html#tilelang.utils.tensor.TensorSupplyType" title="tilelang.utils.tensor.TensorSupplyType"><span class="pre">TensorSupplyType</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Literal</span><span class="p"><span class="pre">[</span></span><span class="s"><span class="pre">'cuda'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'hip'</span></span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.autotuner.JITContext" title="Permalink to this definition">#</a></dt>
 <dd><p>Bases: <code class="xref py py-class docutils literal notranslate"><span class="pre">object</span></code></p>
 <p>Context object for Just-In-Time compilation settings.</p>
 <dl class="py attribute">
@@ -687,12 +687,23 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 </dd></dl>
 
 <dl class="py attribute">
-<dt class="sig sig-object py" id="tilelang.autotuner.JITContext.profiler">
-<span class="sig-name descname"><span class="pre">profiler</span></span><a class="headerlink" href="#tilelang.autotuner.JITContext.profiler" title="Permalink to this definition">#</a></dt>
-<dd><p>Profiler instance for performance measurement.</p>
+<dt class="sig sig-object py" id="tilelang.autotuner.JITContext.kernel">
+<span class="sig-name descname"><span class="pre">kernel</span></span><a class="headerlink" href="#tilelang.autotuner.JITContext.kernel" title="Permalink to this definition">#</a></dt>
+<dd><p>JITKernel instance for performance measurement.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Type<span class="colon">:</span></dt>
-<dd class="field-odd"><p><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler" title="tilelang.profiler.Profiler">tilelang.profiler.Profiler</a></p>
+<dd class="field-odd"><p><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel" title="tilelang.jit.kernel.JITKernel">tilelang.jit.kernel.JITKernel</a></p>
+</dd>
+</dl>
+</dd></dl>
+
+<dl class="py attribute">
+<dt class="sig sig-object py" id="tilelang.autotuner.JITContext.supply_type">
+<span class="sig-name descname"><span class="pre">supply_type</span></span><a class="headerlink" href="#tilelang.autotuner.JITContext.supply_type" title="Permalink to this definition">#</a></dt>
+<dd><p>Type of tensor supply mechanism.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Type<span class="colon">:</span></dt>
+<dd class="field-odd"><p><a class="reference internal" href="tilelang.utils.tensor.html#tilelang.utils.tensor.TensorSupplyType" title="tilelang.utils.tensor.TensorSupplyType">tilelang.utils.tensor.TensorSupplyType</a></p>
 </dd>
 </dl>
 </dd></dl>
@@ -720,17 +731,17 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 
 <dl class="py attribute">
 <dt class="sig sig-object py" id="id8">
-<span class="sig-name descname"><span class="pre">max_mismatched_ratio</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">float</span></em><a class="headerlink" href="#id8" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">kernel</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel" title="tilelang.jit.kernel.JITKernel"><span class="pre">JITKernel</span></a></em><a class="headerlink" href="#id8" title="Permalink to this definition">#</a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
 <dt class="sig sig-object py" id="id9">
-<span class="sig-name descname"><span class="pre">out_idx</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#id9" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">max_mismatched_ratio</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">float</span></em><a class="headerlink" href="#id9" title="Permalink to this definition">#</a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
 <dt class="sig sig-object py" id="id10">
-<span class="sig-name descname"><span class="pre">profiler</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference internal" href="tilelang.profiler.html#tilelang.profiler.Profiler" title="tilelang.profiler.Profiler"><span class="pre">Profiler</span></a></em><a class="headerlink" href="#id10" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">out_idx</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#id10" title="Permalink to this definition">#</a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
@@ -755,7 +766,12 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 
 <dl class="py attribute">
 <dt class="sig sig-object py" id="id15">
-<span class="sig-name descname"><span class="pre">target</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">Literal</span><span class="p"><span class="pre">[</span></span><span class="s"><span class="pre">'cuda'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'hip'</span></span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#id15" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">supply_type</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference internal" href="tilelang.utils.tensor.html#tilelang.utils.tensor.TensorSupplyType" title="tilelang.utils.tensor.TensorSupplyType"><span class="pre">TensorSupplyType</span></a></em><a class="headerlink" href="#id15" title="Permalink to this definition">#</a></dt>
+<dd></dd></dl>
+
+<dl class="py attribute">
+<dt class="sig sig-object py" id="id16">
+<span class="sig-name descname"><span class="pre">target</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">Literal</span><span class="p"><span class="pre">[</span></span><span class="s"><span class="pre">'cuda'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'hip'</span></span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#id16" title="Permalink to this definition">#</a></dt>
 <dd></dd></dl>
 
 </dd></dl>
@@ -917,18 +933,20 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 <li><a class="reference internal" href="#tilelang.autotuner.JITContext.max_mismatched_ratio"><code class="docutils literal notranslate"><span class="pre">JITContext.max_mismatched_ratio</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.autotuner.JITContext.skip_check"><code class="docutils literal notranslate"><span class="pre">JITContext.skip_check</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.autotuner.JITContext.cache_input_tensors"><code class="docutils literal notranslate"><span class="pre">JITContext.cache_input_tensors</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.autotuner.JITContext.profiler"><code class="docutils literal notranslate"><span class="pre">JITContext.profiler</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.autotuner.JITContext.kernel"><code class="docutils literal notranslate"><span class="pre">JITContext.kernel</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.autotuner.JITContext.supply_type"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_type</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.autotuner.JITContext.target"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
 <li><a class="reference internal" href="#id6"><code class="docutils literal notranslate"><span class="pre">JITContext.atol</span></code></a></li>
 <li><a class="reference internal" href="#id7"><code class="docutils literal notranslate"><span class="pre">JITContext.cache_input_tensors</span></code></a></li>
-<li><a class="reference internal" href="#id8"><code class="docutils literal notranslate"><span class="pre">JITContext.max_mismatched_ratio</span></code></a></li>
-<li><a class="reference internal" href="#id9"><code class="docutils literal notranslate"><span class="pre">JITContext.out_idx</span></code></a></li>
-<li><a class="reference internal" href="#id10"><code class="docutils literal notranslate"><span class="pre">JITContext.profiler</span></code></a></li>
+<li><a class="reference internal" href="#id8"><code class="docutils literal notranslate"><span class="pre">JITContext.kernel</span></code></a></li>
+<li><a class="reference internal" href="#id9"><code class="docutils literal notranslate"><span class="pre">JITContext.max_mismatched_ratio</span></code></a></li>
+<li><a class="reference internal" href="#id10"><code class="docutils literal notranslate"><span class="pre">JITContext.out_idx</span></code></a></li>
 <li><a class="reference internal" href="#id11"><code class="docutils literal notranslate"><span class="pre">JITContext.ref_prog</span></code></a></li>
 <li><a class="reference internal" href="#id12"><code class="docutils literal notranslate"><span class="pre">JITContext.rtol</span></code></a></li>
 <li><a class="reference internal" href="#id13"><code class="docutils literal notranslate"><span class="pre">JITContext.skip_check</span></code></a></li>
 <li><a class="reference internal" href="#id14"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_prog</span></code></a></li>
-<li><a class="reference internal" href="#id15"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
+<li><a class="reference internal" href="#id15"><code class="docutils literal notranslate"><span class="pre">JITContext.supply_type</span></code></a></li>
+<li><a class="reference internal" href="#id16"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
 </ul>
 </li>
 <li><a class="reference internal" href="#tilelang.autotuner.autotune"><code class="docutils literal notranslate"><span class="pre">autotune()</span></code></a></li>