Skip to content

Commit 94485e2

Browse files
authored
Add kDLAscend for Huawei Ascend NPU device memory (#189)
This can be used to exchange buffers between Ascend integrations such as torch_npu, TVM-FFI, and TileLang without overloading kDLExtDev. kDLExtDev is shared by unrelated extension backends and cannot uniquely identify Ascend device memory.
1 parent 77aafa4 commit 94485e2

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

include/dlpack/dlpack.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,8 @@ typedef enum {
124124
kDLTPU = 19,
125125
/*! \brief Google TPU Host device (pinned memory) */
126126
kDLTPUHost = 20,
127+
/*! \brief Huawei Ascend NPU device */
128+
kDLAscend = 21,
127129
} DLDeviceType;
128130

129131
/*!

0 commit comments

Comments
 (0)