Merge pull request #826 from IntelPython/samaid_examples_refactor

Add new kernel examples and restructure the existing kernel examples.
IntelPython · Dec 8, 2022 · 3cef533 · 3cef533
2 parents 1a54068 + aa52db8
commit 3cef533
Show file tree

Hide file tree

Showing 12 changed files with 437 additions and 350 deletions.
diff --git a/.github/workflows/conda-package.yml b/.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

diff --git a/numba_dpex/examples/atomic_op.py b/numba_dpex/examples/atomic_op.py
diff --git a/numba_dpex/examples/barrier.py b/numba_dpex/examples/barrier.py
diff --git a/numba_dpex/examples/blacksholes_kernel.py b/numba_dpex/examples/blacksholes_kernel.py
diff --git a/numba_dpex/examples/dpex_func.py b/numba_dpex/examples/dpex_func.py
diff --git a/numba_dpex/examples/dpex_with_context.py b/numba_dpex/examples/dpex_with_context.py
diff --git a/numba_dpex/examples/kernel/atomic_op.py b/numba_dpex/examples/kernel/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()