mpi4py + cupy

19 views
Skip to first unread message

永亮张

unread,
Sep 11, 2023, 8:21:18 AM9/11/23
to CuPy User Group
Hi,
I'm having trouble running the mpi4py cupy oriented sample code and would appreciate any help!
The sample code as following
```
from mpi4py import MPI
import cupy as cp

comm = MPI.COMM_WORLD
size = comm.Get_size()
rank = comm.Get_rank()

sendbuf = cp.arange(10, dtype='i')
recvbuf = cp.empty_like(sendbuf)
cp.cuda.get_current_stream().synchronize()
comm.Allreduce(sendbuf, recvbuf)

assert cp.allclose(recvbuf, sendbuf*size)
```
and I got the error:
YOUR APPLICATION TERMINATED WITH THE EXIT STRING: Segmentation fault (signal 11)
This typically refers to a problem with your application.
Please see the FAQ page for debugging suggestions

I want to konw how to figure it out
Thanke You
Reply all
Reply to author
Forward
0 new messages