Files
ytts/venv/lib/python3.11/site-packages/numba/cuda/simulator/reduction.py
2025-04-02 21:44:17 -07:00

16 lines
300 B
Python

from functools import reduce as pyreduce
def Reduce(func):
def reduce_wrapper(seq, res=None, init=0):
r = pyreduce(func, seq, init)
if res is not None:
res[0] = r
return None
else:
return r
return reduce_wrapper
reduce = Reduce