Update atomic example to be more stable

IntelPython · Jul 27, 2023 · 5eb17ca · 5eb17ca
1 parent 32b0da3
commit 5eb17ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/numba_dpex/examples/kernel/atomic_op.py b/numba_dpex/examples/kernel/atomic_op.py
@@ -10,12 +10,14 @@
 @ndpx.kernel
 def atomic_reduction(a):
     idx = ndpx.get_global_id(0)
-    ndpx.atomic.add(a, 0, a[idx])
+    ndpx.atomic.add(a, 0, a[idx + 1])
 
 
 def main():
     N = 10
-    a = np.arange(N)
+
+    # We are storing sum to the first element
+    a = np.arange(0, N + 1)
 
     print("Using device ...")
     print(a.device)