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)
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)