Skip to content

Commit 21ab6bd

Browse files
Update docs
1 parent 824b895 commit 21ab6bd

File tree

7 files changed

+49
-12
lines changed

7 files changed

+49
-12
lines changed

_sources/autoapi/tilelang/intrinsics/mfma_macro_generator/index.rst.txt

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ Module Contents
2626

2727
.. py:data:: lift
2828
29-
.. py:class:: MatrixCoreIntrinEmitter(a_dtype = T.float16, b_dtype = T.float16, accum_dtype = T.float16, a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, b_preshuffle = False, thread_var = None)
29+
.. py:class:: MatrixCoreIntrinEmitter(a_dtype = T.float16, b_dtype = T.float16, accum_dtype = T.float16, a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, b_preshuffle = False, thread_var = None, target = None)
3030
3131
To eliminate Python syntax within TIR Macro.
3232

@@ -78,6 +78,11 @@ Module Contents
7878

7979

8080

81+
.. py:attribute:: target
82+
:value: None
83+
84+
85+
8186
.. py:attribute:: block_row_warps
8287
:value: 2
8388

@@ -192,7 +197,7 @@ Module Contents
192197

193198

194199

195-
.. py:class:: MatrixCorePreshuffleIntrinEmitter(a_dtype = T.float16, b_dtype = T.float16, accum_dtype = T.float16, a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, a_preshuffle = False, b_preshuffle = False, thread_var = None)
200+
.. py:class:: MatrixCorePreshuffleIntrinEmitter(a_dtype = T.float16, b_dtype = T.float16, accum_dtype = T.float16, a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, a_preshuffle = False, b_preshuffle = False, thread_var = None, target = None)
196201
197202
Bases: :py:obj:`MatrixCoreIntrinEmitter`
198203

_sources/autoapi/tilelang/utils/target/index.rst.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ Functions
3434
tilelang.utils.target.target_is_hopper
3535
tilelang.utils.target.target_is_sm120
3636
tilelang.utils.target.target_is_cdna
37+
tilelang.utils.target.target_is_gfx950
3738
tilelang.utils.target.target_has_async_copy
3839
tilelang.utils.target.target_has_ldmatrix
3940
tilelang.utils.target.target_has_stmatrix
@@ -113,6 +114,8 @@ Module Contents
113114
114115
.. py:function:: target_is_cdna(target)
115116
117+
.. py:function:: target_is_gfx950(target)
118+
116119
.. py:function:: target_has_async_copy(target)
117120
118121
.. py:function:: target_has_ldmatrix(target)

autoapi/tilelang/intrinsics/mfma_macro_generator/index.html

Lines changed: 10 additions & 2 deletions
Large diffs are not rendered by default.

autoapi/tilelang/utils/target/index.html

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -625,19 +625,22 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
625625
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_is_cdna" title="tilelang.utils.target.target_is_cdna"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_is_cdna</span></code></a>(target)</p></td>
626626
<td><p></p></td>
627627
</tr>
628-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_async_copy" title="tilelang.utils.target.target_has_async_copy"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_async_copy</span></code></a>(target)</p></td>
628+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_is_gfx950" title="tilelang.utils.target.target_is_gfx950"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_is_gfx950</span></code></a>(target)</p></td>
629629
<td><p></p></td>
630630
</tr>
631-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_ldmatrix" title="tilelang.utils.target.target_has_ldmatrix"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_ldmatrix</span></code></a>(target)</p></td>
631+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_async_copy" title="tilelang.utils.target.target_has_async_copy"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_async_copy</span></code></a>(target)</p></td>
632632
<td><p></p></td>
633633
</tr>
634-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_stmatrix" title="tilelang.utils.target.target_has_stmatrix"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_stmatrix</span></code></a>(target)</p></td>
634+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_ldmatrix" title="tilelang.utils.target.target_has_ldmatrix"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_ldmatrix</span></code></a>(target)</p></td>
635635
<td><p></p></td>
636636
</tr>
637-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_bulk_copy" title="tilelang.utils.target.target_has_bulk_copy"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_bulk_copy</span></code></a>(target)</p></td>
637+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_stmatrix" title="tilelang.utils.target.target_has_stmatrix"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_stmatrix</span></code></a>(target)</p></td>
638638
<td><p></p></td>
639639
</tr>
640-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_get_warp_size" title="tilelang.utils.target.target_get_warp_size"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_get_warp_size</span></code></a>(target)</p></td>
640+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.utils.target.target_has_bulk_copy" title="tilelang.utils.target.target_has_bulk_copy"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_has_bulk_copy</span></code></a>(target)</p></td>
641+
<td><p></p></td>
642+
</tr>
643+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.utils.target.target_get_warp_size" title="tilelang.utils.target.target_get_warp_size"><code class="xref py py-obj docutils literal notranslate"><span class="pre">target_get_warp_size</span></code></a>(target)</p></td>
641644
<td><p></p></td>
642645
</tr>
643646
</tbody>
@@ -885,6 +888,19 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
885888
</dl>
886889
</dd></dl>
887890

