Implements accumulation functions in dpctl.tensor#1602

Merged

ndgrigorian merged 36 commits intomasterfrom feature/tensor-accumulation

Apr 1, 2024

+3,405-171

Commits on Mar 21, 2024

Use shT instead of std::vector<py::ssize_t> in repeat
ndgrigorian
committed
Add missing host task to host_tasks_list in _reduction.py
ndgrigorian
committed

Commits on Mar 23, 2024

Commits on Mar 25, 2024

Pass host task vector to accumulator kernel calls
ndgrigorian
committed

Commits on Mar 26, 2024

Implements out keyword for accumulators
ndgrigorian
committed
Fixes cumulative_logsumexp when both an intermediate input and result temporary are needed
ndgrigorian
committed
Only permute dims of allocated outputs if accumulated axis is not the trailing axis
ndgrigorian
committed
Enable scalar inputs to accumulation functions
ndgrigorian
committed
Adds test for scalar inputs to cumulative_sum
ndgrigorian
committed
Adds docstrings for cumulative_sum. cumulative_prod, and cumulative_logsumexp
ndgrigorian
committed
Merge branch 'master' into feature/tensor-accumulation
ndgrigorian
committed
Removed redundant dtype kind check in _default_accumulation_dtype
ndgrigorian
committed
Reduce repetition of code allocation out array in _accumulate_common
ndgrigorian
committed

Commits on Mar 28, 2024

Commits on Apr 1, 2024