gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CODEGEN] Now using atomic_rmw code path for atomic_xchg (#222)

Browse Source
This commit is contained in:
Philippe Tillet
2021-08-17 16:33:23 -07:00
committed by GitHub
parent 64b8e7222d
commit 226fde6ea1
11 changed files with 14 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/triton/language.py
View File

@@ -442,7 +442,7 @@ def atomic_cas(pointer, cmp, val, builder=None):
@builtin
def atomic_xchg(pointer, val, builder=None):
def atomic_xchg(pointer, val, mask=None, builder=None):
"""
Swaps the *old* values stored at location :code:`pointer` with the new values given by :code:`val`. Returns the old values.
@@ -450,8 +450,10 @@ def atomic_xchg(pointer, val, builder=None):
:type pointer: Block of dtype=triton.PointerDType
:param val: The new values to store
:type val: Block of dtype=`pointer.dtype.element_ty`
:param mask: If mask[idx] is false, :code:`pointer[idx]` is unaffected.
:type mask: Block of triton.int1, optional
"""
return frontend.atomic_xchg(pointer, val, builder)
return frontend.atomic_xchg(pointer, val, mask, builder)
@builtin