Antoine Pitrou created ARROW-4212:
-------------------------------------
Summary: [Python] [CUDA] Creating a CUDA buffer from Numba device
array should be easier
Key: ARROW-4212
URL: https://issues.apache.org/jira/browse/ARROW-4212
Project: Apache Arrow
Issue Type: Improvement
Components: GPU, Python
Affects Versions: 0.11.1
Reporter: Antoine Pitrou
Currently, to create a CUDA buffer from a Numba device array, you have to write:
{code:python}
cuda.CudaBuffer.from_numba(device_arr.gpu_data)
{code}
It would be easier if you could just write:
{code}
cuda.CudaBuffer.from_numba(device_arr)
{code}
(ideally, any object exposing the [CUDA Array
Interface|https://numba.pydata.org/numba-doc/latest/cuda/cuda_array_interface.html]
would work)
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)