Python device_array Examples

Example #1

File: dispatch.py Project: ArtShp/DataScience

 def device_array(self, shape, dtype, stream):
     if dgpu_present:
         return api.device_array(shape=shape, dtype=dtype)
     else:
         return np.empty(shape=shape, dtype=dtype)

Example #2

File: dispatch.py Project: cpcloud/numba

 def device_array(self, shape, dtype, stream):
     if dgpu_present:
         return api.device_array(shape=shape, dtype=dtype)
     else:
         return np.empty(shape=shape, dtype=dtype)