def _array_to_cpu(array, stream): if array is None: return None if isinstance(array, ndarray): check_cuda_available() with get_device_from_array(array): return array.get(stream) return _cpu._array_to_cpu(array)