Add a unit test for local accessor

Author: Diptorup Deb

numba_dpex/tests/experimental/kernel_api_overloads/spv_overloads/test_local_accessors.py

@@ -0,0 +1,54 @@
+# SPDX-FileCopyrightText: 2023 - 2024 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+
+import dpnp
+import pytest
+
+import numba_dpex as dpex
+import numba_dpex.experimental as dpex_exp
+from numba_dpex.kernel_api import (
+    LocalAccessor,
+    MemoryScope,
+    NdItem,
+    group_barrier,
+)
+from numba_dpex.tests._helper import get_all_dtypes
+
+list_of_supported_dtypes = get_all_dtypes(
+    no_bool=True, no_float16=True, no_none=True, no_complex=True
+)
+
+
+@pytest.mark.parametrize("supported_dtype", list_of_supported_dtypes)
+def test_local_accessor(supported_dtype):
+    """A test for passing a LocalAccessor object as a kernel argument."""
+
+    @dpex_exp.kernel
+    def _kernel(nd_item: NdItem, a, slm):
+        i = nd_item.get_global_linear_id()
+        j = nd_item.get_local_linear_id()
+
+        slm[j] = 0
+        group_barrier(nd_item.get_group(), MemoryScope.WORK_GROUP)
+
+        for m in range(100):
+            slm[j] += i * m
+            group_barrier(nd_item.get_group(), MemoryScope.WORK_GROUP)
+
+        a[i] = slm[j]
+
+    N = 32
+    a = dpnp.empty(N, dtype=supported_dtype)
+    slm = LocalAccessor((32 * 64), dtype=a.dtype)
+
+    # A single work group with 32 work items is launched. Each work item
+    # computes the sum of (0..99) * its get_global_linear_id i.e.,
+    # `4950 * get_global_linear_id` and stores it into the work groups local
+    # memory. The local memory is of size 32*64 elements of the requested dtype.
+    # The result is then stored into `a` in global memory
+    dpex_exp.call_kernel(_kernel, dpex.NdRange((N,), (32,)), a, slm)
+
+    for idx in range(N):
+        assert a[idx] == 4950 * idx