Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TST: Add tests for np.argsort #23846

Merged
merged 5 commits into from Jun 5, 2023
Merged

TST: Add tests for np.argsort #23846

merged 5 commits into from Jun 5, 2023

Conversation

r-devulap
Copy link
Member

@r-devulap r-devulap commented May 30, 2023

Contributing a few tests I had used when developing AVX-512 based argsort. Also added a release note.

numpy/core/tests/test_multiarray.py Outdated Show resolved Hide resolved
numpy/core/tests/test_multiarray.py Outdated Show resolved Hide resolved
r-devulap and others added 2 commits June 1, 2023 09:17
Copy link
Member

@seberg seberg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks everyone, looks good on a quick scan.

I took the liberty of removing the release snippet, I think Chuck is already on it (but may have missed some discussion).

@seberg seberg merged commit ea55ffa into numpy:main Jun 5, 2023
14 of 35 checks passed
@seberg seberg added the 09 - Backport-Candidate PRs tagged should be backported label Jun 5, 2023
charris pushed a commit to charris/numpy that referenced this pull request Jun 5, 2023
Contributing a few tests I had used when developing AVX-512 based argsort.

Co-authored-by: Robert Kern <robert.kern@gmail.com>
@charris
Copy link
Member

charris commented Jun 5, 2023

I think Chuck is already on it

Uh ... Anyway, I don't think a release note is needed for this :)

@charris charris removed the 09 - Backport-Candidate PRs tagged should be backported label Jun 5, 2023
@seberg
Copy link
Member

seberg commented Jun 5, 2023

That was the release note for the faster argsort:

Faster ``np.argsort`` on AVX-512 enabled processors
---------------------------------------------------
32-bit and 64-bit quicksort algorithm for np.argsort gain up to 6x speed up on
processors that support AVX-512 instruction set.

Thanks to `Intel corporation <https://open.intel.com/>`_ for sponsoring this
work.

charris added a commit that referenced this pull request Jun 6, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants