Check in of generic reduction templates and some reductions - #1399

Merged

ndgrigorian merged 27 commits into

feature/reductions-custom-ops

Oct 3, 2023

Commits on Sep 29, 2023

Implements necessary sycl utilities for custom reductions
ndgrigorian
committed
07ac2da
Implements dpctl.tensor.max and dpctl.tensor.min
ndgrigorian
committed
78f7aba
Adds tests for min and max
ndgrigorian
committed
41671ae
Reductions now set max_wg to the minimum of the max work group size and 2048
ndgrigorian
committed
093fcca
max and min nan propagation fixed for CPU devices
ndgrigorian
committed
82688ed
Tweak to test_reduction_kernels
ndgrigorian
committed
e5a39cf
Implements dpctl.tensor.argmax and argmin
ndgrigorian
committed
3af754c
Tests for argmin and argmax
ndgrigorian
committed
7052ad1
Argmin and argmax now handle identities correctly
ndgrigorian
committed
97efe7a
Replaced `std::min` with `idx_reduction_op_`
ndgrigorian
committed
7aef816
reductions now well-behaved for size-zero arrays
ndgrigorian
committed
6c3abcc
removed unnecessary copies in reduction templates
ndgrigorian
committed
a00ac58
Refactors sum to use generic reduction templates
ndgrigorian
committed
2468d8a
Sum now uses a generic Python API
ndgrigorian
committed
78829e7
Docstrings added for argmax, argmin, max, and min
ndgrigorian
committed
f01991b
Small reduction clean-ups
ndgrigorian
committed
8597300
Added test for argmin with keepdims=True
oleksandr-pavlyk
authored and
ndgrigorian
committed
2c18667
Added a test for raised errors in reductions
ndgrigorian
committed
24b54d7

Commits on Sep 30, 2023

Removed `void` overloads from reduction utilities
ndgrigorian
committed
df1c22f

Commits on Oct 1, 2023

Added missing include, Identity to use has_known_identity
oleksandr-pavlyk
committed
478b30c
Adding functor factories for product over axis
oleksandr-pavlyk
committed
0598416
Added Python API for _prod_over_axis
oleksandr-pavlyk
committed
ca0ff64

Commits on Oct 2, 2023

Commits on Oct 3, 2023

Corrected prod docstring
ndgrigorian
committed
60a8ad7
Merge pull request #1426 from IntelPython/implement-product
oleksandr-pavlyk
authored
04ec23b