Exemplos de open_ipc_array em Python

Exemplo n.º 1

 def the_work():
     dtype = np.dtype(np.intp)
     with cuda.open_ipc_array(handle,
                              shape=size // dtype.itemsize,
                              dtype=dtype) as darr:
         # copy the data to host
         return darr.copy_to_host()

Exemplo n.º 2

Arquivo: test_ipc.py Projeto: yuguen/numba

 def the_work():
     dtype = np.dtype(np.intp)
     with cuda.open_ipc_array(handle, shape=size // dtype.itemsize,
                              dtype=dtype) as darr:
         # copy the data to host
         return darr.copy_to_host()