Python device_array示例

示例#1

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

示例#2

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