Update docs

Author: LeiWang1999

api/tilelang.html

@@ -977,6 +977,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l2"><a class="reference internal" href="tilelang.transform.html#module-tilelang.transform">Module contents</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.AnnotateDeviceRegions"><code class="docutils literal notranslate"><span class="pre">AnnotateDeviceRegions()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.ClusterPlanning"><code class="docutils literal notranslate"><span class="pre">ClusterPlanning()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.ConfigIndexBitwidth"><code class="docutils literal notranslate"><span class="pre">ConfigIndexBitwidth()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.FrontendLegalize"><code class="docutils literal notranslate"><span class="pre">FrontendLegalize()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.IfStmtBinding"><code class="docutils literal notranslate"><span class="pre">IfStmtBinding()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.InjectFenceProxy"><code class="docutils literal notranslate"><span class="pre">InjectFenceProxy()</span></code></a></li>

api/tilelang.jit.html

@@ -495,7 +495,30 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.jit.compile">
 <span class="sig-prename descclassname"><span class="pre">tilelang.jit.</span></span><span class="sig-name descname"><span class="pre">compile</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">func</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">PrimFunc</span><span class="p"><span class="pre">]</span></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">None</span></span></em>, <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">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">Union</span><span class="p"><span class="pre">[</span></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><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">int</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></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">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">execution_backend</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">'dlpack'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'ctypes'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'cython'</span></span><span class="p"><span class="pre">]</span></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">'cython'</span></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">Union</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Target</span><span class="p"><span class="pre">]</span></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">'auto'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target_host</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">Union</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Target</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></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">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">verbose</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</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">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">pass_configs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">Dict</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Any</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></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">None</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><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></span><a class="headerlink" href="#tilelang.jit.compile" title="Permalink to this definition">#</a></dt>
-<dd><p>Compile the given TileLang PrimFunc with TVM and build a JITKernel.</p>
+<dd><p>Compile the given TileLang PrimFunc with TVM and build a JITKernel.
+:param func: The TileLang TIR function to compile and wrap.
+:type func: tvm.tir.PrimFunc, optional
+:param out_idx: Index(es) of the output tensors to return (default: None).
+:type out_idx: Union[List[int], int], optional
+:param execution_backend: Execution backend to use for kernel execution (default: “dlpack”).
+:type execution_backend: Literal[“dlpack”, “ctypes”], optional
+:param target: Compilation target, either as a string or a TVM Target object (default: “auto”).
+:type target: Union[str, Target], optional
+:param target_host: Target host for cross-compilation (default: None).
+:type target_host: Union[str, Target], optional
+:param verbose: Whether to enable verbose output (default: False).
+:type verbose: bool, optional
+:param pass_configs: Additional keyword arguments to pass to the Compiler PassContext.</p>
+<blockquote>
+<div><dl class="simple">
+<dt>Available options:</dt><dd><p>“tir.disable_vectorize”: bool, default: False
+“tl.disable_tma_lower”: bool, default: False
+“tl.config_index_bitwidth”: int, default: None</p>
+</dd>
+</dl>
+</div></blockquote>
+<dl class="field-list simple">
+</dl>
 </dd></dl>
 
 <dl class="py function">

api/tilelang.language.proxy.html

@@ -450,7 +450,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">0x7fd5a5a2ffd0&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">0x7f2fa2d47880&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.transform.html

@@ -441,6 +441,21 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.transform.ConfigIndexBitwidth">
+<span class="sig-prename descclassname"><span class="pre">tilelang.transform.</span></span><span class="sig-name descname"><span class="pre">ConfigIndexBitwidth</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.ConfigIndexBitwidth" title="Permalink to this definition">#</a></dt>
+<dd><p>Config index bitwidth.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Returns<span class="colon">:</span></dt>
+<dd class="field-odd"><p><ul class="simple">
+<li><p><strong>fpass</strong> (<em>tvm.transform.Pass</em>) – The result pass</p></li>
+<li><p><em>—-</em></p></li>
+</ul>
+</p>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.transform.FrontendLegalize">
 <span class="sig-prename descclassname"><span class="pre">tilelang.transform.</span></span><span class="sig-name descname"><span class="pre">FrontendLegalize</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.FrontendLegalize" title="Permalink to this definition">#</a></dt>
@@ -811,6 +826,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li><a class="reference internal" href="#module-tilelang.transform">Module contents</a><ul>
 <li><a class="reference internal" href="#tilelang.transform.AnnotateDeviceRegions"><code class="docutils literal notranslate"><span class="pre">AnnotateDeviceRegions()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.ClusterPlanning"><code class="docutils literal notranslate"><span class="pre">ClusterPlanning()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.transform.ConfigIndexBitwidth"><code class="docutils literal notranslate"><span class="pre">ConfigIndexBitwidth()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.FrontendLegalize"><code class="docutils literal notranslate"><span class="pre">FrontendLegalize()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.IfStmtBinding"><code class="docutils literal notranslate"><span class="pre">IfStmtBinding()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.InjectFenceProxy"><code class="docutils literal notranslate"><span class="pre">InjectFenceProxy()</span></code></a></li>

genindex.html

@@ -766,6 +766,8 @@ <h2>C</h2>
         <li><a href="api/tilelang.layout.fragment.html#tilelang.layout.fragment.Fragment.condense_rep_var">condense_rep_var() (tilelang.layout.fragment.Fragment method)</a>
 </li>
         <li><a href="api/tilelang.autotuner.html#id0">config (tilelang.autotuner.AutotuneResult attribute)</a>, <a href="api/tilelang.autotuner.html#tilelang.autotuner.AutotuneResult.config">[1]</a>
+</li>
+        <li><a href="api/tilelang.transform.html#tilelang.transform.ConfigIndexBitwidth">ConfigIndexBitwidth() (in module tilelang.transform)</a>
 </li>
         <li><a href="api/tilelang.carver.roller.shape_inference.tir.html#tilelang.carver.roller.shape_inference.tir.InputShapeInference.construct_dependency_target">construct_dependency_target() (tilelang.carver.roller.shape_inference.tir.InputShapeInference method)</a>
 </li>