fix atomic_cas

2022-05-22 19:43:04 +08:00
parent 79298d61bc
commit 39b1235082
1 changed files with 1 additions and 1 deletions
--- a/python/triton/language/core.py
+++ b/python/triton/language/core.py
@@ -821,7 +821,7 @@ def _add_atomic_docstr(name):
@_add_atomic_docstr("compare-and-swap")
 def atomic_cas(pointer, cmp, val, _builder=None):
    cmp = _to_tensor(cmp, _builder)
-    val = _to_tensor(cmp, _builder)
+    val = _to_tensor(val, _builder)
    return semantic.atomic_cas(pointer, cmp, val, _builder)