Skip to content

Add more unit tests for GPU buffer #2978

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

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

maxrjones
Copy link
Member

This PR adds some tests for the GPU buffer prototype to unlock #2738.

@github-actions github-actions bot added the needs release notes Automatically applied to PRs which haven't added release notes label Apr 11, 2025
@maxrjones
Copy link
Member Author

@TomAugspurger would you expect the sharding codec to work with the gpu buffer prototype? If so, I could debug the failure more but don't want to spend time setting up an environment with cuda unnecessarily

@TomAugspurger
Copy link
Contributor

I haven't used or looked at the sharding code, so I don't know offhand sorry.

If setting up a dev end with cupy is a hassle I can take a look sometime.

Just looking at the traceback, maybe one thing to check: try setting with zarr.config.enable_gpu() in the tests? Maybe we're grabbing a buffer type from the default config somewhere where we should be basing it off another Buffer class?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
needs release notes Automatically applied to PRs which haven't added release notes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants