Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/contrib/nvcc/index.rst.txt

@@ -20,7 +20,6 @@ Functions
    tilelang.contrib.nvcc.get_sass_from_source
    tilelang.contrib.nvcc.find_cuda_path
    tilelang.contrib.nvcc.get_cuda_version
-   tilelang.contrib.nvcc.tilelang_callback_cuda_compile
    tilelang.contrib.nvcc.find_libdevice_path
    tilelang.contrib.nvcc.callback_libdevice_path
    tilelang.contrib.nvcc.get_target_compute_version
@@ -125,11 +124,6 @@ Module Contents
    :rtype: float
 
 
-.. py:function:: tilelang_callback_cuda_compile(code, target)
-
-   use nvcc to generate fatbin code for better optimization
-
-
 .. py:function:: find_libdevice_path(arch)
 
    Utility function to find libdevice

_sources/autoapi/tilelang/engine/lower/index.rst.txt

@@ -48,7 +48,7 @@ Module Contents
 
 .. py:function:: get_host_call(is_device_c = False)
 
-.. py:function:: tilelang_callback_cuda_compile(code, target)
+.. py:function:: tilelang_callback_cuda_compile(code, target, pass_config=None)
 
 .. py:function:: tilelang_callback_hip_compile(code, target)
 

autoapi/tilelang/contrib/nvcc/index.html

@@ -519,49 +519,46 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_cuda_version" title="tilelang.contrib.nvcc.get_cuda_version"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_cuda_version</span></code></a>([cuda_path])</p></td>
 <td><p>Utility function to get cuda version</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.tilelang_callback_cuda_compile" title="tilelang.contrib.nvcc.tilelang_callback_cuda_compile"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang_callback_cuda_compile</span></code></a>(code, target)</p></td>
