Update docs

Author: LeiWang1999

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

@@ -22,6 +22,7 @@ Functions
    tilelang.contrib.nvcc.callback_libdevice_path
    tilelang.contrib.nvcc.get_target_compute_version
    tilelang.contrib.nvcc.parse_compute_version
+   tilelang.contrib.nvcc.get_target_arch
    tilelang.contrib.nvcc.have_fp16
    tilelang.contrib.nvcc.have_int8
    tilelang.contrib.nvcc.have_tensorcore
@@ -117,6 +118,8 @@ Module Contents
              * **minor** (*int*) -- minor version number
 
 
+.. py:function:: get_target_arch(compute_version)
+
 .. py:function:: have_fp16(compute_version)
 
    Either fp16 support is provided in the compute capability or not

autoapi/tilelang/contrib/nvcc/index.html

@@ -489,28 +489,31 @@ <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.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-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-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-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.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-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>
 <td><p>Get the path to the nvcc compiler</p></td>
 </tr>
 </tbody>
@@ -636,6 +639,19 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 </ul>
 </p>
 </dd>
+<dt class="field-odd">Return type<span class="colon">:</span></dt>
+<dd class="field-odd"><p>tuple[int, int]</p>
+</dd>
+</dl>
+</dd></dl>
+
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.contrib.nvcc.get_target_arch">
+<span class="sig-prename descclassname"><span class="pre">tilelang.contrib.nvcc.</span></span><span class="sig-name descname"><span class="pre">get_target_arch</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">compute_version</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.contrib.nvcc.get_target_arch" title="Link to this definition">¶</a></dt>
+<dd><dl class="field-list simple">
+<dt class="field-odd">Return type<span class="colon">:</span></dt>
+<dd class="field-odd"><p>str</p>
+</dd>
 </dl>
 </dd></dl>
 
@@ -795,6 +811,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <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>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.parse_compute_version"><code class="docutils literal notranslate"><span class="pre">parse_compute_version()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.contrib.nvcc.get_target_arch"><code class="docutils literal notranslate"><span class="pre">get_target_arch()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp16"><code class="docutils literal notranslate"><span class="pre">have_fp16()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.have_int8"><code class="docutils literal notranslate"><span class="pre">have_int8()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.contrib.nvcc.have_tensorcore"><code class="docutils literal notranslate"><span class="pre">have_tensorcore()</span></code></a></li>

genindex.html

@@ -2043,6 +2043,8 @@ <h2>G</h2>
         <li><a href="autoapi/tilelang/intrinsics/mma_layout/index.html#tilelang.intrinsics.mma_layout.get_swizzle_layout">get_swizzle_layout() (in module tilelang.intrinsics.mma_layout)</a>
 </li>
         <li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.Node.get_tag">get_tag() (tilelang.carver.roller.node.Node method)</a>
+</li>
+        <li><a href="autoapi/tilelang/contrib/nvcc/index.html#tilelang.contrib.nvcc.get_target_arch">get_target_arch() (in module tilelang.contrib.nvcc)</a>
 </li>
         <li><a href="autoapi/tilelang/contrib/cc/index.html#tilelang.contrib.cc.get_target_by_dump_machine">get_target_by_dump_machine() (in module tilelang.contrib.cc)</a>
 </li>