Dask added an additional keyword-only argument ``method``. method : {'sequential', 'blelloch'}, optional Choose which method to use to perform the cumsum. Default is 'sequential'. * 'sequential' performs the cumsum of each prior block before the current block. * 'blell
(x, axis, dtype=None, out=None, *, method="sequential")
| 215 | |
| 216 | @derived_from(np) |
| 217 | def nancumsum(x, axis, dtype=None, out=None, *, method="sequential"): |
| 218 | """Dask added an additional keyword-only argument ``method``. |
| 219 | |
| 220 | method : {'sequential', 'blelloch'}, optional |
| 221 | Choose which method to use to perform the cumsum. Default is 'sequential'. |
| 222 | |
| 223 | * 'sequential' performs the cumsum of each prior block before the current block. |
| 224 | * 'blelloch' is a work-efficient parallel cumsum. It exposes parallelism by |
| 225 | first taking the sum of each block and combines the sums via a binary tree. |
| 226 | This method may be faster or more memory efficient depending on workload, |
| 227 | scheduler, and hardware. More benchmarking is necessary. |
| 228 | """ |
| 229 | return cumreduction( |
| 230 | chunk.nancumsum, |
| 231 | operator.add, |
| 232 | 0, |
| 233 | x, |
| 234 | axis, |
| 235 | dtype, |
| 236 | out=out, |
| 237 | method=method, |
| 238 | preop=np.nansum, |
| 239 | ) |
| 240 | |
| 241 | |
| 242 | @derived_from(np) |
nothing calls this directly
no test coverage detected
searching dependent graphs…