Skip to content

Commit 5e98a57

Browse files
Update docs
1 parent 58856e1 commit 5e98a57

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,8 @@ Module Contents
120120
:type memory_order: Optional[str]
121121
:param return_prev: If True, return the previous value; if False, return handle (default False).
122122
:type return_prev: bool
123+
:param use_tma: If True, use TMA (cp.reduce) to perform the atomic add. This is available only for sm90+ (default False).
124+
:type use_tma: bool
123125

124126
:returns: A handle representing the atomic addition operation, or the previous value if return_prev is True.
125127
:rtype: PrimExpr

autoapi/tilelang/language/atomic/index.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -617,7 +617,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
617617
<li><p><strong>value</strong> (<em>PrimExpr</em>) – Value to add atomically.</p></li>
618618
<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>
619619
<li><p><strong>return_prev</strong> (<em>bool</em>) – If True, return the previous value; if False, return handle (default False).</p></li>
620-
<li><p><strong>use_tma</strong> (<em>bool</em>)</p></li>
620+
<li><p><strong>use_tma</strong> (<em>bool</em>) – If True, use TMA (cp.reduce) to perform the atomic add. This is available only for sm90+ (default False).</p></li>
621621
</ul>
622622
</dd>
623623
<dt class="field-even">Returns<span class="colon">:</span></dt>

searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)