Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Appearance settings

ENH: Use openmp on x86-simd-sort to speed up np.sort and np.argsort #28619

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.

Sign up for GitHub

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

Merged
merged 10 commits into from
May 14, 2025
+53 −5
Merged

ENH: Use openmp on x86-simd-sort to speed up np.sort and np.argsort #28619

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
130cac5
Update x86-simd-sort module to latest
r-devulap Mar 31, 2025
7fd938b
BLD: Add openmp flags to build x86-simd-sort
r-devulap Mar 31, 2025
f8cfa4e
ENH: Update x86-simd-sort to port openmp support for argsort
r-devulap Apr 11, 2025
02c4728
Add meson option to toggle building with openMP
r-devulap Apr 14, 2025
21bc19f
TST: Add np.argsort test for openmp paths
r-devulap Apr 14, 2025
e425de8
CI: Add openmp flags to test openMP code paths
r-devulap Apr 14, 2025
ac59ea9
Update x86-simd-sort: detect already sorted arrays for np.argsort
r-devulap Apr 15, 2025
8ba425e
DOCS: add release notes
r-devulap Apr 16, 2025
e0f0247
Minor changes to meson.build
r-devulap Apr 16, 2025
6eff29e
Initialize omp to empty variable
r-devulap Apr 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
TST: Add np.argsort test for openmp paths
  • Loading branch information
@r-devulap
r-devulap committed May 1, 2025
commit 21bc19f1349f0bff978cac6dde80ea4db6374f83
11 changes: 10 additions & 1 deletion 11 numpy/_core/tests/test_multiarray.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10296,9 +10296,18 @@ def test_argsort_int(N, dtype):
@pytest.mark.parametrize("dtype", [np.float16, np.float32, np.float64])
def test_sort_largearrays(dtype):
N = 1000000
arr = np.random.rand(N)
rnd = np.random.RandomState(1100710816)
arr = -0.5 + rnd.random(N).astype(dtype)
assert_equal(np.sort(arr, kind='quick'), np.sort(arr, kind='heap'))

# Test large arrays that leverage openMP implementations from x86-simd-sort:
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
def test_argsort_largearrays(dtype):
N = 1000000
rnd = np.random.RandomState(1100710816)
arr = -0.5 + rnd.random(N).astype(dtype)
assert_arg_sorted(arr, np.argsort(arr, kind='quick'))

@pytest.mark.skipif(not HAS_REFCOUNT, reason="Python lacks refcounts")
def test_gh_22683():
b = 777.68760986
Expand Down
Morty Proxy This is a proxified and sanitized view of the page, visit original site.