Esempi in Python per device_array

Esempio n. 1

File: dispatch.py Progetto: 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)

Esempio n. 2

File: dispatch.py Progetto: 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)