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

Merged

oleksandr-pavlyk merged 34 commits intomasterfrom feature/sorting

Jan 13, 2024

+3,388-140

Commits on Jan 8, 2024

Initial check in of sort/argsort per array API spec
oleksandr-pavlyk
committed
Added docstrings
oleksandr-pavlyk
committed
Added license header comments
oleksandr-pavlyk
committed
Corrected variable name used in assertion
oleksandr-pavlyk
committed
Ensure that we use no more SLM than is available
oleksandr-pavlyk
committed
Implementation of set functions
oleksandr-pavlyk
committed
Exposed set functions to dpctl.tensor namespace
oleksandr-pavlyk
committed
Added __all__ for _sorting.py
oleksandr-pavlyk
committed
Expand tests for sorting to improve coverage
oleksandr-pavlyk
committed
Tests for set functions
oleksandr-pavlyk
committed
Added license header for test file
oleksandr-pavlyk
committed
Do not use shorcut API for copy with dependency
oleksandr-pavlyk
committed
Implement set functions using synchronizing ops
oleksandr-pavlyk
committed
Renamed unused _set_functions.py and added it to omit list for coverage
oleksandr-pavlyk
committed
Use sequential sorting for small arrays
oleksandr-pavlyk
committed
Do not use shorthand queue::copy
oleksandr-pavlyk
committed
Do not use queue::fill convenience shortcut functions
oleksandr-pavlyk
committed
Accumulate implementations API acquired host_task_events argument
oleksandr-pavlyk
committed
Changed inclusive sum implementation from recursive to iterative
oleksandr-pavlyk
committed
Use async implementation of set functions
oleksandr-pavlyk
committed
Permit copying when flattening for unique funcs

ndgrigorian
authored and
oleksandr-pavlyk
committed
Fix typos in set functions

ndgrigorian
authored and
oleksandr-pavlyk
committed
Fixes to unique functions, adds tests

ndgrigorian
authored and
oleksandr-pavlyk
committed
Sorting, unique functions properly handle 0D arrays

ndgrigorian
authored and
oleksandr-pavlyk
committed
unique_all and unique_inverse inverse indices shape fixed

ndgrigorian
authored and
oleksandr-pavlyk
committed
Set stable to True for sorting functions by default

ndgrigorian
authored and
oleksandr-pavlyk
committed
Adds tests for 0D array cases for sorting

ndgrigorian
authored and
oleksandr-pavlyk
committed
Adds tests for unique functions on strided inputs

ndgrigorian
authored and
oleksandr-pavlyk
committed
Tests for strided 1D data input for unique functions

ndgrigorian
authored and
oleksandr-pavlyk
committed
Changes to _set_functions_sync.py for test suite to pass
oleksandr-pavlyk
committed
Removed sync implementation of set functions
oleksandr-pavlyk
committed
Renamed _set_functions_async -> _set_functions
oleksandr-pavlyk
committed

Commits on Jan 9, 2024

Use extended comparison operators to define weak order on real/complex FP types
oleksandr-pavlyk
committed
Adding tests for sorting of FP arrays with NaNs
oleksandr-pavlyk
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

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

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