Skip to content

feat: Add ak.sort() for CUDA backend#3750

Merged
ianna merged 4 commits intoscikit-hep:mainfrom
shwina:add-cuda-backend-sort
Jan 6, 2026
Merged

feat: Add ak.sort() for CUDA backend#3750
ianna merged 4 commits intoscikit-hep:mainfrom
shwina:add-cuda-backend-sort

Conversation

@shwina
Copy link
Copy Markdown
Contributor

@shwina shwina commented Nov 26, 2025

Closes #3749.

Following the study #3748, this PR adds support for ak.sort() for the CUDA backend, using https://nvidia.github.io/cccl/python/compute.html.

⚠️ This PR should not be merged in its current state - I'm opening it as a POC at this time.

@codecov
Copy link
Copy Markdown

codecov bot commented Nov 26, 2025

Codecov Report

❌ Patch coverage is 16.00000% with 42 lines in your changes missing coverage. Please review.
✅ Project coverage is 82.64%. Comparing base (c7e21ea) to head (b6d0031).
⚠️ Report is 1 commits behind head on main.

Files with missing lines Patch % Lines
src/awkward/_connect/cuda/_compute.py 0.00% 24 Missing ⚠️
src/awkward/_backends/cupy.py 22.22% 14 Missing ⚠️
src/awkward/_kernels.py 50.00% 4 Missing ⚠️
Additional details and impacted files
Files with missing lines Coverage Δ
src/awkward/_kernels.py 69.78% <50.00%> (-1.43%) ⬇️
src/awkward/_backends/cupy.py 47.61% <22.22%> (-19.05%) ⬇️
src/awkward/_connect/cuda/_compute.py 0.00% <0.00%> (ø)

... and 2 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@shwina shwina changed the title Add ak.sort() for CUDA backend feat: Add ak.sort() for CUDA backend Nov 26, 2025
@github-actions
Copy link
Copy Markdown

The documentation preview is ready to be viewed at http://preview.awkward-array.org.s3-website.us-east-1.amazonaws.com/PR3750

Copy link
Copy Markdown
Member

@ianna ianna left a comment

Choose a reason for hiding this comment

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

@shwina - this is great! Thanks for implementing it. Let’s keep it open and discuss.

@shwina shwina force-pushed the add-cuda-backend-sort branch from 173b048 to 8fbd26f Compare December 1, 2025 19:19
@shwina shwina force-pushed the add-cuda-backend-sort branch from 8fbd26f to f434568 Compare December 4, 2025 11:34
@ianna ianna merged commit 03905ba into scikit-hep:main Jan 6, 2026
39 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Missing ak.sort() for CUDA backend

5 participants