Skip to content

Commit ad42409

Browse files
Update docs
1 parent 18cca05 commit ad42409

5 files changed

Lines changed: 30 additions & 2 deletions

File tree

_sources/autoapi/tilelang/tileop/gemm/gemm_tcgen05/index.rst.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,12 @@ Module Contents
2727
of operands A and B.
2828

2929

30+
.. py:method:: infer_shared_layout(continuity, k_major)
31+
32+
Build a shared-memory layout constructor for TCGEN05 operands.
33+
34+
35+
3036
.. py:method:: infer_layout(target, thread_nums)
3137
3238
Infer swizzled layouts for operands and accumulator.

autoapi/tilelang/tileop/gemm/gemm_tcgen05/index.html

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -581,6 +581,23 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
581581
<p>Supports the SS (Shared-Shared) and TS (TensorMemory-Shared) variants.
582582
Layout inference and lowering are dispatched based on the memory scopes
583583
of operands A and B.</p>
584+
<dl class="py method">
585+
<dt class="sig sig-object py" id="tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_shared_layout">
586+
<span class="sig-name descname"><span class="pre">infer_shared_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">continuity</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">k_major</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_shared_layout" title="Link to this definition"></a></dt>
587+
<dd><p>Build a shared-memory layout constructor for TCGEN05 operands.</p>
588+
<dl class="field-list simple">
589+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
590+
<dd class="field-odd"><ul class="simple">
591+
<li><p><strong>continuity</strong> (<em>int</em>)</p></li>
592+
<li><p><strong>k_major</strong> (<a class="reference internal" href="../../../language/dtypes/index.html#tilelang.language.dtypes.bool" title="tilelang.language.dtypes.bool"><em>bool</em></a>)</p></li>
593+
</ul>
594+
</dd>
595+
<dt class="field-even">Return type<span class="colon">:</span></dt>
596+
<dd class="field-even"><p>Callable[[tvm.tir.Buffer], tilelang.layout.Layout]</p>
597+
</dd>
598+
</dl>
599+
</dd></dl>
600+
584601
<dl class="py method">
585602
<dt class="sig sig-object py" id="tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_layout">
586603
<span class="sig-name descname"><span class="pre">infer_layout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">thread_nums</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_layout" title="Link to this definition"></a></dt>
@@ -677,6 +694,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
677694
<li><a class="reference internal" href="#classes">Classes</a></li>
678695
<li><a class="reference internal" href="#module-contents">Module Contents</a><ul>
679696
<li><a class="reference internal" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5"><code class="docutils literal notranslate"><span class="pre">GemmTCGEN5</span></code></a><ul>
697+
<li><a class="reference internal" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_shared_layout"><code class="docutils literal notranslate"><span class="pre">GemmTCGEN5.infer_shared_layout()</span></code></a></li>
680698
<li><a class="reference internal" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_layout"><code class="docutils literal notranslate"><span class="pre">GemmTCGEN5.infer_layout()</span></code></a></li>
681699
<li><a class="reference internal" href="#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.lower"><code class="docutils literal notranslate"><span class="pre">GemmTCGEN5.lower()</span></code></a></li>
682700
</ul>

genindex.html

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3606,8 +3606,12 @@ <h2>I</h2>
36063606
<li><a href="autoapi/tilelang/carver/roller/policy/tensorcore/index.html#tilelang.carver.roller.policy.tensorcore.TensorCorePolicy.infer_node_smem_usage">(tilelang.carver.roller.policy.tensorcore.TensorCorePolicy method)</a>
36073607
</li>
36083608
</ul></li>
3609-
<li><a href="autoapi/tilelang/tileop/gemm/gemm_wgmma/index.html#tilelang.tileop.gemm.gemm_wgmma.GemmWGMMA.infer_shared_layout">infer_shared_layout() (tilelang.tileop.gemm.gemm_wgmma.GemmWGMMA method)</a>
3609+
<li><a href="autoapi/tilelang/tileop/gemm/gemm_tcgen05/index.html#tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5.infer_shared_layout">infer_shared_layout() (tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5 method)</a>
3610+
3611+
<ul>
3612+
<li><a href="autoapi/tilelang/tileop/gemm/gemm_wgmma/index.html#tilelang.tileop.gemm.gemm_wgmma.GemmWGMMA.infer_shared_layout">(tilelang.tileop.gemm.gemm_wgmma.GemmWGMMA method)</a>
36103613
</li>
3614+
</ul></li>
36113615
<li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.PrimFuncNode.infer_tensorcore_axis">infer_tensorcore_axis() (tilelang.carver.roller.node.PrimFuncNode method)</a>
36123616
</li>
36133617
<li><a href="autoapi/tilelang/language/tir/ir/index.html#tilelang.language.tir.ir.infinity">infinity (in module tilelang.language.tir.ir)</a>

objects.inv

6 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)