Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/language/v2/builder/index.rst.txt

@@ -179,6 +179,9 @@ Module Contents
       :type:  dict[str, AnyFrame]
 
 
+   .. py:attribute:: arg_annotations
+
+
    .. py:method:: current()
       :classmethod:
 
@@ -187,7 +190,7 @@ Module Contents
    .. py:method:: prim_func(name)
 
 
-   .. py:method:: macro(name=None)
+   .. py:method:: macro(name=None, annotations=None)
 
 
    .. py:method:: get()
@@ -265,6 +268,12 @@ Module Contents
    .. py:method:: rval(name, value)
 
 
+   .. py:method:: macro_arg(name, value)
+
+
+   .. py:method:: prim_func_arg(name, value)
+
+
    .. py:method:: arg(name, value)
 
 
@@ -369,6 +378,10 @@ Module Contents
       :type:  tilelang.language.v2.ast.IRGenerator[_P, _T]
 
 
+   .. py:attribute:: annotations
+      :type:  dict[str, Any]
+
+
    .. py:property:: source
       :type: str
 

autoapi/tilelang/language/v2/builder/index.html

@@ -766,6 +766,11 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <span class="sig-name descname"><span class="pre">name_inside_frame</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">dict</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">AnyFrame</span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#tilelang.language.v2.builder.Builder.name_inside_frame" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
+<dl class="py attribute">
+<dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.arg_annotations">
+<span class="sig-name descname"><span class="pre">arg_annotations</span></span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.arg_annotations" title="Link to this definition">¶</a></dt>
+<dd></dd></dl>
+
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.current">
 <em class="property"><span class="pre">classmethod</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">current</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.current" title="Link to this definition">¶</a></dt>
@@ -783,7 +788,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.macro">
-<span class="sig-name descname"><span class="pre">macro</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.macro" title="Link to this definition">¶</a></dt>
+<span class="sig-name descname"><span class="pre">macro</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">annotations</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.macro" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
 <dl class="py method">
@@ -954,6 +959,16 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 </dl>
 </dd></dl>
 
+<dl class="py method">
+<dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.macro_arg">
+<span class="sig-name descname"><span class="pre">macro_arg</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.macro_arg" title="Link to this definition">¶</a></dt>
+<dd></dd></dl>
+
+<dl class="py method">
+<dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.prim_func_arg">
+<span class="sig-name descname"><span class="pre">prim_func_arg</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.prim_func_arg" title="Link to this definition">¶</a></dt>
+<dd></dd></dl>
+
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.v2.builder.Builder.arg">
 <span class="sig-name descname"><span class="pre">arg</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.v2.builder.Builder.arg" title="Link to this definition">¶</a></dt>
@@ -1077,6 +1092,11 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <span class="sig-name descname"><span class="pre">ir_gen</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference internal" href="../ast/index.html#tilelang.language.v2.ast.IRGenerator" title="tilelang.language.v2.ast.IRGenerator"><span class="pre">tilelang.language.v2.ast.IRGenerator</span></a><span class="p"><span class="pre">[</span></span><span class="pre">_P</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">_T</span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#tilelang.language.v2.builder.Macro.ir_gen" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
+<dl class="py attribute">
+<dt class="sig sig-object py" id="tilelang.language.v2.builder.Macro.annotations">
+<span class="sig-name descname"><span class="pre">annotations</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">dict</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Any</span><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#tilelang.language.v2.builder.Macro.annotations" title="Link to this definition">¶</a></dt>
+<dd></dd></dl>
+
 <dl class="py property">
 <dt class="sig sig-object py" id="tilelang.language.v2.builder.Macro.source">
 <em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">source</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">str</span></em><a class="headerlink" href="#tilelang.language.v2.builder.Macro.source" title="Link to this definition">¶</a></dt>
@@ -1313,6 +1333,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.frames"><code class="docutils literal notranslate"><span class="pre">Builder.frames</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.ir_builder"><code class="docutils literal notranslate"><span class="pre">Builder.ir_builder</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.name_inside_frame"><code class="docutils literal notranslate"><span class="pre">Builder.name_inside_frame</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.arg_annotations"><code class="docutils literal notranslate"><span class="pre">Builder.arg_annotations</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.current"><code class="docutils literal notranslate"><span class="pre">Builder.current()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.prim_func"><code class="docutils literal notranslate"><span class="pre">Builder.prim_func()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.macro"><code class="docutils literal notranslate"><span class="pre">Builder.macro()</span></code></a></li>
@@ -1341,6 +1362,8 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.ctx_with"><code class="docutils literal notranslate"><span class="pre">Builder.ctx_with()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.assert_expr"><code class="docutils literal notranslate"><span class="pre">Builder.assert_expr()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.rval"><code class="docutils literal notranslate"><span class="pre">Builder.rval()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.macro_arg"><code class="docutils literal notranslate"><span class="pre">Builder.macro_arg()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.prim_func_arg"><code class="docutils literal notranslate"><span class="pre">Builder.prim_func_arg()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.arg"><code class="docutils literal notranslate"><span class="pre">Builder.arg()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Builder.override"><code class="docutils literal notranslate"><span class="pre">Builder.override()</span></code></a></li>
 </ul>
@@ -1361,6 +1384,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.name"><code class="docutils literal notranslate"><span class="pre">Macro.name</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.orig_func"><code class="docutils literal notranslate"><span class="pre">Macro.orig_func</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.ir_gen"><code class="docutils literal notranslate"><span class="pre">Macro.ir_gen</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.annotations"><code class="docutils literal notranslate"><span class="pre">Macro.annotations</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.source"><code class="docutils literal notranslate"><span class="pre">Macro.source</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.v2.builder.Macro.__call__"><code class="docutils literal notranslate"><span class="pre">Macro.__call__()</span></code></a></li>
 </ul>

genindex.html

@@ -832,8 +832,12 @@ <h2>A</h2>
 </li>
         <li><a href="autoapi/tilelang/transform/index.html#tilelang.transform.AnnotateWarpGroupRegAlloc">AnnotateWarpGroupRegAlloc() (in module tilelang.transform)</a>
 </li>
-        <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.SerialForWithStep.annotations">annotations (tilelang.language.v2.builder.SerialForWithStep attribute)</a>
+        <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Macro.annotations">annotations (tilelang.language.v2.builder.Macro attribute)</a>
+
+        <ul>
+          <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.SerialForWithStep.annotations">(tilelang.language.v2.builder.SerialForWithStep attribute)</a>
 </li>
+        </ul></li>
         <li><a href="autoapi/tilelang/language/tir/op/index.html#tilelang.language.tir.op.any">any() (in module tilelang.language.tir.op)</a>
 </li>
         <li><a href="autoapi/tilelang/language/logical/index.html#tilelang.language.logical.any_of">any_of() (in module tilelang.language.logical)</a>
@@ -884,6 +888,8 @@ <h2>A</h2>
           <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.arg">(tilelang.language.v2.builder.Builder method)</a>
 </li>
         </ul></li>
+        <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.arg_annotations">arg_annotations (tilelang.language.v2.builder.Builder attribute)</a>
+</li>
         <li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.PrimFuncNode.args">args (tilelang.carver.roller.node.PrimFuncNode attribute)</a>
 </li>
         <li><a href="autoapi/tilelang/utils/language/index.html#tilelang.utils.language.array_reduce">array_reduce() (in module tilelang.utils.language)</a>
@@ -3706,6 +3712,8 @@ <h2>M</h2>
           <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.macro">(tilelang.language.v2.builder.Builder method)</a>
 </li>
         </ul></li>
+        <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.macro_arg">macro_arg() (tilelang.language.v2.builder.Builder method)</a>
+</li>
         <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.MacroFrame">MacroFrame (class in tilelang.language.v2.builder)</a>
 </li>
         <li><a href="autoapi/tilelang/testing/index.html#tilelang.testing.main">main() (in module tilelang.testing)</a>
@@ -4675,6 +4683,8 @@ <h2>P</h2>
           <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.prim_func">(tilelang.language.v2.builder.Builder method)</a>
 </li>
         </ul></li>
+        <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.Builder.prim_func_arg">prim_func_arg() (tilelang.language.v2.builder.Builder method)</a>
+</li>
         <li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.PrimFunc">PrimFunc (class in tilelang.language.v2.builder)</a>
 </li>
         <li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.PrimFuncNode">PrimFuncNode (class in tilelang.carver.roller.node)</a>