Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/jit/adapter/wrapper/index.rst.txt

@@ -472,7 +472,7 @@ Module Contents
    .. py:method:: generate_l2_persistent_map(function_name)
 
 
-   .. py:method:: generate_tma_descriptor_args(desc_name_map)
+   .. py:method:: generate_tma_descriptor_args(desc_name_map, desc_name_var_map)
 
 
    .. py:method:: parse_source_information()
@@ -496,6 +496,9 @@ Module Contents
    .. py:property:: device_func
 
 
+   .. py:property:: host_func
+
+
 .. py:class:: TLNVRTCSourceWrapper(scheduled_ir_module, source, target, device_mod = None, host_mod = None, pass_configs = None)
 
    Bases: :py:obj:`TLCUDASourceWrapper`

autoapi/tilelang/jit/adapter/wrapper/index.html

@@ -915,10 +915,13 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.generate_tma_descriptor_args">
-<span class="sig-name descname"><span class="pre">generate_tma_descriptor_args</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">desc_name_map</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.generate_tma_descriptor_args" title="Link to this definition">¶</a></dt>
+<span class="sig-name descname"><span class="pre">generate_tma_descriptor_args</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">desc_name_map</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">desc_name_var_map</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.generate_tma_descriptor_args" 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"><p><strong>desc_name_map</strong> (<em>Dict</em><em>[</em><em>str</em><em>, </em><em>str</em><em>]</em>)</p>
+<dd class="field-odd"><ul class="simple">
+<li><p><strong>desc_name_map</strong> (<em>Dict</em><em>[</em><em>str</em><em>, </em><em>str</em><em>]</em>)</p></li>
+<li><p><strong>desc_name_var_map</strong> (<em>Dict</em><em>[</em><em>str</em><em>, </em><em>tilelang.tvm.tir.Var</em><em>]</em>)</p></li>
+</ul>
 </dd>
 <dt class="field-even">Return type<span class="colon">:</span></dt>
 <dd class="field-even"><p>str</p>
@@ -971,6 +974,11 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">device_func</span></span><a class="headerlink" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.device_func" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
+<dl class="py property">
+<dt class="sig sig-object py" id="tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.host_func">
+<em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">host_func</span></span><a class="headerlink" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.host_func" title="Link to this definition">¶</a></dt>
+<dd></dd></dl>
+
 </dd></dl>
 
 <dl class="py class">
@@ -1407,6 +1415,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.get_stream_type"><code class="docutils literal notranslate"><span class="pre">TLCUDASourceWrapper.get_stream_type()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.prim_func"><code class="docutils literal notranslate"><span class="pre">TLCUDASourceWrapper.prim_func</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.device_func"><code class="docutils literal notranslate"><span class="pre">TLCUDASourceWrapper.device_func</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.host_func"><code class="docutils literal notranslate"><span class="pre">TLCUDASourceWrapper.host_func</span></code></a></li>
 </ul>
 </li>
 <li><a class="reference internal" href="#tilelang.jit.adapter.wrapper.TLNVRTCSourceWrapper"><code class="docutils literal notranslate"><span class="pre">TLNVRTCSourceWrapper</span></code></a><ul>

genindex.html

@@ -2309,10 +2309,10 @@ <h2>H</h2>
 </li>
         <li><a href="autoapi/tilelang/contrib/rocm/index.html#tilelang.contrib.rocm.have_matrixcore">have_matrixcore() (in module tilelang.contrib.rocm)</a>
 </li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/tilelang/contrib/nvcc/index.html#tilelang.contrib.nvcc.have_tensorcore">have_tensorcore() (in module tilelang.contrib.nvcc)</a>
 </li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/tilelang/contrib/nvcc/index.html#tilelang.contrib.nvcc.have_tma">have_tma() (in module tilelang.contrib.nvcc)</a>
 </li>
         <li><a href="autoapi/tilelang/carver/template/flashattention/index.html#tilelang.carver.template.flashattention.FlashAttentionTemplate.head_dim">head_dim (tilelang.carver.template.flashattention.FlashAttentionTemplate attribute)</a>
@@ -2324,7 +2324,11 @@ <h2>H</h2>
         <li><a href="autoapi/tilelang/engine/lower/index.html#tilelang.engine.lower.host_codegen">host_codegen() (in module tilelang.engine.lower)</a>
 </li>
         <li><a href="autoapi/tilelang/jit/adapter/libgen/index.html#tilelang.jit.adapter.libgen.PyLibraryGenerator.host_func">host_func (tilelang.jit.adapter.libgen.PyLibraryGenerator attribute)</a>
+
+        <ul>
+          <li><a href="autoapi/tilelang/jit/adapter/wrapper/index.html#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.host_func">(tilelang.jit.adapter.wrapper.TLCUDASourceWrapper property)</a>
 </li>
+        </ul></li>
         <li><a href="autoapi/tilelang/engine/param/index.html#tilelang.engine.param.CompiledArtifact.host_mod">host_mod (tilelang.engine.param.CompiledArtifact attribute)</a>
 
         <ul>