Skip to content

Commit 84ba92c

Browse files
Fixes for gpu
1 parent c2795aa commit 84ba92c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_common.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,9 @@ def test_common_atomicadd(test_context):
5252

5353
expected = 1000
5454
result = np.array([0], dtype=np.float64)
55-
test_context.kernels.test_atomic_add(out=result)
55+
result_ctx = test_context.nparray_to_context_array(result)
56+
test_context.kernels.test_atomic_add(out=result_ctx)
57+
result = test_context.nparray_form_context_array(result_ctx)[0]
5658

5759
assert result == expected
5860

0 commit comments

Comments
 (0)