pyamgx icon indicating copy to clipboard operation
pyamgx copied to clipboard

Published 20 hours ago • verified publisher icon shwina

MPI pyamgx examples

Open tzahtila opened this issue 5 years ago • 4 comments

Hi there,

I was wondering if there were any examples MPI based examples on pyamgx?

Kind regards,

Tony

tzahtila avatar Jun 06 '19 05:06 tzahtila

Currently pyamgx does not support MPI, although that would be a welcome contribution!

shwina avatar Jun 21 '19 22:06 shwina

Hey Shwina,

Sorry to revive an old question but I was hoping to use pyamgx on a node with 4 GPUs.

My problem doesn't require communication between nodes, rather I was hoping for each corner to be launched onto a separate GPU.

Should this be possible? All that I see that is necessary is to get the ID of each GPU but in your examples I can't quite figure out how to do this.

tzahtila avatar Jan 13 '20 21:01 tzahtila

Hello @tzahtila - I apologize for the delay in responding. Could you please help me understand this use case further? Which AMGX C-API uses the GPU ID?

shwina avatar Jan 26 '20 21:01 shwina

Typically for things like this in the past, I've run Python with mpirun, and used mpi4py to get the MPI rank associated with each process. That MPI rank can also be used to calculate the GPU ID.

shwina avatar Jan 26 '20 21:01 shwina