Implementation of sort/argsort and set functions per array API spec - #1483

Merged

oleksandr-pavlyk merged 34 commits into

masterfrom

feature/sorting

Jan 13, 2024

Commits on Jan 8, 2024

Initial check in of sort/argsort per array API spec
oleksandr-pavlyk
committed
c2d21d0
Added docstrings
oleksandr-pavlyk
committed
84437f1
Added license header comments
oleksandr-pavlyk
committed
8076adf
Corrected variable name used in assertion
oleksandr-pavlyk
committed
3cb7b15
Ensure that we use no more SLM than is available
oleksandr-pavlyk
committed
d79962d
Implementation of set functions
oleksandr-pavlyk
committed
aa1df8b
Exposed set functions to dpctl.tensor namespace
oleksandr-pavlyk
committed
709ed1e
Added __all__ for _sorting.py
oleksandr-pavlyk
committed
5635d85
Expand tests for sorting to improve coverage
oleksandr-pavlyk
committed
88ad12f
Tests for set functions
oleksandr-pavlyk
committed
734cf77
Added license header for test file
oleksandr-pavlyk
committed
11344b1
Do not use shorcut API for copy with dependency
oleksandr-pavlyk
committed
b1d1006
Implement set functions using synchronizing ops
oleksandr-pavlyk
committed
34b50af
Renamed unused _set_functions.py and added it to omit list for coverage
oleksandr-pavlyk
committed
1c7efc7
Use sequential sorting for small arrays
oleksandr-pavlyk
committed
fad8470
Do not use shorthand queue::copy
oleksandr-pavlyk
committed
fa3d8fe
Do not use queue::fill convenience shortcut functions
oleksandr-pavlyk
committed
abfc280
Accumulate implementations API acquired host_task_events argument
oleksandr-pavlyk
committed
5e0e5eb
Changed inclusive sum implementation from recursive to iterative
oleksandr-pavlyk
committed
6abb9fe
Use async implementation of set functions
oleksandr-pavlyk
committed
b149096
Permit copying when flattening for `unique` funcs
ndgrigorian
authored and
oleksandr-pavlyk
committed
eeb30f6
Fix typos in set functions
ndgrigorian
authored and
oleksandr-pavlyk
committed
0035558
Fixes to unique functions, adds tests
ndgrigorian
authored and
oleksandr-pavlyk
committed
a7e7625
Sorting, unique functions properly handle 0D arrays
ndgrigorian
authored and
oleksandr-pavlyk
committed
2fe8620
`unique_all` and `unique_inverse` inverse indices shape fixed
ndgrigorian
authored and
oleksandr-pavlyk
committed
8a6581a
Set `stable` to True for sorting functions by default
ndgrigorian
authored and
oleksandr-pavlyk
committed
82be69a
Adds tests for 0D array cases for sorting
ndgrigorian
authored and
oleksandr-pavlyk
committed
81610d6
Adds tests for `unique` functions on strided inputs
ndgrigorian
authored and
oleksandr-pavlyk
committed
b0cc135
Tests for strided 1D data input for `unique` functions
ndgrigorian
authored and
oleksandr-pavlyk
committed
8a6282f
Changes to _set_functions_sync.py for test suite to pass
oleksandr-pavlyk
committed
dd71e4e
Removed sync implementation of set functions
oleksandr-pavlyk
committed
b1b0af1
Renamed _set_functions_async -> _set_functions
oleksandr-pavlyk
committed
08e5dac

Navigation Menu

Uh oh!

Implementation of sort/argsort and set functions per array API spec - #1483

Implementation of sort/argsort and set functions per array API spec#1483
oleksandr-pavlyk merged 34 commits into
masterfrom
feature/sorting

Commits on Jan 8, 2024

Initial check in of sort/argsort per array API spec

Added docstrings

Added license header comments

Corrected variable name used in assertion

Ensure that we use no more SLM than is available

Implementation of set functions

Exposed set functions to dpctl.tensor namespace

Added all for _sorting.py

Expand tests for sorting to improve coverage

Tests for set functions

Added license header for test file

Do not use shorcut API for copy with dependency

Implement set functions using synchronizing ops

Renamed unused _set_functions.py and added it to omit list for coverage

Use sequential sorting for small arrays

Do not use shorthand queue::copy

Do not use queue::fill convenience shortcut functions

Accumulate implementations API acquired host_task_events argument

Changed inclusive sum implementation from recursive to iterative

Use async implementation of set functions

Permit copying when flattening for `unique` funcs

Fix typos in set functions

Fixes to unique functions, adds tests

Sorting, unique functions properly handle 0D arrays

`unique_all` and `unique_inverse` inverse indices shape fixed

Set `stable` to True for sorting functions by default

Adds tests for 0D array cases for sorting

Adds tests for `unique` functions on strided inputs

Tests for strided 1D data input for `unique` functions

Changes to _set_functions_sync.py for test suite to pass

Removed sync implementation of set functions

Renamed _set_functions_async -> _set_functions

Commits on Jan 9, 2024

Use extended comparison operators to define weak order on real/complex FP types

Adding tests for sorting of FP arrays with NaNs