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

fix atomic_cas

Browse Source
This commit is contained in:
Yan Da
2022-05-22 19:43:04 +08:00
parent 79298d61bc
commit 39b1235082
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/language/core.py
View File

@@ -821,7 +821,7 @@ def _add_atomic_docstr(name):
@_add_atomic_docstr("compare-and-swap") @_add_atomic_docstr("compare-and-swap")
def atomic_cas(pointer, cmp, val, _builder=None): def atomic_cas(pointer, cmp, val, _builder=None):
cmp = _to_tensor(cmp, _builder) cmp = _to_tensor(cmp, _builder)
val = _to_tensor(cmp, _builder) val = _to_tensor(val, _builder)
return semantic.atomic_cas(pointer, cmp, val, _builder) return semantic.atomic_cas(pointer, cmp, val, _builder)