This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch asf-site
in repository https://gitbox.apache.org/repos/asf/tvm-site.git
The following commit(s) were added to refs/heads/asf-site by this push:
new b260a21fdad deploying tirx docs
(0e51e7b08a3f8330a07190d94527838867578a8a)
b260a21fdad is described below
commit b260a21fdade72578b0b6f87aec6016229e25abf
Author: tqchen <[email protected]>
AuthorDate: Tue Oct 20 02:58:13 2026 +0000
deploying tirx docs (0e51e7b08a3f8330a07190d94527838867578a8a)
---
.../reference/python/backend.trn.transform.rst | 1 +
...kend.trn.transform.AnnotateKernelArguments.rst} | 27 ++++--------
tirx/commit_hash | 2 +-
tirx/genindex.html | 2 +
tirx/objects.inv | Bin 45908 -> 45962 bytes
tirx/reference/python/backend.trn.transform.html | 14 +++---
...end.trn.transform.AnnotateKernelArguments.html} | 47 +++++++++++----------
.../tvm_tirx.backend.trn.transform.LowerTIRx.html | 1 +
....backend.trn.transform.LowerTrainiumLayout.html | 1 +
...rx.backend.trn.transform.TrnNaiveAllocator.html | 7 +--
...ackend.trn.transform.TrnPrivateTensorAlloc.html | 1 +
.../generated/tvm_tirx.tirx.tile.Operand.html | 2 +-
tirx/searchindex.js | 2 +-
13 files changed, 56 insertions(+), 51 deletions(-)
diff --git a/tirx/_sources/reference/python/backend.trn.transform.rst
b/tirx/_sources/reference/python/backend.trn.transform.rst
index 7aa5ea95f08..5bcf802e0bb 100644
--- a/tirx/_sources/reference/python/backend.trn.transform.rst
+++ b/tirx/_sources/reference/python/backend.trn.transform.rst
@@ -29,6 +29,7 @@ Classes
.. autosummary::
:toctree: generated/
+ AnnotateKernelArguments
TrnNaiveAllocator
TrnPrivateTensorAlloc
diff --git a/tirx/_sources/reference/python/backend.trn.transform.rst
b/tirx/_sources/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.rst
similarity index 67%
copy from tirx/_sources/reference/python/backend.trn.transform.rst
copy to
tirx/_sources/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.rst
index 7aa5ea95f08..9cb595f58db 100644
--- a/tirx/_sources/reference/python/backend.trn.transform.rst
+++
b/tirx/_sources/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.rst
@@ -1,4 +1,4 @@
-.. Licensed to the Apache Software Foundation (ASF) under one
+.. Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
@@ -15,28 +15,19 @@
specific language governing permissions and limitations
under the License.
-tvm_tirx.backend.trn.transform
-==============================
-
-.. automodule:: tvm_tirx.backend.trn.transform
- :no-members:
+tvm\_tirx.backend.trn.transform.AnnotateKernelArguments
+=======================================================
.. currentmodule:: tvm_tirx.backend.trn.transform
-Classes
--------
+.. autoclass:: AnnotateKernelArguments
+
+
+
+
+
-.. autosummary::
- :toctree: generated/
- TrnNaiveAllocator
- TrnPrivateTensorAlloc
-Functions
----------
-.. autosummary::
- :toctree: generated/
- LowerTrainiumLayout
- LowerTIRx
diff --git a/tirx/commit_hash b/tirx/commit_hash
index 16f3061b9a8..a0c46cd77e9 100644
--- a/tirx/commit_hash
+++ b/tirx/commit_hash
@@ -1 +1 @@
-ac996ef5cd8c0b054904e2da272a3e9a5638015a
+0e51e7b08a3f8330a07190d94527838867578a8a
diff --git a/tirx/genindex.html b/tirx/genindex.html
index 6a9d24d5339..5690ab27f7c 100644
--- a/tirx/genindex.html
+++ b/tirx/genindex.html
@@ -549,6 +549,8 @@
<li><a
href="reference/python/generated/tvm_tirx.tirx.script.and_.html#tvm_tirx.tirx.script.and_">and_()
(in module tvm_tirx.tirx.script)</a>
</li>
<li><a
href="reference/python/generated/tvm_tirx.tirx.transform.AnnotateEntryFunc.html#tvm_tirx.tirx.transform.AnnotateEntryFunc">AnnotateEntryFunc()
(in module tvm_tirx.tirx.transform)</a>
+</li>
+ <li><a
href="reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html#tvm_tirx.backend.trn.transform.AnnotateKernelArguments">AnnotateKernelArguments
(class in tvm_tirx.backend.trn.transform)</a>
</li>
<li><a
href="reference/python/generated/tvm_tirx.script.printer.doc.AssignDoc.html#tvm_tirx.script.printer.doc.AssignDoc.annotation">annotation
(tvm_tirx.script.printer.doc.AssignDoc property)</a>
</li>
diff --git a/tirx/objects.inv b/tirx/objects.inv
index 35e05236346..cd3399fde7a 100644
Binary files a/tirx/objects.inv and b/tirx/objects.inv differ
diff --git a/tirx/reference/python/backend.trn.transform.html
b/tirx/reference/python/backend.trn.transform.html
index d0d8d4cd9a7..a78a51d8009 100644
--- a/tirx/reference/python/backend.trn.transform.html
+++ b/tirx/reference/python/backend.trn.transform.html
@@ -47,7 +47,7 @@
<script>DOCUMENTATION_OPTIONS.search_as_you_type = false;</script>
<link rel="index" title="Index" href="../../genindex.html" />
<link rel="search" title="Search" href="../../search.html" />
- <link rel="next" title="tvm_tirx.backend.trn.transform.TrnNaiveAllocator"
href="generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html" />
+ <link rel="next"
title="tvm_tirx.backend.trn.transform.AnnotateKernelArguments"
href="generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html" />
<link rel="prev" title="tvm_tirx.backend.trn.op.nki_affine_select"
href="generated/tvm_tirx.backend.trn.op.nki_affine_select.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
@@ -226,6 +226,7 @@
<li class="toctree-l2"><a class="reference internal"
href="backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="current reference
internal" href="#">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul>
+<li class="toctree-l3"><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3"><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3"><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3"><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
@@ -423,10 +424,13 @@
<h2>Classes<a class="headerlink" href="#classes" title="Link to this
heading">#</a></h2>
<div class="pst-scrollable-table-container"><table class="autosummary
longtable table">
<tbody>
-<tr class="row-odd"><td><p><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html#tvm_tirx.backend.trn.transform.TrnNaiveAllocator"
title="tvm_tirx.backend.trn.transform.TrnNaiveAllocator"><code class="xref py
py-obj docutils literal notranslate"><span
class="pre">TrnNaiveAllocator</span></code></a>(*args, **kwargs)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html#tvm_tirx.backend.trn.transform.AnnotateKernelArguments"
title="tvm_tirx.backend.trn.transform.AnnotateKernelArguments"><code
class="xref py py-obj docutils literal notranslate"><span
class="pre">AnnotateKernelArguments</span></code></a>(*args, **kwargs)</p></td>
+<td><p>Order kernel inputs before outputs and annotate their per-kernel input
count.</p></td>
+</tr>
+<tr class="row-even"><td><p><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html#tvm_tirx.backend.trn.transform.TrnNaiveAllocator"
title="tvm_tirx.backend.trn.transform.TrnNaiveAllocator"><code class="xref py
py-obj docutils literal notranslate"><span
class="pre">TrnNaiveAllocator</span></code></a>(*args, **kwargs)</p></td>
<td><p>Assign consecutive byte offsets to unaddressed tensor
allocations.</p></td>
</tr>
-<tr class="row-even"><td><p><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html#tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc"
title="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc"><code class="xref
py py-obj docutils literal notranslate"><span
class="pre">TrnPrivateTensorAlloc</span></code></a>(*args, **kwargs)</p></td>
+<tr class="row-odd"><td><p><a class="reference internal"
href="generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html#tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc"
title="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc"><code class="xref
py py-obj docutils literal notranslate"><span
class="pre">TrnPrivateTensorAlloc</span></code></a>(*args, **kwargs)</p></td>
<td><p>Allocate and initialize instruction-private Trainium
workspaces.</p></td>
</tr>
</tbody>
@@ -470,11 +474,11 @@
</div>
</a>
<a class="right-next"
- href="generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html"
+
href="generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html"
title="next page">
<div class="prev-next-info">
<p class="prev-next-subtitle">next</p>
- <p
class="prev-next-title">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</p>
+ <p
class="prev-next-title">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</p>
</div>
<i class="fa-solid fa-angle-right"></i>
</a>
diff --git
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html
similarity index 85%
copy from
tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
copy to
tirx/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html
index f339a59617d..14ca821da79 100644
---
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
+++
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html
@@ -8,7 +8,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0"
/><meta name="viewport" content="width=device-width, initial-scale=1" />
- <title>tvm_tirx.backend.trn.transform.LowerTIRx — tvm-tirx</title>
+ <title>tvm_tirx.backend.trn.transform.AnnotateKernelArguments —
tvm-tirx</title>
@@ -43,12 +43,12 @@
<script src="../../../_static/doctools.js?v=fd6eb6e6"></script>
<script src="../../../_static/sphinx_highlight.js?v=6ffebe34"></script>
<script
src="../../../_static/scripts/sphinx-book-theme.js?v=fab101a9"></script>
- <script>DOCUMENTATION_OPTIONS.pagename =
'reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx';</script>
+ <script>DOCUMENTATION_OPTIONS.pagename =
'reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments';</script>
<script>DOCUMENTATION_OPTIONS.search_as_you_type = false;</script>
<link rel="index" title="Index" href="../../../genindex.html" />
<link rel="search" title="Search" href="../../../search.html" />
- <link rel="next" title="tvm_tirx.support" href="../support.html" />
- <link rel="prev"
title="tvm_tirx.backend.trn.transform.LowerTrainiumLayout"
href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html" />
+ <link rel="next" title="tvm_tirx.backend.trn.transform.TrnNaiveAllocator"
href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html" />
+ <link rel="prev" title="tvm_tirx.backend.trn.transform"
href="../backend.trn.transform.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docsearch:version" content="" />
@@ -226,10 +226,11 @@
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="reference
internal"
href="../backend.trn.transform.html">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul class="current">
+<li class="toctree-l3 current active"><a class="current reference internal"
href="#">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
-<li class="toctree-l3 current active"><a class="current reference internal"
href="#">tvm_tirx.backend.trn.transform.LowerTIRx</a></li>
+<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.LowerTIRx.html">tvm_tirx.backend.trn.transform.LowerTIRx</a></li>
</ul>
</details></li>
<li class="toctree-l2"><a class="reference internal"
href="../support.html">tvm_tirx.support</a></li>
@@ -310,7 +311,7 @@
- <li><a
href="../../../_sources/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.rst"
target="_blank"
+ <li><a
href="../../../_sources/reference/python/generated/tvm_tirx.backend.trn.transform.AnnotateKernelArguments.rst"
target="_blank"
class="btn btn-sm btn-download-source-button dropdown-item"
title="Download source file"
data-bs-placement="left" data-bs-toggle="tooltip"
@@ -393,7 +394,7 @@
<div id="jb-print-docs-body" class="onlyprint">
- <h1>tvm_tirx.backend.trn.transform.LowerTIRx</h1>
+ <h1>tvm_tirx.backend.trn.transform.AnnotateKernelArguments</h1>
<!-- Table of contents -->
<div id="print-main-content">
<div id="jb-print-toc">
@@ -403,7 +404,7 @@
</div>
<nav aria-label="Page">
<ul class="pst-show_toc_level nav section-nav flex-column">
-<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link"
href="#tvm_tirx.backend.trn.transform.LowerTIRx"><code class="docutils literal
notranslate"><span class="pre">LowerTIRx()</span></code></a></li>
+<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link"
href="#tvm_tirx.backend.trn.transform.AnnotateKernelArguments"><code
class="docutils literal notranslate"><span
class="pre">AnnotateKernelArguments</span></code></a></li>
</ul>
</nav>
</div>
@@ -415,15 +416,17 @@
<div id="searchbox"></div>
<article class="bd-article">
- <section id="tvm-tirx-backend-trn-transform-lowertirx">
-<h1>tvm_tirx.backend.trn.transform.LowerTIRx<a class="headerlink"
href="#tvm-tirx-backend-trn-transform-lowertirx" title="Link to this
heading">#</a></h1>
-<dl class="py function">
-<dt class="sig sig-object py" id="tvm_tirx.backend.trn.transform.LowerTIRx">
-<span class="sig-prename descclassname"><span
class="pre">tvm_tirx.backend.trn.transform.</span></span><span class="sig-name
descname"><span class="pre">LowerTIRx</span></span><span
class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink"
href="#tvm_tirx.backend.trn.transform.LowerTIRx" title="Link to this
definition">#</a></dt>
-<dd><p>Run <code class="docutils literal notranslate"><span
class="pre">TileDispatch</span></code> followed by <code class="docutils
literal notranslate"><span class="pre">LowerTrainiumLayout</span></code>.</p>
-<p>Run <code class="docutils literal notranslate"><span
class="pre">TrnPrivateTensorAlloc</span></code> and <code class="docutils
literal notranslate"><span class="pre">TrnNaiveAllocator</span></code> first
when tiles
-need workspaces or allocation addresses. Tile dispatch requires a
-Trainium target on the function or in the current target scope.</p>
+ <section id="tvm-tirx-backend-trn-transform-annotatekernelarguments">
+<h1>tvm_tirx.backend.trn.transform.AnnotateKernelArguments<a
class="headerlink"
href="#tvm-tirx-backend-trn-transform-annotatekernelarguments" title="Link to
this heading">#</a></h1>
+<dl class="py class">
+<dt class="sig sig-object py"
id="tvm_tirx.backend.trn.transform.AnnotateKernelArguments">
+<span class="property"><span class="k"><span
class="pre">class</span></span><span class="w"> </span></span><span
class="sig-prename descclassname"><span
class="pre">tvm_tirx.backend.trn.transform.</span></span><span class="sig-name
descname"><span class="pre">AnnotateKernelArguments</span></span><span
class="sig-paren">(</span><em class="sig-param"><span class="o"><span
class="pre">*</span></span><span class="n"><span
class="pre">args</span></span></em>, <em class="sig-param"><span class [...]
+<dd><p>Bases: <a class="reference internal"
href="tvm_tirx.ir.transform.ModulePass.html#tvm_tirx.ir.transform.ModulePass"
title="tvm_tirx.ir.transform.core.ModulePass"><code class="xref py py-class
docutils literal notranslate"><span class="pre">ModulePass</span></code></a></p>
+<p>Order kernel inputs before outputs and annotate their per-kernel input
count.</p>
+<p>Run after <code class="docutils literal notranslate"><span
class="pre">SplitHostDevice</span></code> and before <code class="docutils
literal notranslate"><span class="pre">MakePackedAPI</span></code>. The source
+<code class="docutils literal notranslate"><span
class="pre">trn.num_input_args</span></code> contract remains on each host
function until its
+kernel calls have been matched to the surviving host parameters and captures.
+Functions without extracted kernels retain their own contract.</p>
</dd></dl>
</section>
@@ -440,20 +443,20 @@ Trainium target on the function or in the current target
scope.</p>
<div class="prev-next-area">
<a class="left-prev"
- href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html"
+ href="../backend.trn.transform.html"
title="previous page">
<i class="fa-solid fa-angle-left"></i>
<div class="prev-next-info">
<p class="prev-next-subtitle">previous</p>
- <p
class="prev-next-title">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</p>
+ <p class="prev-next-title">tvm_tirx.backend.trn.transform</p>
</div>
</a>
<a class="right-next"
- href="../support.html"
+ href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html"
title="next page">
<div class="prev-next-info">
<p class="prev-next-subtitle">next</p>
- <p class="prev-next-title">tvm_tirx.support</p>
+ <p
class="prev-next-title">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</p>
</div>
<i class="fa-solid fa-angle-right"></i>
</a>
@@ -475,7 +478,7 @@ Trainium target on the function or in the current target
scope.</p>
</div>
<nav id="pst-page-toc-nav" class="page-toc"
aria-labelledby="pst-page-navigation-heading-2">
<ul class="pst-show_toc_level nav section-nav flex-column">
-<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link"
href="#tvm_tirx.backend.trn.transform.LowerTIRx"><code class="docutils literal
notranslate"><span class="pre">LowerTIRx()</span></code></a></li>
+<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link"
href="#tvm_tirx.backend.trn.transform.AnnotateKernelArguments"><code
class="docutils literal notranslate"><span
class="pre">AnnotateKernelArguments</span></code></a></li>
</ul>
</nav></div>
diff --git
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
index f339a59617d..cde5b5c78c3 100644
---
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
+++
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTIRx.html
@@ -226,6 +226,7 @@
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="reference
internal"
href="../backend.trn.transform.html">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul class="current">
+<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
diff --git
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html
index 821131e5298..5cc29e6524f 100644
---
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html
+++
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html
@@ -226,6 +226,7 @@
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="reference
internal"
href="../backend.trn.transform.html">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul class="current">
+<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3 current active"><a class="current reference internal"
href="#">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
diff --git
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html
index e13d2314333..6228fced296 100644
---
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html
+++
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html
@@ -48,7 +48,7 @@
<link rel="index" title="Index" href="../../../genindex.html" />
<link rel="search" title="Search" href="../../../search.html" />
<link rel="next"
title="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc"
href="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html" />
- <link rel="prev" title="tvm_tirx.backend.trn.transform"
href="../backend.trn.transform.html" />
+ <link rel="prev"
title="tvm_tirx.backend.trn.transform.AnnotateKernelArguments"
href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<meta name="docsearch:language" content="en"/>
<meta name="docsearch:version" content="" />
@@ -226,6 +226,7 @@
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="reference
internal"
href="../backend.trn.transform.html">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul class="current">
+<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3 current active"><a class="current reference internal"
href="#">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
@@ -462,12 +463,12 @@ allocations, without lifetime-based reuse.</p>
<div class="prev-next-area">
<a class="left-prev"
- href="../backend.trn.transform.html"
+ href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html"
title="previous page">
<i class="fa-solid fa-angle-left"></i>
<div class="prev-next-info">
<p class="prev-next-subtitle">previous</p>
- <p class="prev-next-title">tvm_tirx.backend.trn.transform</p>
+ <p
class="prev-next-title">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</p>
</div>
</a>
<a class="right-next"
diff --git
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html
index dd4fd699c3d..1eede48afaa 100644
---
a/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html
+++
b/tirx/reference/python/generated/tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc.html
@@ -226,6 +226,7 @@
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.html">tvm_tirx.backend.trn</a></li>
<li class="toctree-l2"><a class="reference internal"
href="../backend.trn.op.html">tvm_tirx.backend.trn.op</a></li>
<li class="toctree-l2 current active has-children"><a class="reference
internal"
href="../backend.trn.transform.html">tvm_tirx.backend.trn.transform</a><details
open="open"><summary><span class="toctree-toggle" role="presentation"><i
class="fa-solid fa-chevron-down"></i></span></summary><ul class="current">
+<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.AnnotateKernelArguments.html">tvm_tirx.backend.trn.transform.AnnotateKernelArguments</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.TrnNaiveAllocator.html">tvm_tirx.backend.trn.transform.TrnNaiveAllocator</a></li>
<li class="toctree-l3 current active"><a class="current reference internal"
href="#">tvm_tirx.backend.trn.transform.TrnPrivateTensorAlloc</a></li>
<li class="toctree-l3"><a class="reference internal"
href="tvm_tirx.backend.trn.transform.LowerTrainiumLayout.html">tvm_tirx.backend.trn.transform.LowerTrainiumLayout</a></li>
diff --git a/tirx/reference/python/generated/tvm_tirx.tirx.tile.Operand.html
b/tirx/reference/python/generated/tvm_tirx.tirx.tile.Operand.html
index d4045a37193..68066f9013b 100644
--- a/tirx/reference/python/generated/tvm_tirx.tirx.tile.Operand.html
+++ b/tirx/reference/python/generated/tvm_tirx.tirx.tile.Operand.html
@@ -428,7 +428,7 @@
<h1>tvm_tirx.tirx.tile.Operand<a class="headerlink"
href="#tvm-tirx-tirx-tile-operand" title="Link to this heading">#</a></h1>
<dl class="py class">
<dt class="sig sig-object py" id="tvm_tirx.tirx.tile.Operand">
-<span class="property"><span class="k"><span
class="pre">class</span></span><span class="w"> </span></span><span
class="sig-prename descclassname"><span
class="pre">tvm_tirx.tirx.tile.</span></span><span class="sig-name
descname"><span class="pre">Operand</span></span><span
class="sig-paren">(</span><em class="sig-param"><span class="n"><span
class="pre">name</span></span><span class="p"><span
class="pre">:</span></span><span class="w"> </span><span class="n"><span
class="pre">str</span> [...]
+<span class="property"><span class="k"><span
class="pre">class</span></span><span class="w"> </span></span><span
class="sig-prename descclassname"><span
class="pre">tvm_tirx.tirx.tile.</span></span><span class="sig-name
descname"><span class="pre">Operand</span></span><span
class="sig-paren">(</span><em class="sig-param"><span class="n"><span
class="pre">name</span></span><span class="p"><span
class="pre">:</span></span><span class="w"> </span><span class="n"><span
class="pre">str</span> [...]
<dd><p>Bases: <code class="xref py py-class docutils literal
notranslate"><span class="pre">object</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
diff --git a/tirx/searchindex.js b/tirx/searchindex.js
index d038a54a964..ae3833144f3 100644
--- a/tirx/searchindex.js
+++ b/tirx/searchindex.js
@@ -1 +1 @@
-Search.setIndex({"alltitles":{"CallEffectKind":[[416,"calleffectkind"]],"CallingConv":[[413,"callingconv"]],"Classes":[[4,"classes"],[6,"classes"],[7,"classes"],[8,"classes"],[9,"classes"],[10,"classes"],[11,"classes"],[12,"classes"],[13,"classes"],[14,"classes"],[15,"classes"],[22,"classes"],[28,"classes"],[37,"classes"],[39,"classes"],[42,"classes"],[1460,"classes"],[1463,"classes"],[1464,"classes"],[1465,"classes"],[1466,"classes"],[1468,"classes"],[1470,"classes"],[1471,"classes"],[1
[...]
\ No newline at end of file
+Search.setIndex({"alltitles":{"CallEffectKind":[[417,"calleffectkind"]],"CallingConv":[[414,"callingconv"]],"Classes":[[4,"classes"],[6,"classes"],[7,"classes"],[8,"classes"],[9,"classes"],[10,"classes"],[11,"classes"],[12,"classes"],[13,"classes"],[14,"classes"],[15,"classes"],[22,"classes"],[28,"classes"],[37,"classes"],[39,"classes"],[42,"classes"],[1461,"classes"],[1464,"classes"],[1465,"classes"],[1466,"classes"],[1467,"classes"],[1469,"classes"],[1471,"classes"],[1472,"classes"],[1
[...]
\ No newline at end of file
