Exemplos de device_array em Python

Exemplo n.º 1

Arquivo: dispatch.py Projeto: 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)

Exemplo n.º 2

Arquivo: dispatch.py Projeto: 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)