Skip to content

Use CuPy for arrays & CUDA Python for low-level CUDA functions #4958

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

Merged
merged 8 commits into from
Mar 19, 2025

Conversation

jakirkham
Copy link
Member

@jakirkham jakirkham commented Mar 5, 2025

Switch to using CuPy for array creation when needed. Can still pass these to Numba or use __cuda_array_interface__ for low-level operations.

Also switch to CUDA Python for low-level CUDA functions calls in Python.

@jakirkham jakirkham requested a review from a team as a code owner March 5, 2025 00:48
@github-actions github-actions bot added the python label Mar 5, 2025
@jakirkham jakirkham added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change and removed python labels Mar 5, 2025
@github-actions github-actions bot added the python label Mar 5, 2025
@ChuckHastings
Copy link
Collaborator

@jakirkham - looks like there are ripple effects into the python testing.

@jakirkham jakirkham changed the title Use CuPy array for pos in force_atlas2 Use CuPy for arrays & CUDA Python for low-level CUDA functions Mar 8, 2025
Copy link
Contributor

Choose a reason for hiding this comment

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

Thanks. FYI this file will be going away in 25.06 due to support being added in pylibcugraph for 25.04. cc @jnke2016

@rlratzel
Copy link
Contributor

/merge

@rapids-bot rapids-bot bot merged commit d790d42 into rapidsai:branch-25.04 Mar 19, 2025
76 checks passed
@jakirkham jakirkham deleted the use_cp branch March 19, 2025 19:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement / enhancement to an existing function non-breaking Non-breaking change python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants