Update docs

Author: github-actions[bot]

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

@@ -17,6 +17,7 @@ Submodules
 
    /autoapi/tilelang/analysis/ast_printer/index
    /autoapi/tilelang/analysis/fragment_loop_checker/index
+   /autoapi/tilelang/analysis/layout_visual/index
    /autoapi/tilelang/analysis/nested_loop_checker/index
 
 

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

@@ -0,0 +1,31 @@
+tilelang.analysis.layout_visual
+===============================
+
+.. py:module:: tilelang.analysis.layout_visual
+
+
+Functions
+---------
+
+.. autoapisummary::
+
+   tilelang.analysis.layout_visual.print_fragment_format
+   tilelang.analysis.layout_visual.LayoutVisual
+
+
+Module Contents
+---------------
+
+.. py:function:: print_fragment_format(layout)
+
+   Format fragment layout information into a human-readable string.
+
+   :param layout: The fragment layout to format
+   :type layout: T.Fragment
+
+   :returns: Formatted string showing shape, thread mapping, and index mapping
+   :rtype: str
+
+
+.. py:function:: LayoutVisual(formats = '')
+

_sources/autoapi/tilelang/engine/phase/index.rst.txt

@@ -16,6 +16,9 @@ Functions
    tilelang.engine.phase.allow_global_thread_synchronization
    tilelang.engine.phase.should_enable_aggressive_merge
    tilelang.engine.phase.should_force_let_inline
+   tilelang.engine.phase.should_enable_layout_visual
+   tilelang.engine.phase.get_layout_visual_formats
+   tilelang.engine.phase.LayoutVisual
    tilelang.engine.phase.PreLowerSemanticCheck
    tilelang.engine.phase.LowerAndLegalize
    tilelang.engine.phase.OptimizeForTarget
@@ -38,6 +41,15 @@ Module Contents
 
 .. py:function:: should_force_let_inline(pass_ctx = None)
 
+.. py:function:: should_enable_layout_visual(pass_ctx = None)
+
+.. py:function:: get_layout_visual_formats(pass_ctx = None)
+
+.. py:function:: LayoutVisual(mod)
+
+   Apply layout visualization pass if enabled.
+
+
 .. py:function:: PreLowerSemanticCheck(mod)
 
    Check whether the module is valid before lowering. If not, raise a user-friendly error

_sources/autoapi/tilelang/tools/plot_layout/index.rst.txt

@@ -15,7 +15,7 @@ Functions
 Module Contents
 ---------------
 
-.. py:function:: plot_layout(layout, save_directory='./tmp', name = 'layout', colormap = 'RdPu', verbose = False)
+.. py:function:: plot_layout(layout, save_directory='./tmp', name = 'layout', colormap = 'RdPu', verbose = False, formats = 'png')
 
    Plot the layout of a buffer.
 
@@ -29,6 +29,8 @@ Module Contents
    :type colormap: str, optional
    :param verbose: If True, prints additional information about the mapping (default is False).
    :type verbose: bool, optional
+   :param formats: The formats to save the image in (default is "png").
+   :type formats: str | list[str], optional
 
    :rtype: None
 

_sources/autoapi/tilelang/transform/pass_config/index.rst.txt

@@ -188,6 +188,23 @@ Module Contents
       :type: Force TileLang to inline let bindings during simplification. Default
 
 
+   .. py:attribute:: TL_LAYOUT_VISUALIZATION_ENABLE
+      :value: 'tl.layout_visualization_enable'
+
+
+      False
+
+      :type: Enable layout inference visualization. Default
+
+
+   .. py:attribute:: TL_LAYOUT_VISUALIZATION_FORMATS
+      :value: 'tl.layout_visualization_formats'
+
+
+      Layout visualization formats.
+      Acceptable values: "pdf", "png", "svg", "all"
+
+
    .. py:attribute:: TL_STORAGE_REWRITE_DETECT_INPLACE
       :value: 'tl.storage_rewrite_detect_inplace'
 

_sources/tutorials/debug_tools_for_tilelang.md.txt

@@ -171,6 +171,32 @@ The output messages will include something like:
 msg='hello world' BlockIdx=(0, 0, 0), ThreadIdx=(0, 0, 0): 0
 ```
 
+### Visual Layout Inference For TileLang
+ The **Visual Layout Inference** tool automatically generates visual diagrams that illustrate the mapping between logical indices, thread IDs, and register file locations.
+
+When TileLang performs layout inference, it determines how fragment buffers are distributed across threads. The visual layout tool captures this information and generates:
+1. **Textual output**: A human-readable description of the layout mapping
+2. **Visual diagrams**: Color-coded plots showing the thread-to-data mapping
+
+The visual layout inference tool is controlled through the `TL_LAYOUT_VISUALIZATION_ENABLE` and `TL_LAYOUT_VISUALIZATION_FORMATS` pass configuration. By default, `TL_LAYOUT_VISUALIZATION_ENABLE` is **disabled** to avoid performance overhead during compilation.
+
+When enabled, `TL_LAYOUT_VISUALIZATION_FORMATS` accepts string values to control output formats:
+- "txt": Text output only (same as default)
+- "all": Generates all formats (TXT, PDF, PNG, SVG)
+- "png": Generate PNG format only
+- "pdf": Generate PDF format only
+- "svg": Generate SVG format only
+- "txt,svg": Generate multiple formats (comma-separated) in addition to text output
+
+The output messages of "txt" will include something like:
+```
+C_local inferenced layout:
+  Shape: [32, 32] -> [8]
+  Thread: _j // 16 * 64 + _i // 16 * 32 + _i % 8 * 4 + _j % 8 // 2
+  Index:  [_j % 16 // 8 * 4 + _i % 16 // 8 * 2 + _j % 2]
+```
+
+
 ## Conclusion
 
 By carefully examining intermediate representations (IR) before final code generation—and by leveraging runtime printing through `T.print`—one can quickly diagnose where index calculations, copy logic, or other kernel operations deviate from the intended behavior. This two-pronged approach (inspecting IR transformations and using runtime prints) is often sufficient for resolving generation and correctness issues in TileLang programs.

autoapi/index.html

@@ -229,6 +229,7 @@
 <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>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </li>
@@ -517,6 +518,7 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Link to this
 <li class="toctree-l2"><a class="reference internal" href="tilelang/analysis/index.html">tilelang.analysis</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang/analysis/nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </li>

autoapi/tilelang/analysis/ast_printer/index.html

@@ -229,6 +229,7 @@
 <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">
 <li class="toctree-l3 current current-page"><a class="current reference internal" href="#">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l3"><a class="reference internal" href="../layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </li>

autoapi/tilelang/analysis/fragment_loop_checker/index.html

@@ -3,7 +3,7 @@
   <head><meta charset="utf-8">
     <meta name="viewport" content="width=device-width,initial-scale=1">
     <meta name="color-scheme" content="light dark"><meta name="viewport" content="width=device-width, initial-scale=1" />
-<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.ast_printer" href="../ast_printer/index.html">
+<link rel="index" title="Index" href="../../../../genindex.html"><link rel="search" title="Search" href="../../../../search.html"><link rel="next" title="tilelang.analysis.layout_visual" href="../layout_visual/index.html"><link rel="prev" title="tilelang.analysis.ast_printer" href="../ast_printer/index.html">
         <link rel="prefetch" href="../../../../_static/img/logo-row.svg" as="image">
         <link rel="prefetch" href="../../../../_static/img/logo-row.svg" as="image">
 
@@ -229,6 +229,7 @@
 <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">
 <li class="toctree-l3"><a class="reference internal" href="../ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l3 current current-page"><a class="current reference internal" href="#">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l3"><a class="reference internal" href="../layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l3"><a class="reference internal" href="../nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </li>
@@ -567,12 +568,12 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
       <footer>
 
         <div class="related-pages">
-          <a class="next-page" href="../nested_loop_checker/index.html">
+          <a class="next-page" href="../layout_visual/index.html">
               <div class="page-info">
                 <div class="context">
                   <span>Next</span>
                 </div>
-                <div class="title">tilelang.analysis.nested_loop_checker</div>
+                <div class="title">tilelang.analysis.layout_visual</div>
               </div>
               <svg class="furo-related-icon"><use href="#svg-arrow-right"></use></svg>
             </a>

autoapi/tilelang/analysis/index.html

@@ -229,6 +229,7 @@
 <li class="toctree-l2 current has-children current-page"><a class="current reference internal" href="#">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>
 <li class="toctree-l3"><a class="reference internal" href="ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l3"><a class="reference internal" href="fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l3"><a class="reference internal" href="layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l3"><a class="reference internal" href="nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </li>
@@ -517,6 +518,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Link to this headi
 <ul>
 <li class="toctree-l1"><a class="reference internal" href="ast_printer/index.html">tilelang.analysis.ast_printer</a></li>
 <li class="toctree-l1"><a class="reference internal" href="fragment_loop_checker/index.html">tilelang.analysis.fragment_loop_checker</a></li>
+<li class="toctree-l1"><a class="reference internal" href="layout_visual/index.html">tilelang.analysis.layout_visual</a></li>
 <li class="toctree-l1"><a class="reference internal" href="nested_loop_checker/index.html">tilelang.analysis.nested_loop_checker</a></li>
 </ul>
 </div>