Python MPI广播排

Question

因此，我正在尝试将一排Numpy阵列广播到每个节点，并在其自己的矩阵中将每个节点更新该特定行更新。

因此，代码如下：

comm.Bcast([A[i, i:], A[i, i:].size, MPI.DOUBLE], root = root)

其中a的类型为numpy.Array，其中i是迭代变量。 当我尝试这样做时，我以a成为无结束。

做到这一点的正确方法是什么？

Answer 1

这需要一些数字骗局。您需要使用缓冲区的.data，然后将其变成带有偏移量的新的Numpy缓冲区。这应该接近您想要的内容：

datatype = np.intc
elementsize = datatype().itemsize
## or np.dtype('intc').itemsize
typechar = datatype().dtype.char
if procid==0:
    print("int size:",elementsize)
    print(typechar)

buffer = np.zeros( [nprocs,nprocs], dtype=datatype)
buffer[:,:] = -1
for proc in range(nprocs):
    if procid==proc:
        buffer[proc,:] = proc
    comm.Bcast\
        ( [ np.frombuffer\
            ( buffer.data,
              dtype=datatype,
              offset=(proc*nprocs+proc)*elementsize ),
            nprocs-proc, typechar ],
          root=proc )
    if procid==nprocs-1:
        print(buffer)

int size: 4
i
[[ 0  0  0  0]
 [-1 -1 -1 -1]
 [-1 -1 -1 -1]
 [-1 -1 -1 -1]]
[[ 0  0  0  0]
 [-1  1  1  1]
 [-1 -1 -1 -1]
 [-1 -1 -1 -1]]
[[ 0  0  0  0]
 [-1  1  1  1]
 [-1 -1  2  2]
 [-1 -1 -1 -1]]
[[ 0  0  0  0]
 [-1  1  1  1]
 [-1 -1  2  2]
 [ 3  3  3  3]]

请注意，这需要您选择的元素是连续的。对于发送一列，您需要派生类型。