Пример #1
0
 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
0
 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()