

def allocate_outputs(self, schedule, outdtype):
    if (self._need_device_conversion or (self.output is None)):
        retval = self.device_array(shape=schedule.output_shapes[0], dtype=outdtype)
    else:
        retval = self.output
    self.kernel_returnvalue = retval

dataset/ETHPy150Open numba/numba/numba/npyufunc/deviceufunc.py GUFuncCallSteps.allocate_outputs/original