Skip to content

Add CUDA IPC functions#294

Merged
LegNeato merged 1 commit intoRust-GPU:mainfrom
andfoy:add_ipc
Oct 27, 2025
Merged

Add CUDA IPC functions#294
LegNeato merged 1 commit intoRust-GPU:mainfrom
andfoy:add_ipc

Conversation

@andfoy
Copy link
Copy Markdown
Contributor

@andfoy andfoy commented Oct 22, 2025

This PR exposes the functions cudaIpcGetMemHandle, cudaIpcOpenMemHandle and cudaIpcCloseMemHandle, it also adds the definition for cudaIpcMemHandle_t, which are required to share device memory pointers between processes. This can be helpful for cross-language IPC access, e.g., creating a CuPy array in Python and then accessing the values as a tch::Tensor.

@LegNeato
Copy link
Copy Markdown
Contributor

Not familiar with this but looks fine, thank you!

@LegNeato LegNeato merged commit 05392f3 into Rust-GPU:main Oct 27, 2025
12 checks passed
@andfoy andfoy deleted the add_ipc branch October 28, 2025 05:03
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