Skip to content

Commit 18cca05

Browse files
Update docs
1 parent d54af02 commit 18cca05

File tree

237 files changed

+1023
-11
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

237 files changed

+1023
-11
lines changed
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
tilelang.tileop.gemm.gemm_scalar
2+
================================
3+
4+
.. py:module:: tilelang.tileop.gemm.gemm_scalar
5+
6+
7+
Classes
8+
-------
9+
10+
.. autoapisummary::
11+
12+
tilelang.tileop.gemm.gemm_scalar.GemmScalar
13+
14+
15+
Module Contents
16+
---------------
17+
18+
.. py:class:: GemmScalar
19+
20+
Bases: :py:obj:`tilelang.tileop.gemm.gemm_base.GemmBase`
21+
22+
23+
CPU scalar fallback: triple nested loop gemm.
24+
25+
26+
.. py:method:: infer_layout(target, thread_nums)
27+
28+
29+
.. py:method:: lower(layout_map, target, thread_bounds, thread_var)
30+
31+

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ Submodules
1414
/autoapi/tilelang/tileop/gemm/gemm_mfma/index
1515
/autoapi/tilelang/tileop/gemm/gemm_mma/index
1616
/autoapi/tilelang/tileop/gemm/gemm_mma_sm70/index
17+
/autoapi/tilelang/tileop/gemm/gemm_scalar/index
1718
/autoapi/tilelang/tileop/gemm/gemm_tcgen05/index
1819
/autoapi/tilelang/tileop/gemm/gemm_wgmma/index
1920
/autoapi/tilelang/tileop/gemm/inst/index

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@ Module Contents
4343

4444

4545

46+
.. py:attribute:: Scalar
47+
:value: 4
48+
49+
50+
4651
.. py:method:: is_mma()
4752
4853
@@ -55,6 +60,9 @@ Module Contents
5560
.. py:method:: is_mfma()
5661
5762
63+
.. py:method:: is_scalar()
64+
65+
5866
.. py:method:: __repr__()
5967
6068
Return repr(self).

autoapi/index.html

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>
@@ -798,6 +799,7 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Link to this
798799
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
799800
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
800801
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
802+
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
801803
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
802804
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
803805
<li class="toctree-l4"><a class="reference internal" href="tilelang/tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/analysis/ast_printer/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/analysis/fragment_loop_checker/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/analysis/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/analysis/layout_visual/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/analysis/nested_loop_checker/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

autoapi/tilelang/autodd/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,7 @@
474474
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mfma/index.html">tilelang.tileop.gemm.gemm_mfma</a></li>
475475
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mma/index.html">tilelang.tileop.gemm.gemm_mma</a></li>
476476
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_mma_sm70/index.html">tilelang.tileop.gemm.gemm_mma_sm70</a></li>
477+
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_scalar/index.html">tilelang.tileop.gemm.gemm_scalar</a></li>
477478
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_tcgen05/index.html">tilelang.tileop.gemm.gemm_tcgen05</a></li>
478479
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/gemm_wgmma/index.html">tilelang.tileop.gemm.gemm_wgmma</a></li>
479480
<li class="toctree-l4"><a class="reference internal" href="../tileop/gemm/inst/index.html">tilelang.tileop.gemm.inst</a></li>

0 commit comments

Comments
 (0)