Python open_ipc_array示例

示例#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()

示例#2

文件： test_ipc.py 项目： 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()