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

Let user decide which cupy version to use #67

Merged

Conversation

relativityhd
Copy link
Contributor

Implement solution for #66 as discussed.

Remove cupy from dependencies and add three extras where each requires a different cupy version: cupy (building from source), and the two prebuild cupy-cuda12x and cupy-cuda11x

Closes #66

relativityhd and others added 3 commits November 24, 2024 15:15
Remove cupy from dependencies and add three extras where each requires a different cupy version: cupy (building from source), and the two prebuild cupy-cuda12x and cupy-cuda11x
Copy link
Collaborator

@jacobtomlinson jacobtomlinson left a comment

Choose a reason for hiding this comment

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

I felt like the general concensus on #66 was to remove the dependency altogether and add an import time check no?

relativityhd and others added 3 commits November 25, 2024 13:13
Co-authored-by: Wei Ji <[email protected]>
Co-authored-by: Wei Ji <[email protected]>
Co-authored-by: Wei Ji <[email protected]>
@relativityhd
Copy link
Contributor Author

@jacobtomlinson This would indeed remove the dependency and add an import time check. Additionally, this would add the possibility for the user to install cupy with a pip-extra.

@jacobtomlinson
Copy link
Collaborator

Sure, but I'm not sure the extras is a sustainable way to move forwards. I suspect there will be more turmoil around packaging as GPU packages continue to mature. I would recommend just linking to the cupy docs and leaving it at that. Otherwise things may get stale here quickly.

@relativityhd
Copy link
Contributor Author

Okay, I removed the possibility of installing cupy via extras.

Copy link
Collaborator

@jacobtomlinson jacobtomlinson left a comment

Choose a reason for hiding this comment

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

This looks good to me!

@dcherian dcherian requested a review from weiji14 November 26, 2024 17:51
Copy link
Member

@weiji14 weiji14 left a comment

Choose a reason for hiding this comment

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

Thanks @relativityhd! I might not get to a release this week, but will try to find time next month to do so.

@weiji14 weiji14 merged commit 3f84de0 into xarray-contrib:main Nov 27, 2024
4 checks passed
@relativityhd relativityhd deleted the feature/outsource-cupy-dependency branch November 27, 2024 15:40
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.

Let users decide which cupy to use
4 participants