Skip to content

instantiating an Array from a GPU memory buffer without GPU mem -> system mem -> GPU mem roundtrip #20506

Answered by jakevdp
hrbigelow asked this question in Q&A
Discussion options

You must be logged in to vote

The primary method for zero-copy transfer of GPU buffers into JAX is via the dlpack data structure, using the jax.from_dlpack function.

If dlpack is not an option, JAX also has support for __cuda_array_interface__.

Replies: 1 comment 4 replies

Comment options

You must be logged in to vote
4 replies
@hrbigelow
Comment options

@jakevdp
Comment options

@hrbigelow
Comment options

@gszep
Comment options

Answer selected by hrbigelow
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants