Implement HIP backend

Author: ndgrigorian

dpctl/_backend.pxd

@@ -43,6 +43,7 @@ cdef extern from "syclinterface/dpctl_sycl_enum_types.h":
     ctypedef enum _backend_type 'DPCTLSyclBackendType':
         _ALL_BACKENDS    'DPCTL_ALL_BACKENDS'
         _CUDA            'DPCTL_CUDA'
+        _HIP             'DPCTL_HIP'
         _LEVEL_ZERO      'DPCTL_LEVEL_ZERO'
         _OPENCL          'DPCTL_OPENCL'
         _UNKNOWN_BACKEND 'DPCTL_UNKNOWN_BACKEND'

dpctl/_sycl_device.pyx

@@ -172,6 +172,8 @@ cdef list _get_devices(DPCTLDeviceVectorRef DVRef):
 cdef str _backend_type_to_filter_string_part(_backend_type BTy):
     if BTy == _backend_type._CUDA:
         return "cuda"
+    elif BTy == _backend_type._HIP:
+        return "hip"
     elif BTy == _backend_type._LEVEL_ZERO:
         return "level_zero"
     elif BTy == _backend_type._OPENCL:
@@ -425,6 +427,8 @@ cdef class SyclDevice(_SyclDevice):
         )
         if BTy == _backend_type._CUDA:
             return backend_type.cuda
+        elif BTy == _backend_type._HIP:
+            return backend_type.hip
         elif BTy == _backend_type._LEVEL_ZERO:
             return backend_type.level_zero
         elif BTy == _backend_type._OPENCL:

dpctl/_sycl_device_factory.pyx

@@ -71,6 +71,8 @@ cdef _backend_type _string_to_dpctl_sycl_backend_ty(str backend_str):
         return _backend_type._ALL_BACKENDS
     elif backend_str == "cuda":
         return _backend_type._CUDA
+    elif backend_str == "hip":
+        return _backend_type._HIP
     elif backend_str == "level_zero":
         return _backend_type._LEVEL_ZERO
     elif backend_str == "opencl":
@@ -100,6 +102,8 @@ cdef _device_type _string_to_dpctl_sycl_device_ty(str dty_str):
 cdef _backend_type _enum_to_dpctl_sycl_backend_ty(BTy):
     if BTy == backend_type.all:
         return _backend_type._ALL_BACKENDS
+    elif BTy == backend_type.hip:
+        return _backend_type._HIP
     elif BTy == backend_type.cuda:
         return _backend_type._CUDA
     elif BTy == backend_type.level_zero:
@@ -153,7 +157,7 @@ cpdef list get_devices(backend=backend_type.all, device_type=device_type_t.all):
         backend (optional):
             A :class:`dpctl.backend_type` enum value or a string that
             specifies a SYCL backend. Currently, accepted values are: "cuda",
-            "opencl", "level_zero", or "all".
+            "hip", "opencl", "level_zero", or "all".
             Default: ``dpctl.backend_type.all``.
         device_type (optional):
             A :class:`dpctl.device_type` enum value or a string that
@@ -209,7 +213,7 @@ cpdef int get_num_devices(
         backend (optional):
             A :class:`dpctl.backend_type` enum value or a string that
             specifies a SYCL backend. Currently, accepted values are: "cuda",
-            "opencl", "level_zero", or "all".
+            "hip", "opencl", "level_zero", or "all".
             Default: ``dpctl.backend_type.all``.
         device_type (optional):
             A :class:`dpctl.device_type` enum value or a string that

dpctl/_sycl_event.pyx

@@ -333,6 +333,8 @@ cdef class SyclEvent(_SyclEvent):
             return backend_type.level_zero
         elif BE == _backend_type._CUDA:
             return backend_type.cuda
+        elif BE == _backend_type._HIP:
+            return backend_type.hip
         else:
             raise ValueError("Unknown backend type.")
 

dpctl/_sycl_platform.pyx

@@ -294,6 +294,8 @@ cdef class SyclPlatform(_SyclPlatform):
         )
         if BTy == _backend_type._CUDA:
             return backend_type.cuda
+        elif BTy == _backend_type._HIP:
+            return backend_type.hip
         elif BTy == _backend_type._LEVEL_ZERO:
             return backend_type.level_zero
         elif BTy == _backend_type._OPENCL:

dpctl/_sycl_queue.pyx

@@ -886,6 +886,8 @@ cdef class SyclQueue(_SyclQueue):
             return backend_type.level_zero
         elif BE == _backend_type._CUDA:
             return backend_type.cuda
+        elif BE == _backend_type._HIP:
+            return backend_type.hip
         else:
             raise ValueError("Unknown backend type.")
 

dpctl/enum_types.py

@@ -69,6 +69,7 @@ class backend_type(Enum):
 
     all = auto()
     cuda = auto()
+    hip = auto()
     level_zero = auto()
     opencl = auto()
 

dpctl/tests/test_sycl_device_factory.py

@@ -60,6 +60,8 @@ def string_to_device_type(dty_str):
 def string_to_backend_type(bty_str):
     if bty_str == "cuda":
         return bty.cuda
+    elif bty_str == "hip":
+        return bty.hip
     elif bty_str == "host":
         return bty.host
     elif bty_str == "level_zero":

libsyclinterface/helper/source/dpctl_utils_helper.cpp

@@ -94,6 +94,8 @@ backend DPCTL_DPCTLBackendTypeToSyclBackend(DPCTLSyclBackendType BeTy)
         return backend::opencl;
     case DPCTLSyclBackendType::DPCTL_ALL_BACKENDS:
         return backend::all;
+    case DPCTLSyclBackendType::DPCTL_HIP:
+        return backend::ext_oneapi_hip;
     default:
         throw std::runtime_error("Unsupported backend type");
     }
@@ -108,6 +110,8 @@ DPCTLSyclBackendType DPCTL_SyclBackendToDPCTLBackendType(backend B)
         return DPCTLSyclBackendType::DPCTL_LEVEL_ZERO;
     case backend::opencl:
         return DPCTLSyclBackendType::DPCTL_OPENCL;
+    case backend::ext_oneapi_hip:
+        return DPCTLSyclBackendType::DPCTL_HIP;
     default:
         return DPCTLSyclBackendType::DPCTL_UNKNOWN_BACKEND;
     }
@@ -467,6 +471,9 @@ std::string DPCTL_GetDeviceFilterString(const device &Device)
     case backend::opencl:
         ss << "opencl";
         break;
+    case backend::ext_oneapi_hip:
+        ss << "hip";
+        break;
     default:
         ss << "unknown";
     };

libsyclinterface/include/syclinterface/dpctl_sycl_enum_types.h

@@ -54,7 +54,8 @@ typedef enum
     DPCTL_LEVEL_ZERO      = 1 << 17,
     DPCTL_OPENCL          = 1 << 18,
     DPCTL_UNKNOWN_BACKEND = 0,
-    DPCTL_ALL_BACKENDS    = ((1<<5)-1) << 16
+    DPCTL_ALL_BACKENDS    = ((1<<5)-1) << 16,
+    DPCTL_HIP             = 1 << 19,
     // clang-format on
 } DPCTLSyclBackendType;