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)

Reply via email to