cupy/cupy

Helper class for IPC memory handling?

Offen

#8.029 geöffnet am 13.12.2023

 (5 Kommentare) (2 Reaktionen) (0 zugewiesene Personen)Python (730 Forks)batch import
cat:featurecontribution welcomegood first issueprio:medium

Repository-Metriken

Stars
 (7.381 Sterne)
PR-Merge-Metriken
 (Durchschn. Merge 7T 18h) (64 gemergte PRs in 30 T)

Beschreibung

Regarding #8027 and the recent question on Gitter/Matrix, I'm wondering whether it will make sense to provide the feature.

The initial idea: allow users to do a_ipc = cupy.arange(10).ipc(). Users will get a_ipc, an instance of a serializable class that holds array information (shape, strides, dtype) & IPC handle, and can be passed to other processes (e.g. via multiprocessing).

c.f. https://numba.pydata.org/numba-doc/dev/cuda/ipc.html

Contributor Guide