891+
<dl class="py function">
892+
<dt class="sig sig-object py" id="tilelang.utils.target.target_is_gfx950">
893+
<span class="sig-prename descclassname"><span class="pre">tilelang.utils.target.</span></span><span class="sig-name descname"><span class="pre">target_is_gfx950</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.utils.target.target_is_gfx950" title="Link to this definition"></a></dt>
894+
<dd><dl class="field-list simple">
895+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
896+
<dd class="field-odd"><p><strong>target</strong> (<em>tvm.target.Target</em>)</p>
897+
</dd>
898+
<dt class="field-even">Return type<span class="colon">:</span></dt>
899+
<dd class="field-even"><p><a class="reference internal" href="../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool">bool</a></p>
900+
</dd>
901+
</dl>
902+
</dd></dl>
903+
888904
<dl class="py function">
889905
<dt class="sig sig-object py" id="tilelang.utils.target.target_has_async_copy">
890906
<span class="sig-prename descclassname"><span class="pre">tilelang.utils.target.</span></span><span class="sig-name descname"><span class="pre">target_has_async_copy</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.utils.target.target_has_async_copy" title="Link to this definition"></a></dt>
@@ -1030,6 +1046,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
10301046
<li><a class="reference internal" href="#tilelang.utils.target.target_is_hopper"><code class="docutils literal notranslate"><span class="pre">target_is_hopper()</span></code></a></li>
10311047
<li><a class="reference internal" href="#tilelang.utils.target.target_is_sm120"><code class="docutils literal notranslate"><span class="pre">target_is_sm120()</span></code></a></li>
10321048
<li><a class="reference internal" href="#tilelang.utils.target.target_is_cdna"><code class="docutils literal notranslate"><span class="pre">target_is_cdna()</span></code></a></li>
1049+
<li><a class="reference internal" href="#tilelang.utils.target.target_is_gfx950"><code class="docutils literal notranslate"><span class="pre">target_is_gfx950()</span></code></a></li>
10331050
<li><a class="reference internal" href="#tilelang.utils.target.target_has_async_copy"><code class="docutils literal notranslate"><span class="pre">target_has_async_copy()</span></code></a></li>
10341051
<li><a class="reference internal" href="#tilelang.utils.target.target_has_ldmatrix"><code class="docutils literal notranslate"><span class="pre">target_has_ldmatrix()</span></code></a></li>
10351052
<li><a class="reference internal" href="#tilelang.utils.target.target_has_stmatrix"><code class="docutils literal notranslate"><span class="pre">target_has_stmatrix()</span></code></a></li>

genindex.html

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7005,6 +7005,8 @@ <h2>T</h2>
70057005
<li><a href="autoapi/tilelang/carver/arch/cuda/index.html#tilelang.carver.arch.cuda.CUDA.target">(tilelang.carver.arch.cuda.CUDA attribute)</a>
70067006
</li>
70077007
<li><a href="autoapi/tilelang/carver/arch/metal/index.html#tilelang.carver.arch.metal.METAL.target">(tilelang.carver.arch.metal.METAL attribute)</a>
7008+
</li>
7009+
<li><a href="autoapi/tilelang/intrinsics/mfma_macro_generator/index.html#tilelang.intrinsics.mfma_macro_generator.MatrixCoreIntrinEmitter.target">(tilelang.intrinsics.mfma_macro_generator.MatrixCoreIntrinEmitter attribute)</a>
70087010
</li>
70097011
<li><a href="autoapi/tilelang/jit/adapter/cutedsl/adapter/index.html#tilelang.jit.adapter.cutedsl.adapter.CuTeDSLKernelAdapter.target">(tilelang.jit.adapter.cutedsl.adapter.CuTeDSLKernelAdapter attribute)</a>
70107012
</li>
@@ -7054,6 +7056,8 @@ <h2>T</h2>
70547056
<li><a href="autoapi/tilelang/utils/target/index.html#tilelang.utils.target.target_is_cdna">target_is_cdna() (in module tilelang.utils.target)</a>
70557057
</li>
70567058
<li><a href="autoapi/tilelang/utils/target/index.html#tilelang.utils.target.target_is_cuda">target_is_cuda() (in module tilelang.utils.target)</a>
7059+
</li>
7060+
<li><a href="autoapi/tilelang/utils/target/index.html#tilelang.utils.target.target_is_gfx950">target_is_gfx950() (in module tilelang.utils.target)</a>
70577061
</li>
70587062
<li><a href="autoapi/tilelang/utils/target/index.html#tilelang.utils.target.target_is_hip">target_is_hip() (in module tilelang.utils.target)</a>
70597063
</li>
@@ -8111,15 +8115,15 @@ <h2>T</h2>
81118115
<li><a href="autoapi/tilelang/language/annotations/index.html#module-tilelang.language.annotations">module</a>
81128116
</li>
81138117
</ul></li>
8118+
</ul></td>
8119+
<td style="width: 33%; vertical-align: top;"><ul>
81148120
<li>
81158121
tilelang.language.atomic
81168122

81178123
<ul>
81188124
<li><a href="autoapi/tilelang/language/atomic/index.html#module-tilelang.language.atomic">module</a>
81198125
</li>
81208126
</ul></li>
8121-
</ul></td>
8122-
<td style="width: 33%; vertical-align: top;"><ul>
81238127
<li>
81248128
tilelang.language.builtin
81258129

objects.inv

13 Bytes
Binary file not shown.

searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)