Skip to content

Conversation

@btjanaka
Copy link
Member

@btjanaka btjanaka commented Jul 15, 2025

Description

This PR supports CuPy via the Array API. At first I thought this might be challenging since CuPy has a different model of devices, i.e., it depends on context managers rather than passing in devices. However, array-api-compat seems to have solved these issues and made CuPy compatible in terms of devices. Here are some references that I looked at:

TODO

  • Add cupy backend to conftest
  • Add instructions for installing cupy
  • Test ArrayStore with CuPy on GPU
  • Update ArrayStore to consider CuPy in pandas return type
  • Add cupy tests to CI -- I installed CuPy but the CuPy tests won't actually run because we don't have a GPU runner set up -- following scipy (https://github.com/scipy/scipy/blob/main/.github/workflows/gpu-ci.yml), I also set the CI to only run on ubuntu, since installing cupy on mac and windows seems to be problematic

Status

  • I have read the guidelines in
    CONTRIBUTING.md
  • I have formatted my code using yapf
  • I have tested my code by running pytest
  • I have linted my code with pylint
  • I have added a one-line description of my change to the changelog in
    HISTORY.md
  • This PR is ready to go

@btjanaka btjanaka merged commit 38fd31d into master Jul 15, 2025
17 checks passed
@btjanaka btjanaka deleted the cupy-array-store branch July 15, 2025 07:45
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.

2 participants