From 39b12350821be36ca71f97bd0d3e67dd96bf4d1e Mon Sep 17 00:00:00 2001 From: Yan Da Date: Sun, 22 May 2022 19:43:04 +0800 Subject: [PATCH] fix atomic_cas --- python/triton/language/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/triton/language/core.py b/python/triton/language/core.py index ab9c194a0..282f06db4 100644 --- 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)