MPI_GATHERV处理不同大小数组的机制及调用报错问题排查
MPI_Gatherv数组合并问题解答
核心错误原因
初始版本调用MPI_Gatherv时仅传入了接收计数数组,未提供正确的位移数组(recvdispls参数),仅将位移参数填为0,导致MPI无法确定各进程发送数据在接收缓冲区中的存储位置,所有进程的数据都会从接收缓冲区的起始位置开始写入,出现数据覆盖、顺序混乱的问题,严重时会因为内存越界触发段错误。
MPI_Gatherv root端正确调用逻辑
要实现不等长数组的收集,root进程需要提前准备两个长度等于通信域进程总数的数组:
- 接收计数数组recvcounts:存储每个进程发送的元素个数,你通过
MPI_Gather收集各进程本地数组长度Ilocalcnt得到的Icntarr就属于这类数组。 - 位移数组recvdispls:存储每个进程的数据在接收缓冲区中的起始偏移量,以元素为单位,要求从0开始计数。你后续编写的累加计算逻辑完全正确:
Istartdis=0 DO Idisp=1,size Idisarray(Idisp)=Istartdis Istartdis=Istartdis+Icntarr(Idisp) ENDDO
由于Fortran数组默认1起始,而MPI位移从0计数,二者刚好适配:传入接收缓冲区F的起始地址对应F(1),位移0对应F(1),偏移Icntarr(1)刚好对应第二个进程数据的起始位置F(Icntarr(1)+1),不会出现索引错位。
其他说明
非root进程调用MPI_Gatherv时,接收缓冲区、接收计数数组、位移数组传入NULL的写法是正确的,无需额外处理。
内容的提问来源于stack exchange,提问作者user6872802
相关产品推荐
相关产品推荐

