使用单向通信但没有任何内容加载到目标中的排名总和

Question

我的目标是在类似于 allreduce 程序的环中创建排名总和，但使用单方面通信。 例如，如果这个系统中有四个进程。输出将是：

PE0:    Sum = 6
PE2:    Sum = 6
PE3:    Sum = 6
PE1:    Sum = 6

然而，对于我当前的单向通信解决方案，所有总和均为 0。 到目前为止我的代码：

#!/usr/bin/env python3
from mpi4py import MPI
import numpy as np

rcv_buf = np.empty((), dtype=np.intc) # uninitialized 0 dimensional integer array
status = MPI.Status()

comm_world = MPI.COMM_WORLD
my_rank = comm_world.Get_rank()
size = comm_world.Get_size()

right = (my_rank+1)      % size;
left  = (my_rank-1+size) % size;

snd_buf = np.array(my_rank, dtype=np.intc) # 0 dimensional integer array with 1 element initialized with the value of my_rank
sum = 0
copy = 0

# create a window
win = MPI.Win.Create(snd_buf, 1, MPI.INFO_NULL, comm=comm_world)

# we need a master process 

# sync remote get call
for i in range(size):
    win.Fence(0)
    win.Get(snd_buf, left, copy)
    win.Fence(0)
    
    sum += copy

win.Free()


print(f"PE{my_rank}:\tSum = {copy}")

我不确定如何检查 Get 调用是否正常工作，如果是，是否有其他方法来加载和存储。

Answer 1

我错误地使用了 win.Get 调用。在文档，Get 调用的第一个参数被指定为 origin (BufSpec)，我误以为窗口中的原始值是 snd_buf，但它应该是您希望存储答案的缓冲区。我还必须包含一个 Put 调用，将排名值发送到下一个进程。这使得最终代码：

#!/usr/bin/env python3
from mpi4py import MPI
import numpy as np

rcv_buf = np.empty((), dtype=np.intc) # uninitialized 0 dimensional integer array
status = MPI.Status()

comm_world = MPI.COMM_WORLD
my_rank = comm_world.Get_rank()
size = comm_world.Get_size()

right = (my_rank+1)      % size;
left  = (my_rank-1+size) % size;

snd_buf = np.array(my_rank, dtype=np.intc) # 0 dimensional integer array with 1 element initialized with the value of my_rank
sum = 0

# create a window
win = MPI.Win.Create(snd_buf, 1, MPI.INFO_NULL, comm=comm_world)


# sync remote get call
for i in range(size):
    win.Fence(0)
    win.Put(snd_buf, left)
    win.Fence(0)
    win.Get(rcv_buf, right)
    win.Fence(0)
    sum += rcv_buf


print(f"PE{my_rank}:\tSum = {sum}")


win.Free()