-
Notifications
You must be signed in to change notification settings - Fork 236
Open
Labels
triageNeeds the team's attentionNeeds the team's attention
Description
Description
Add support for exporting StridedMemoryView objects via the DLPack protocol.
Requested Features
Both the following protocols should be supported:
-
__dlpack__/__dlpack_device__- The standard Python DLPack protocol for interoperability with frameworks like NumPy, PyTorch, JAX, CuPy, etc. -
__c_dlpack_exchange_api__- The C-level DLPack exchange API for efficient zero-copy data sharing at the C/Cython level.
Motivation
Currently, StridedMemoryView can consume DLPack-compatible objects, but it cannot export itself as DLPack tensors. Adding export support would enable it to be seamlessly passed to any library that accepts DLPack inputs, improving interoperability across the Python data science ecosystem.
Additional Context
- DLPack spec: https://dmlc.github.io/dlpack/latest/
__c_dlpack_exchange_api__: https://dmlc.github.io/dlpack/latest/c_api.html#c.DLPackExchangeAPI- Related to the ongoing discussions around
StridedMemoryViewdesign (RFC: RemoveStridedMemoryView(to somewhere else)? #1371)
Metadata
Metadata
Assignees
Labels
triageNeeds the team's attentionNeeds the team's attention