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.

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

Q: preferred way to implement GPU communication #12452

Open
angainor opened this issue Apr 4, 2024 · 0 comments
Open

Q: preferred way to implement GPU communication #12452

angainor opened this issue Apr 4, 2024 · 0 comments
Assignees
Labels

Comments

@angainor
Copy link

angainor commented Apr 4, 2024

I am implementing a communication pattern where GPUs exchange parts of their local data vector. The exchanged vector entries are 'unstructured' (arbitrary indices) with block size of ~8KB: for each communication index the sender GPU sends ~1k contiguous doubles.

I implemented this pattern with MPI_Type_indexed + MPI_Isend and it works. My question is, which of the following implementations is expected to be most efficient:

  • use MPI_Isend directly, without packing, with the newly defined indexed type (I guess OpenMPI allocates some internal buffer?)
  • use MPI_Pack with the newly defined type, copying data to a GPU buffer, then use MPI_Isend on the packed buffer
  • use MPI_Pack with the newly defined type, copying data to a CPU buffer, then use MPI_Isend on the packed buffer

Or is there any other, better way to implement such scenario?

Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

3 participants