Commit 94485e2
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
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
124 | 124 | | |
125 | 125 | | |
126 | 126 | | |
| 127 | + | |
| 128 | + | |
127 | 129 | | |
128 | 130 | | |
129 | 131 | | |
| |||
0 commit comments