-<td><p>use nvcc to generate fatbin code for better optimization</p></td>
-</tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.find_libdevice_path" title="tilelang.contrib.nvcc.find_libdevice_path"><code class="xref py py-obj docutils literal notranslate"><span class="pre">find_libdevice_path</span></code></a>(arch)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.find_libdevice_path" title="tilelang.contrib.nvcc.find_libdevice_path"><code class="xref py py-obj docutils literal notranslate"><span class="pre">find_libdevice_path</span></code></a>(arch)</p></td>
 <td><p>Utility function to find libdevice</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.callback_libdevice_path" title="tilelang.contrib.nvcc.callback_libdevice_path"><code class="xref py py-obj docutils literal notranslate"><span class="pre">callback_libdevice_path</span></code></a>(arch)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.callback_libdevice_path" title="tilelang.contrib.nvcc.callback_libdevice_path"><code class="xref py py-obj docutils literal notranslate"><span class="pre">callback_libdevice_path</span></code></a>(arch)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_compute_version" title="tilelang.contrib.nvcc.get_target_compute_version"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_target_compute_version</span></code></a>([target])</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_compute_version" title="tilelang.contrib.nvcc.get_target_compute_version"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_target_compute_version</span></code></a>([target])</p></td>
 <td><p>Utility function to get compute capability of compilation target.</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.parse_compute_version" title="tilelang.contrib.nvcc.parse_compute_version"><code class="xref py py-obj docutils literal notranslate"><span class="pre">parse_compute_version</span></code></a>(compute_version)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.parse_compute_version" title="tilelang.contrib.nvcc.parse_compute_version"><code class="xref py py-obj docutils literal notranslate"><span class="pre">parse_compute_version</span></code></a>(compute_version)</p></td>
 <td><p>Parse compute capability string to divide major and minor version</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_arch" title="tilelang.contrib.nvcc.get_target_arch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_target_arch</span></code></a>(compute_version)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_arch" title="tilelang.contrib.nvcc.get_target_arch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_target_arch</span></code></a>(compute_version)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp16" title="tilelang.contrib.nvcc.have_fp16"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_fp16</span></code></a>(compute_version)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp16" title="tilelang.contrib.nvcc.have_fp16"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_fp16</span></code></a>(compute_version)</p></td>
 <td><p>Either fp16 support is provided in the compute capability or not</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_int8" title="tilelang.contrib.nvcc.have_int8"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_int8</span></code></a>(compute_version)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_int8" title="tilelang.contrib.nvcc.have_int8"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_int8</span></code></a>(compute_version)</p></td>
 <td><p>Either int8 support is provided in the compute capability or not</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_tensorcore" title="tilelang.contrib.nvcc.have_tensorcore"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_tensorcore</span></code></a>([compute_version, target])</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_tensorcore" title="tilelang.contrib.nvcc.have_tensorcore"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_tensorcore</span></code></a>([compute_version, target])</p></td>
 <td><p>Either TensorCore support is provided in the compute capability or not</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_cudagraph" title="tilelang.contrib.nvcc.have_cudagraph"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_cudagraph</span></code></a>()</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_cudagraph" title="tilelang.contrib.nvcc.have_cudagraph"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_cudagraph</span></code></a>()</p></td>
 <td><p>Either CUDA Graph support is provided</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_bf16" title="tilelang.contrib.nvcc.have_bf16"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_bf16</span></code></a>(compute_version)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_bf16" title="tilelang.contrib.nvcc.have_bf16"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_bf16</span></code></a>(compute_version)</p></td>
 <td><p>Either bf16 support is provided in the compute capability or not</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp8" title="tilelang.contrib.nvcc.have_fp8"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_fp8</span></code></a>(compute_version)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp8" title="tilelang.contrib.nvcc.have_fp8"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_fp8</span></code></a>(compute_version)</p></td>
 <td><p>Whether fp8 support is provided in the specified compute capability or not</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_tma" title="tilelang.contrib.nvcc.have_tma"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_tma</span></code></a>(target)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.have_tma" title="tilelang.contrib.nvcc.have_tma"><code class="xref py py-obj docutils literal notranslate"><span class="pre">have_tma</span></code></a>(target)</p></td>
 <td><p>Whether TMA support is provided in the specified compute capability or not</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.is_hopper" title="tilelang.contrib.nvcc.is_hopper"><code class="xref py py-obj docutils literal notranslate"><span class="pre">is_hopper</span></code></a>(target)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.is_hopper" title="tilelang.contrib.nvcc.is_hopper"><code class="xref py py-obj docutils literal notranslate"><span class="pre">is_hopper</span></code></a>(target)</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_nvcc_compiler" title="tilelang.contrib.nvcc.get_nvcc_compiler"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_nvcc_compiler</span></code></a>()</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.contrib.nvcc.get_nvcc_compiler" title="tilelang.contrib.nvcc.get_nvcc_compiler"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_nvcc_compiler</span></code></a>()</p></td>
 <td><p>Get the path to the nvcc compiler</p></td>
 </tr>
 </tbody>
@@ -687,12 +684,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 </dl>
 </dd></dl>
 
-<dl class="py function">
-<dt class="sig sig-object py" id="tilelang.contrib.nvcc.tilelang_callback_cuda_compile">
-<span class="sig-prename descclassname"><span class="pre">tilelang.contrib.nvcc.</span></span><span class="sig-name descname"><span class="pre">tilelang_callback_cuda_compile</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">code</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.contrib.nvcc.tilelang_callback_cuda_compile" title="Link to this definition">¶</a></dt>
-<dd><p>use nvcc to generate fatbin code for better optimization</p>
-</dd></dl>
-
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.contrib.nvcc.find_libdevice_path">
 <span class="sig-prename descclassname"><span class="pre">tilelang.contrib.nvcc.</span></span><span class="sig-name descname"><span class="pre">find_libdevice_path</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">arch</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.contrib.nvcc.find_libdevice_path" title="Link to this definition">¶</a></dt>
