Python API of addressing #587

Adds `dpctl.SyclDevice.global_mem_size` and
`dpctl.SyclDevice.local_mem_size`

Author: oleksandr-pavlyk

dpctl/_backend.pxd

@@ -21,7 +21,7 @@
 types defined by dpctl's C API.
 """
 
-from libc.stdint cimport int64_t, uint32_t, uint64_t
+from libc.stdint cimport int64_t, uint32_t
 from libcpp cimport bool
 
 
@@ -149,6 +149,8 @@ cdef extern from "dpctl_sycl_device_interface.h":
     cdef _backend_type DPCTLDevice_GetBackend(const DPCTLSyclDeviceRef)
     cdef _device_type DPCTLDevice_GetDeviceType(const DPCTLSyclDeviceRef)
     cdef const char *DPCTLDevice_GetDriverVersion(const DPCTLSyclDeviceRef DRef)
+    cdef size_t DPCTLDevice_GetGlobalMemSize(const DPCTLSyclDeviceRef DRef)
+    cdef size_t DPCTLDevice_GetLocalMemSize(const DPCTLSyclDeviceRef DRef)
     cdef uint32_t DPCTLDevice_GetMaxComputeUnits(const DPCTLSyclDeviceRef DRef)
     cdef uint32_t DPCTLDevice_GetMaxNumSubGroups(const DPCTLSyclDeviceRef DRef)
     cdef size_t DPCTLDevice_GetMaxWorkGroupSize(const DPCTLSyclDeviceRef DRef)
@@ -239,9 +241,9 @@ cdef extern from "dpctl_sycl_event_interface.h":
         size_t index)
     cdef DPCTLEventVectorRef DPCTLEvent_GetWaitList(
         DPCTLSyclEventRef ERef)
-    cdef uint64_t DPCTLEvent_GetProfilingInfoSubmit(DPCTLSyclEventRef ERef)
-    cdef uint64_t DPCTLEvent_GetProfilingInfoStart(DPCTLSyclEventRef ERef)
-    cdef uint64_t DPCTLEvent_GetProfilingInfoEnd(DPCTLSyclEventRef ERef)
+    cdef size_t DPCTLEvent_GetProfilingInfoSubmit(DPCTLSyclEventRef ERef)
+    cdef size_t DPCTLEvent_GetProfilingInfoStart(DPCTLSyclEventRef ERef)
+    cdef size_t DPCTLEvent_GetProfilingInfoEnd(DPCTLSyclEventRef ERef)
 
 
 cdef extern from "dpctl_sycl_kernel_interface.h":

dpctl/_sycl_device.pyx

@@ -34,11 +34,13 @@ from ._backend cimport (  # noqa: E211
     DPCTLDevice_GetBackend,
     DPCTLDevice_GetDeviceType,
     DPCTLDevice_GetDriverVersion,
+    DPCTLDevice_GetGlobalMemSize,
     DPCTLDevice_GetImage2dMaxHeight,
     DPCTLDevice_GetImage2dMaxWidth,
     DPCTLDevice_GetImage3dMaxDepth,
     DPCTLDevice_GetImage3dMaxHeight,
     DPCTLDevice_GetImage3dMaxWidth,
+    DPCTLDevice_GetLocalMemSize,
     DPCTLDevice_GetMaxComputeUnits,
     DPCTLDevice_GetMaxNumSubGroups,
     DPCTLDevice_GetMaxReadImageArgs,
@@ -656,6 +658,22 @@ cdef class SyclDevice(_SyclDevice):
         """
         return DPCTLDevice_GetPreferredVectorWidthHalf(self._device_ref)
 
+    @property
+    def global_mem_size(self):
+        """ Returns the size of global memory on this device in bytes.
+        """
+        cdef size_t global_mem_size = 0
+        global_mem_size = DPCTLDevice_GetGlobalMemSize(self._device_ref)
+        return global_mem_size
+
+    @property
+    def local_mem_size(self):
+        """ Returns the size of local memory on this device in bytes.
+        """
+        cdef size_t local_mem_size = 0
+        local_mem_size = DPCTLDevice_GetLocalMemSize(self._device_ref)
+        return local_mem_size
+
     @property
     def vendor(self):
         """ Returns the device vendor name as a string.

dpctl/tests/test_sycl_device.py

@@ -61,6 +61,16 @@ def check_get_max_compute_units(device):
     assert max_compute_units > 0
 
 
+def check_get_global_mem_size(device):
+    global_mem_size = device.global_mem_size
+    assert global_mem_size > 0
+
+
+def check_get_local_mem_size(device):
+    local_mem_size = device.local_mem_size
+    assert local_mem_size > 0
+
+
 def check_get_max_work_item_dims(device):
     max_work_item_dims = device.max_work_item_dims
     assert max_work_item_dims > 0
@@ -529,6 +539,8 @@ def check_repr(device):
     check_create_sub_devices_by_affinity_next_partitionable,
     check_print_device_info,
     check_repr,
+    check_get_global_mem_size,
+    check_get_local_mem_size,
 ]