Merge pull request #826 from IntelPython/samaid_examples_refactor

Add new kernel examples and restructure the existing kernel examples.

Author: Diptorup Deb

.github/workflows/conda-package.yml

@@ -152,7 +152,7 @@ jobs:
           conda activate numba_dpex_env
           # echo "libintelocl.so" | tee /etc/OpenCL/vendors/intel-cpu.icd
           export OCL_ICD_FILENAMES=libintelocl.so
-          for script in $(find . \( -not -name "_*" -not -name "vectorize.py" -and -name "*.py" \))
+          for script in $(find . \( -not -name "_*" -not -name "vectorize.py" -not -name "scan.py" -and -name "*.py" \))
           do
             echo "Executing ${script}"
             python ${script} || exit 1

numba_dpex/examples/atomic_op.py

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: 2020 - 2022 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import dpnp as np
+
+import numba_dpex as ndpex
+
+
+@ndpex.kernel
+def atomic_reduction(a):
+    idx = ndpex.get_global_id(0)
+    ndpex.atomic.add(a, 0, a[idx])
+
+
+def main():
+    N = 10
+    a = np.arange(N)
+
+    print("Using device ...")
+    print(a.device)
+
+    atomic_reduction[N, ndpex.DEFAULT_LOCAL_SIZE](a)
+    print("Reduction sum =", a[0])
+
+    print("Done...")
+
+
+if __name__ == "__main__":
+    main()