@@ -924,7 +915,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.get_sass_from_source"><code class="docutils literal notranslate"><span class="pre">get_sass_from_source()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.find_cuda_path"><code class="docutils literal notranslate"><span class="pre">find_cuda_path()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.get_cuda_version"><code class="docutils literal notranslate"><span class="pre">get_cuda_version()</span></code></a></li>
-<li><a class="reference internal" href="#tilelang.contrib.nvcc.tilelang_callback_cuda_compile"><code class="docutils literal notranslate"><span class="pre">tilelang_callback_cuda_compile()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.find_libdevice_path"><code class="docutils literal notranslate"><span class="pre">find_libdevice_path()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.callback_libdevice_path"><code class="docutils literal notranslate"><span class="pre">callback_libdevice_path()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_compute_version"><code class="docutils literal notranslate"><span class="pre">get_target_compute_version()</span></code></a></li>

autoapi/tilelang/engine/lower/index.html

@@ -519,7 +519,7 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <tr class="row-even"><td><p><a class="reference internal" href="#tilelang.engine.lower.get_host_call" title="tilelang.engine.lower.get_host_call"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_host_call</span></code></a>([is_device_c])</p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.engine.lower.tilelang_callback_cuda_compile" title="tilelang.engine.lower.tilelang_callback_cuda_compile"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang_callback_cuda_compile</span></code></a>(code, target)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.engine.lower.tilelang_callback_cuda_compile" title="tilelang.engine.lower.tilelang_callback_cuda_compile"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang_callback_cuda_compile</span></code></a>(code, target[, pass_config])</p></td>
 <td><p></p></td>
 </tr>
 <tr class="row-even"><td><p><a class="reference internal" href="#tilelang.engine.lower.tilelang_callback_hip_compile" title="tilelang.engine.lower.tilelang_callback_hip_compile"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang_callback_hip_compile</span></code></a>(code, target)</p></td>
@@ -618,7 +618,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.engine.lower.tilelang_callback_cuda_compile">
-<span class="sig-prename descclassname"><span class="pre">tilelang.engine.lower.</span></span><span class="sig-name descname"><span class="pre">tilelang_callback_cuda_compile</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">code</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.engine.lower.tilelang_callback_cuda_compile" title="Link to this definition">¶</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.engine.lower.</span></span><span class="sig-name descname"><span class="pre">tilelang_callback_cuda_compile</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">code</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">pass_config</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.engine.lower.tilelang_callback_cuda_compile" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
 <dl class="py function">

genindex.html

@@ -6399,15 +6399,15 @@ <h2>T</h2>
           <li><a href="autoapi/tilelang/language/logical/index.html#module-tilelang.language.logical">module</a>
 </li>
         </ul></li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     tilelang.language.loop
 
         <ul>
           <li><a href="autoapi/tilelang/language/loop/index.html#module-tilelang.language.loop">module</a>
 </li>
         </ul></li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     tilelang.language.math_intrinsics
 
@@ -6815,12 +6815,8 @@ <h2>T</h2>
 </li>
         <li><a href="autoapi/tilelang/env/index.html#tilelang.env.Environment.TILELANG_CACHE_DIR">TILELANG_CACHE_DIR (tilelang.env.Environment attribute)</a>
 </li>
-        <li><a href="autoapi/tilelang/contrib/nvcc/index.html#tilelang.contrib.nvcc.tilelang_callback_cuda_compile">tilelang_callback_cuda_compile() (in module tilelang.contrib.nvcc)</a>
-
-        <ul>
-          <li><a href="autoapi/tilelang/engine/lower/index.html#tilelang.engine.lower.tilelang_callback_cuda_compile">(in module tilelang.engine.lower)</a>
+        <li><a href="autoapi/tilelang/engine/lower/index.html#tilelang.engine.lower.tilelang_callback_cuda_compile">tilelang_callback_cuda_compile() (in module tilelang.engine.lower)</a>
 </li>
-        </ul></li>
         <li><a href="autoapi/tilelang/contrib/hipcc/index.html#tilelang.contrib.hipcc.tilelang_callback_hip_compile">tilelang_callback_hip_compile() (in module tilelang.contrib.hipcc)</a>
 
         <ul>