def cuda_array_from_list(a):
a = np.array(a, dtype=np.int32, order='F')
host_data = a.ctypes.data_as(ct.POINTER(ct.c_int))
elem_size = ct.sizeof(ct.c_int)
nbytes = a.size * elem_size
data = cudart.cuda_malloc(nbytes, ct.c_int)
cudart.cuda_memcpy(data, host_data, nbytes, 'default')
return data
def cuda_array_from_list(a):
a = np.array(a, dtype=np.int32, order='F')
host_data = a.ctypes.data_as(ct.POINTER(ct.c_int))
elem_size = ct.sizeof(ct.c_int)
nbytes = a.size * elem_size
data = cudart.cuda_malloc(nbytes, ct.c_int)
cudart.cuda_memcpy(data, host_data, nbytes, 'default')
return data
def list_from_cuda_array(a, n, release_memory=True):
c_int_p = ct.POINTER(ct.c_int)
host_array = (c_int_p * n)()
host_ptr = ct.cast(host_array, c_int_p)
elem_size = ct.sizeof(ct.c_int)
cudart.cuda_memcpy(host_ptr, a, n * elem_size, 'default')
if release_memory:
cudart.cuda_free(a)
a = np.ndarray(shape=(n, ), dtype=np.int32, buffer=host_array, order='F')
return a.tolist()
def list_from_cuda_array(a, n, release_memory=True):
c_int_p = ct.POINTER(ct.c_int)
host_array = (c_int_p * n)()
host_ptr = ct.cast(host_array, c_int_p)
elem_size = ct.sizeof(ct.c_int)
cudart.cuda_memcpy(host_ptr, a, n * elem_size, 'default')
if release_memory:
cudart.cuda_free(a)
a = np.ndarray(shape=(n, ), dtype=np.int32, buffer=host_array, order='F')
return a.tolist()
