Skip to content

Commit b158b61

Browse files
Update docs
1 parent 9ea2b93 commit b158b61

File tree

185 files changed

+882
-9
lines changed

Some content is hidden

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

185 files changed

+882
-9
lines changed
Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
tilelang.analysis.fragment_loop_checker
2+
=======================================
3+
4+
.. py:module:: tilelang.analysis.fragment_loop_checker
5+
6+
7+
Functions
8+
---------
9+
10+
.. autoapisummary::
11+
12+
tilelang.analysis.fragment_loop_checker.collect_local_buffer_accesses
13+
tilelang.analysis.fragment_loop_checker.FragmentLoopChecker
14+
15+
16+
Module Contents
17+
---------------
18+
19+
.. py:function:: collect_local_buffer_accesses(statement)
20+
21+
Collect local buffer accesses in the loop body.
22+
23+
:param statement: The TIR statement to analyze
24+
25+
:returns: Tuple of buffer accesses in the loop body.
26+
27+
28+
.. py:function:: FragmentLoopChecker()
29+
30+
When using T.Parallel over a local/fragment buffer, there are several restrictions:
31+
to ensure that the parallelization is valid.
32+
33+
1. The range of loop can not be symbolic.
34+
35+
:returns: A prim_func_pass that applies the transformation
36+
37+

_sources/autoapi/tilelang/analysis/index.rst.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ Submodules
1616
:maxdepth: 1
1717

1818
/autoapi/tilelang/analysis/ast_printer/index
19+
/autoapi/tilelang/analysis/fragment_loop_checker/index
1920
/autoapi/tilelang/analysis/nested_loop_checker/index
2021

2122

autoapi/index.html

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,7 @@
226226
<li class="toctree-l1 has-children"><a class="reference internal" href="tilelang/index.html">tilelang</a><input aria-label="Toggle navigation of tilelang" class="toctree-checkbox" id="toctree-checkbox-1" name="toctree-checkbox-1" role="switch" type="checkbox"/><label for="toctree-checkbox-1"><span class="icon"><svg><use href="#svg-arrow-right"></use></svg></span></label><ul>
227227
<li class="toctree-l2 has-children"><a class="reference internal" href="tilelang/analysis/index.html">tilelang.analysis</a><input aria-label="Toggle navigation of tilelang.analysis" class="toctree-checkbox" id="toctree-checkbox-2" name="toctree-checkbox-2" role="switch" type="checkbox"/><label for="toctree-checkbox-2"><span class="icon"><svg><use href="#svg-arrow-right"></use></svg></span></label><ul>
228228
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
229+
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
229230
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
230231
</ul>
231232
</li>
@@ -506,6 +507,7 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Link to this
506507
<li class="toctree-l1"><a class="reference internal" href="tilelang/index.html">tilelang</a><ul>
507508
<li class="toctree-l2"><a class="reference internal" href="tilelang/analysis/index.html">tilelang.analysis</a><ul>
508509
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
510+
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
509511
<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
510512
</ul>
511513
</li>

autoapi/tilelang/analysis/ast_printer/index.html

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
<head><meta charset="utf-8">
44
<meta name="viewport" content="width=device-width,initial-scale=1">
55
<meta name="color-scheme" content="light dark"><meta name="viewport" content="width=device-width, initial-scale=1" />
6-
<link rel="index" title="Index" href="../../../../genindex.html"><link rel="search" title="Search" href="../../../../search.html"><link rel="next" title="tilelang.analysis.nested_loop_checker" href="../nested_loop_checker/index.html"><link rel="prev" title="tilelang.analysis" href="../index.html">
6+
<link rel="index" title="Index" href="../../../../genindex.html"><link rel="search" title="Search" href="../../../../search.html"><link rel="next" title="tilelang.analysis.fragment_loop_checker" href="../fragment_loop_checker/index.html"><link rel="prev" title="tilelang.analysis" href="../index.html">
77
<link rel="prefetch" href="../../../../_static/img/logo-row.svg" as="image">
88
<link rel="prefetch" href="../../../../_static/img/logo-row.svg" as="image">
99

@@ -226,6 +226,7 @@
226226
<li class="toctree-l1 current has-children"><a class="reference internal" href="../../index.html">tilelang</a><input aria-label="Toggle navigation of tilelang" checked="" class="toctree-checkbox" id="toctree-checkbox-1" name="toctree-checkbox-1" role="switch" type="checkbox"/><label for="toctree-checkbox-1"><span class="icon"><svg><use href="#svg-arrow-right"></use></svg></span></label><ul class="current">
227227
<li class="toctree-l2 current has-children"><a class="reference internal" href="../index.html">tilelang.analysis</a><input aria-label="Toggle navigation of tilelang.analysis" checked="" class="toctree-checkbox" id="toctree-checkbox-2" name="toctree-checkbox-2" role="switch" type="checkbox"/><label for="toctree-checkbox-2"><span class="icon"><svg><use href="#svg-arrow-right"></use></svg></span></label><ul class="current">
228228
<li class="toctree-l3 current current-page"><a class="current reference internal" href="#">tilelang.analysis.ast_printer</a></li>
229+
<li class="toctree-l3"><a class="reference internal" href="../fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
229230
<li class="toctree-l3"><a class="reference internal" href="../nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
230231
</ul>
231232
</li>
@@ -528,12 +529,12 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
528529
<footer>
529530

530531
<div class="related-pages">
531-
<a class="next-page" href="../nested_loop_checker/index.html">
532+
<a class="next-page" href="../fragment_loop_checker/index.html">
532533
<div class="page-info">
533534
<div class="context">
534535
<span>Next</span>
535536
</div>
536-
<div class="title">tilelang.analysis.nested_loop_checker</div>
537+
<div class="title">tilelang.analysis.fragment_loop_checker</div>
537538
</div>
538539
<svg class="furo-related-icon"><use href="#svg-arrow-right"></use></svg>
539540
</a>

0 commit comments

Comments
 (0)