Python cuda_memcpyの例

コード例 #1

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

コード例 #2

ファイル: test_events.py プロジェクト: Sandy4321/quagga

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

コード例 #3

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()

コード例 #4

ファイル: test_events.py プロジェクト: Sandy4321/quagga

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()