Commit 25b3db7
Fix managed memory DLPack device type on buffer-side export paths
Update setup_dl_tensor_device() and Buffer.__dlpack_device__() to emit
kDLCUDAManaged for managed memory, closing the gap where the Buffer ->
DLPack capsule -> StridedMemoryView path still misclassified managed
buffers as kDLCUDAHost. Add cross-reference comments to keep the three
classification sites aligned.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>1 parent 0650f4b commit 25b3db7
File tree
3 files changed
+6
-2
lines changed- cuda_core/cuda/core
- _memory
3 files changed
+6
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
95 | 95 | | |
96 | 96 | | |
97 | 97 | | |
98 | | - | |
| 98 | + | |
| 99 | + | |
99 | 100 | | |
100 | 101 | | |
101 | 102 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
328 | 328 | | |
329 | 329 | | |
330 | 330 | | |
331 | | - | |
| 331 | + | |
| 332 | + | |
| 333 | + | |
332 | 334 | | |
333 | 335 | | |
334 | 336 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
607 | 607 | | |
608 | 608 | | |
609 | 609 | | |
| 610 | + | |
610 | 611 | | |
611 | 612 | | |
612 | 613 | | |
| |||
0 commit comments