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.

Sign up for GitHub

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

Jump to bottom

[Python] Expand the the C Device Interface bindings to support CUDA device #40384

Closed
jorisvandenbossche opened this issue Mar 6, 2024 · 1 comment
Closed

[Python] Expand the the C Device Interface bindings to support CUDA device #40384

jorisvandenbossche opened this issue Mar 6, 2024 · 1 comment
Assignees
@jorisvandenbossche
Labels
Component: Python
Milestone
17.0.0

Comments

@jorisvandenbossche
Copy link
Member

jorisvandenbossche commented Mar 6, 2024 •
edited
Loading

Follow-up on #39979 which added _export_to_c_device/_import_from_c_device methods, but for now only for CPU devices. Expand this to support the CUDA device.
@github-actions github-actions bot mentioned this issue Mar 6, 2024
Merged
@jorisvandenbossche jorisvandenbossche added this to the 16.0.0 milestone Apr 1, 2024
@jorisvandenbossche jorisvandenbossche removed this from the 16.0.0 milestone Apr 11, 2024
jorisvandenbossche added a commit that referenced this issue Jun 19, 2024
@jorisvandenbossche
GH-40384: [Python] Expand the C Device Interface bindings to support …
89d6354 
…import on CUDA device (#40385)

### Rationale for this change

Follow-up on #39979 which added `_export_to_c_device`/`_import_from_c_device` methods, but for now only for CPU devices.

### What changes are included in this PR?

* Ensure `pyarrow.cuda` is imported before importing data through the C Interface, to ensure the CUDA device is registered
* Add tests for exporting/importing with the device interface on CUDA

### Are these changes tested?

Yes, added tests for CUDA.

* GitHub Issue: #40384

Authored-by: Joris Van den Bossche <[email protected]>
Signed-off-by: Joris Van den Bossche <[email protected]>
@jorisvandenbossche jorisvandenbossche added this to the 17.0.0 milestone Jun 19, 2024
@jorisvandenbossche
Copy link
Member Author

Issue resolved by pull request 40385
#40385

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@jorisvandenbossche jorisvandenbossche

Labels
Component: Python
Projects
None yet
Milestone
17.0.0
Development

No branches or pull requests
1 participant
@jorisvandenbossche