Skip to content

Commit d841090

Browse files
Update docs
1 parent 4b418ea commit d841090

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

_sources/autoapi/tilelang/language/customize/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ Module Contents
111111
:type value: PrimExpr
112112
:param memory_order: Optional memory-order name (e.g. "relaxed", "acquire", "seq_cst").
113113
If provided, it is translated to the corresponding numeric memory-order id before the call.
114-
:type memory_order: str | None
114+
:type memory_order: Optional[str]
115115

116116
:returns: A handle/expression representing the issued atomic maximum operation.
117117
:rtype: PrimExpr
@@ -126,7 +126,7 @@ Module Contents
126126
to integer IDs). If memory_order is None, the extern is invoked without an explicit memory-order argument.
127127

128128
:param memory_order: Optional memory-order name controlling the atomic operation's ordering.
129-
:type memory_order: str | None
129+
:type memory_order: Optional[str]
130130

131131
:returns: A handle expression representing the atomic-min operation.
132132
:rtype: PrimExpr

autoapi/tilelang/language/customize/index.html

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
628628
<dd class="field-odd"><ul class="simple">
629629
<li><p><strong>dst</strong> (<em>Buffer</em>) – Destination buffer/address to apply the atomic max.</p></li>
630630
<li><p><strong>value</strong> (<em>PrimExpr</em>) – Value to compare/store atomically.</p></li>
631-
<li><p><strong>memory_order</strong> (<em>str</em><em> | </em><em>None</em>) – Optional memory-order name (e.g. “relaxed”, “acquire”, “seq_cst”).
631+
<li><p><strong>memory_order</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>) – Optional memory-order name (e.g. “relaxed”, “acquire”, “seq_cst”).
632632
If provided, it is translated to the corresponding numeric memory-order id before the call.</p></li>
633633
</ul>
634634
</dd>
@@ -651,7 +651,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
651651
<dl class="field-list simple">
652652
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
653653
<dd class="field-odd"><ul class="simple">
654-
<li><p><strong>memory_order</strong> (<em>str</em><em> | </em><em>None</em>) – Optional memory-order name controlling the atomic operation’s ordering.</p></li>
654+
<li><p><strong>memory_order</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>) – Optional memory-order name controlling the atomic operation’s ordering.</p></li>
655655
<li><p><strong>dst</strong> (<em>tvm.tir.Buffer</em>)</p></li>
656656
<li><p><strong>value</strong> (<em>tvm.tir.PrimExpr</em>)</p></li>
657657
</ul>
@@ -681,7 +681,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
681681
<dd class="field-odd"><ul class="simple">
682682
<li><p><strong>dst</strong> (<em>tvm.tir.Buffer</em>)</p></li>
683683
<li><p><strong>value</strong> (<em>tvm.tir.PrimExpr</em>)</p></li>
684-
<li><p><strong>memory_order</strong> (<em>str</em><em> | </em><em>None</em>)</p></li>
684+
<li><p><strong>memory_order</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>)</p></li>
685685
</ul>
686686
</dd>
687687
</dl>

0 commit comments

Comments
 (0)