2024年4月3日发(作者：)

fortran中的mpi_gather的用法

MPI_Gather是Fortran中的一种MPI函数，用于将所有进程的

数据收集到一个进程中。具体用法如下：

MPI_Gather(sendbuf, sendcount, sendtype, recvbuf,

recvcount, recvtype, root, comm, ierror)

其中，sendbuf表示发送缓冲区的起始地址，sendcount表示每

个进程要发送的元素数，sendtype表示发送元素的类型。

recvbuf表示接收缓冲区的起始地址，recvcount表示每个进程

要接收的元素数，recvtype表示接收元素的类型。

root表示接收缓冲区所在的进程的编号，comm表示通信子，

ierror表示错误码。

使用MPI_Gather时，每个进程都会向root进程发送数据。root

进程会将所有数据收集到recvbuf中。由于每个进程发送的元素数可

能不同，因此在接收时，需要指定每个进程要接收的元素数

（recvcount）。如果接收缓冲区的大小不足以容纳所有数据，则可能

会发生错误。

下面是一个简单的例子，展示了MPI_Gather的用法：

program main

include 'mpif.h'

integer :: rank, size, ierror

integer, allocatable :: sendbuf(:), recvbuf(:)

integer :: sendcount, recvcount, root, i

- 1 -

call MPI_Init(ierror)

call MPI_Comm_rank(MPI_COMM_WORLD, rank, ierror)

call MPI_Comm_size(MPI_COMM_WORLD, size, ierror)

sendcount = 1

recvcount = size

root = 0

allocate(sendbuf(sendcount))

allocate(recvbuf(recvcount))

sendbuf = rank

call MPI_Gather(sendbuf, sendcount, MPI_INTEGER, &

recvbuf, recvcount, MPI_INTEGER, root, &

MPI_COMM_WORLD, ierror)

if (rank == root) then

do i = 1, size

write(*,*) recvbuf(i)

end do

end if

deallocate(sendbuf)

deallocate(recvbuf)

call MPI_Finalize(ierror)

end program

在上面的例子中，每个进程发送一个整数（即其rank），root进

- 2 -

程将所有整数收集到一个数组中，并输出。注意，发送和接收的元素

类型必须匹配。本例中的MPI_INTEGER表示整数类型。

- 3 -