Inconsistent lifetime management across `from_handle` / `from_*` APIs
@Andy-Jost arbeitet bereits daran.
Seit 01.5.2026.
- Vorherrschende Sprache
- Cython
- Sterne
- 3.4k
- Forks
- 329
- Ø Merge
- 1 T. 23 Std.
- Gemergte PRs (30 T.)
- 116
Beschreibung
Summary
Audit of all public from_handle / from_* methods in cuda.core reveals several inconsistencies in how lifetime management is exposed to users. These should be resolved before the 1.0 stable release.
Inventory
| Class | Method | Ownership Model |
|---|---|---|
Stream |
from_handle(handle) |
Always non-owning (borrowed) |
Kernel |
from_handle(handle, mod=) |
Non-owning ref; optional _keepalive holds ObjectCode |
Buffer |
from_handle(ptr, size, mr=, owner=) |
3 modes: non-owning / mr-deallocates / owner-ref |
Event |
from_ipc_descriptor(...) |
Owns imported event |
Buffer |
from_ipc_descriptor(...) |
Owns imported buffer |
GraphicsResource |
from_gl_buffer(...) / from_gl_image(...) |
Owns GL registration |
DeviceMemoryResource |
from_allocation_handle(...) / from_registry(uuid) |
Owns imported pool / retrieves existing |
PinnedMemoryResource |
from_allocation_handle(...) / from_registry(uuid) |
Owns imported pool / retrieves existing |
ObjectCode |
from_cubin/ptx/ltoir/fatbin/object/library() |
Owns module data |
StridedMemoryView |
from_dlpack/cai/array_interface/buffer/any_interface() |
Always non-owning (view) |
Inconsistencies
1. Stream.from_handle cannot opt into keeping an owner alive
Buffer.from_handleacceptsowner=to hold a Python reference and prevent GC of the foreign allocator.Kernel.from_handleacceptsmod=to hold a reference and prevent GC of the parentObjectCode/library.Stream.from_handlehas no equivalent parameter. It creates a borrowed stream with no way for the user to tie lifetime to an owner object.
The internal machinery already supports this — create_stream_handle_with_owner() in C++ accepts a Python owner and does Py_INCREF/DECREF. But from_handle() only passes the ephemeral _stream_holder wrapper as the owner, not the user's actual owner object.
Suggestion: Add an owner parameter to Stream.from_handle, consistent with Buffer.from_handle.
2. Event has no public from_handle()
Stream,Kernel, andBufferall expose publicfrom_handle()for wrapping foreign CUDA handles.Eventonly hasfrom_ipc_descriptor(). There is an internal_from_handle()(cdef), but no public way to wrap a foreignCUevent.
Suggestion: Add Event.from_handle(handle) for symmetry, with the same borrowed/non-owning semantics.
3. Kernel.from_handle docstring doesn't clearly document ownership
Stream.from_handledocstring explicitly says: "Stream lifetime is not managed, foreign object must remain alive while this stream is active."Buffer.from_handledocstring explicitly documents all three ownership modes and when the pointer is/isn't freed.Kernel.from_handledocstring mentionsmod"provides library lifetime for foreign kernels" but doesn't clearly state that the kernel handle itself is non-owning and won't be destroyed on close/GC.
Suggestion: Improve the Kernel.from_handle docstring to match the clarity of the other two.
Additional observation
All from_handle() methods use @staticmethod, while from_ipc_descriptor() / from_gl_*() / from_allocation_handle() use @classmethod. This may be intentional (the latter group uses cls for subclassing support), but worth confirming as a deliberate convention and documenting if so.
-- Leo's bot
Beitragsleitfaden
Erste Schritte
- Lies das ganze Issue und danach den Beitragsleitfaden des Projekts.
- Schreib ins Issue, dass du es übernimmst — das erspart doppelte Arbeit.
- Forke das Repository und arbeite in einem Branch.
- Öffne einen Pull Request, der die Issue-Nummer nennt.
Bewertung
Dieses Issue wurde noch nicht bewertet.