Update dpctl Cython API for new DPCTLKernelArgType

Author: Diptorup Deb

dpctl/_backend.pxd

@@ -57,21 +57,16 @@ cdef extern from "syclinterface/dpctl_sycl_enum_types.h":
         _UNKNOWN_DEVICE 'DPCTL_UNKNOWN_DEVICE'
 
     ctypedef enum _arg_data_type 'DPCTLKernelArgType':
-        _CHAR               'DPCTL_CHAR',
-        _SIGNED_CHAR        'DPCTL_SIGNED_CHAR',
-        _UNSIGNED_CHAR      'DPCTL_UNSIGNED_CHAR',
-        _SHORT              'DPCTL_SHORT',
-        _INT                'DPCTL_INT',
-        _UNSIGNED_INT       'DPCTL_UNSIGNED_INT',
-        _UNSIGNED_INT8      'DPCTL_UNSIGNED_INT8',
-        _LONG               'DPCTL_LONG',
-        _UNSIGNED_LONG      'DPCTL_UNSIGNED_LONG',
-        _LONG_LONG          'DPCTL_LONG_LONG',
-        _UNSIGNED_LONG_LONG 'DPCTL_UNSIGNED_LONG_LONG',
-        _SIZE_T             'DPCTL_SIZE_T',
-        _FLOAT              'DPCTL_FLOAT',
-        _DOUBLE             'DPCTL_DOUBLE',
-        _LONG_DOUBLE        'DPCTL_DOUBLE',
+        _INT8_T             'DPCTL_INT8_T',
+        _UINT8_T            'DPCTL_UINT8_T',
+        _INT16_T            'DPCTL_INT16_T',
+        _UINT16_T           'DPCTL_UINT16_T',
+        _INT32_T            'DPCTL_INT32_T',
+        _UINT32_T           'DPCTL_UINT32_T',
+        _INT64_T            'DPCTL_INT64_T',
+        _UINT64_T           'DPCTL_UINT64_T',
+        _FLOAT              'DPCTL_FLOAT32_T',
+        _DOUBLE             'DPCTL_FLOAT64_T',
         _VOID_PTR           'DPCTL_VOID_PTR'
 
     ctypedef enum _queue_property_type 'DPCTLQueuePropertyType':

dpctl/_sycl_queue.pyx

@@ -631,34 +631,28 @@ cdef class SyclQueue(_SyclQueue):
         for idx, arg in enumerate(args):
             if isinstance(arg, ctypes.c_char):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._CHAR
-            elif isinstance(arg, ctypes.c_int):
+                kargty[idx] = _arg_data_type._INT8_T
+            elif isinstance(arg, ctypes.c_uint8):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._INT
-            elif isinstance(arg, ctypes.c_uint):
+                kargty[idx] = _arg_data_type._UINT8_T
+            elif isinstance(arg, ctypes.c_short):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._UNSIGNED_INT
-            elif isinstance(arg, ctypes.c_uint8):
+                kargty[idx] = _arg_data_type._INT16_T
+            elif isinstance(arg, ctypes.c_ushort):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._UNSIGNED_INT8
-            elif isinstance(arg, ctypes.c_long):
+                kargty[idx] = _arg_data_type._UINT16_T
+            elif isinstance(arg, ctypes.c_int):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._LONG
-            elif isinstance(arg, ctypes.c_ulong):
+                kargty[idx] = _arg_data_type._INT32_T
+            elif isinstance(arg, ctypes.c_uint):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._UNSIGNED_LONG
+                kargty[idx] = _arg_data_type._UINT32_T
             elif isinstance(arg, ctypes.c_longlong):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._LONG_LONG
+                kargty[idx] = _arg_data_type._INT64_T
             elif isinstance(arg, ctypes.c_ulonglong):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._UNSIGNED_LONG_LONG
-            elif isinstance(arg, ctypes.c_short):
-                kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._SHORT
-            elif isinstance(arg, ctypes.c_size_t):
-                kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
-                kargty[idx] = _arg_data_type._SIZE_T
+                kargty[idx] = _arg_data_type._UINT64_T
             elif isinstance(arg, ctypes.c_float):
                 kargs[idx] = <void*><size_t>(ctypes.addressof(arg))
                 kargty[idx] = _arg_data_type._FLOAT