Define LocalAccessor type to use to specify local accessor kernel arguments

LocalAccessor(ndim, elemental_type_str, dim0, dim1, dim2)

The elemental type can be one of the following:
 "i1", "u1", "i2", "u2", "i4", "u4", "i8", "u8", "f4", "f8"

Author: oleksandr-pavlyk

dpctl/_sycl_queue.pyx

@@ -59,6 +59,7 @@ from ._backend cimport (  # noqa: E211
     DPCTLWorkGroupMemory_Delete,
     _arg_data_type,
     _backend_type,
+    _md_local_accessor,
     _queue_property_type,
 )
 from .memory._memory cimport _Memory
@@ -125,6 +126,47 @@ cdef class kernel_arg_type_attribute:
         return self.attr_value
 
 
+cdef class LocalAccessor:
+    cdef _md_local_accessor lacc
+
+    def __cinit__(self, size_t ndim, str type, size_t dim0, size_t dim1, size_t dim2):
+       self.lacc.ndim = ndim
+       self.lacc.dim0 = dim0
+       self.lacc.dim1 = dim1
+       self.lacc.dim2 = dim2
+
+       if ndim < 1 or ndim > 3:
+           raise ValueError
+       if type == 'i1':
+           self.lacc.dpctl_type_id = _arg_data_type._INT8_T
+       elif type == 'u1':
+           self.lacc.dpctl_type_id = _arg_data_type._UINT8_T
+       elif type == 'i2':
+           self.lacc.dpctl_type_id = _arg_data_type._INT16_T
+       elif type == 'u2':
+           self.lacc.dpctl_type_id = _arg_data_type._UINT16_T
+       elif type == 'i4':
+           self.lacc.dpctl_type_id = _arg_data_type._INT32_T
+       elif type == 'u4':
+           self.lacc.dpctl_type_id = _arg_data_type._UINT32_T
+       elif type == 'i8':
+           self.lacc.dpctl_type_id = _arg_data_type._INT64_T
+       elif type == 'u8':
+           self.lacc.dpctl_type_id = _arg_data_type._UINT64_T
+       elif type == 'f4':
+           self.lacc.dpctl_type_id = _arg_data_type._FLOAT
+       elif type == 'f8':
+           self.lacc.dpctl_type_id = _arg_data_type._DOUBLE
+       else:
+           raise ValueError(f"Unrecornigzed type value: '{type}'")
+
+    def __repr__(self):
+        return "LocalAccessor(" + self.ndim + ")"
+
+    cdef size_t addressof(self):
+        return <size_t>&self.lacc
+
+
 cdef class _kernel_arg_type:
     """
     An enumeration of supported kernel argument types in
@@ -865,6 +907,9 @@ cdef class SyclQueue(_SyclQueue):
             elif isinstance(arg, WorkGroupMemory):
                 kargs[idx] = <void*>(<size_t>arg._ref)
                 kargty[idx] = _arg_data_type._WORK_GROUP_MEMORY
+            elif isinstance(arg, LocalAccessor):
+                kargs[idx] = <void*>((<LocalAccessor>arg).addressof())
+                kargty[idx] = _arg_data_type._LOCAL_ACCESSOR
             else:
                 ret = -1
         return ret