Add support for MsNpu

dmlc · Mar 2, 2023 · 59e5218 · 59e5218
1 parent ca4d00a
commit 59e5218
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
+    kDLMsNpu = 17
 
     def __str__(self):
         return {
@@ -32,6 +33,7 @@ def __str__(self):
             self.kDLROCMHost: "ROMCHost",
             self.kDLCUDAManaged: "CUDAManaged",
             self.kDLOneAPI: "oneAPI",
+            self.kDLMsNpu: "MsNpu",
             }[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 NPU */
+  kDLMsNpu = 17,
 } DLDeviceType;
 
 /*!