[RFC] [DeviceType] Add support for MAIA (#120)

* Add support for MsNpu * Renamed MsNPU to MAIA
dmlc · Feb 5, 2024 · 4c10fec · 4c10fec
1 parent 2a7e9f1
commit 4c10fec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/apps/numpy_dlpack/dlpack/dlpack.py b/apps/numpy_dlpack/dlpack/dlpack.py
@@ -18,6 +18,7 @@ class DLDeviceType(ctypes.c_int):
     kDLROCMHost = 11
     kDLCUDAManaged = 13
     kDLOneAPI = 14
+    kDLMAIA = 17
 
     def __str__(self):
         return {
@@ -32,6 +33,7 @@ def __str__(self):
             self.kDLROCMHost: "ROMCHost",
             self.kDLCUDAManaged: "CUDAManaged",
             self.kDLOneAPI: "oneAPI",
+            self.kDLMAIA: "MAIA",
             }[self.value]
 
 

diff --git a/include/dlpack/dlpack.h b/include/dlpack/dlpack.h
@@ -116,6 +116,8 @@ typedef enum {
   kDLWebGPU = 15,
   /*! \brief Qualcomm Hexagon DSP */
   kDLHexagon = 16,
+  /*! \brief Microsoft MAIA devices */
+  kDLMAIA = 17,
 } DLDeviceType;
 
 /*!