我正在尝试实现一个简单的MPI
程序,在其中我调用一个函数foo()
,在该函数上,-n
进程在n
数组周围传递,直到所有数组都传递给所有进程(n steps
)。该实现采用环形通信的形式,其中process#0
发送到process#2
并从process#n-1
接收,等等。这是我的代码:
#include <stdio.h>
#include <stdlib.h>
#include <mpi.h>
void foo(double* X, int n,int d,int k)
{
int id, world_size;
double *array = X;
double *array_buff = malloc(n*d*sizeof(double));
MPI_Comm_rank(MPI_COMM_WORLD, &id);
MPI_Comm_size(MPI_COMM_WORLD, &world_size);
MPI_Status status;
array[0] = id;
for(int step = 0 ; step < world_size ; step++)
{
int temp = id-step;
if(temp<0)
{
temp = world_size+temp;
}
if(array[0]!= temp)
{
printf("[%d][%d] %f (start)\n", id, step, array[0]);
}
//even processes send first while odd process receive
//first in order to avoid deadlock - achieve synchronization
MPI_Request reqsend, reqrecv;
int dst = (id+1)%world_size;
if(id%2 == 0)
{
int src = id-1;
if(id == 0)
{
src = world_size-1;
}
MPI_Isend(array, n*d, MPI_DOUBLE, dst, 0, MPI_COMM_WORLD, &reqsend);
MPI_Irecv(array_buff, n*d, MPI_DOUBLE, src, 0, MPI_COMM_WORLD, &reqrecv);
}
else
{
MPI_Irecv(array_buff, n*d, MPI_DOUBLE, id-1, 0, MPI_COMM_WORLD, &reqrecv);
MPI_Isend(array, n*d, MPI_DOUBLE, dst, 0, MPI_COMM_WORLD, &reqsend);
}
//..some array[] related work is been done here...
if(array[0]!=temp)
{
printf("[%d][%d] %f (end)\n", id, step, array[0]);
}
//update array asynchronously
MPI_Wait(&reqrecv, &status);
array = array_buff;
}
free(array);
}
为了避免死锁,偶数个进程先发送然后接收,而奇数个进程先接收然后发送。
初始化:array[0] = id;
旨在控制通讯是否正确完成。这就是为什么我在函数的开头使用两个打印,然后在函数的结尾使用两个打印,以观察array[]
是否在array = array_buff;
赋值(其中array[]
取缓冲值)发生之前更改内容。
我得到的输出之一是:
[0][2] 1.000000 (start)
[0][2] 1.000000 (end)
[0][3] 0.000000 (start)
[0][3] 0.000000 (end)
[1][3] 1.000000 (start)
[1][3] 1.000000 (end)
[3][1] 1.000000 (end)
[3][2] 0.000000 (end)
[3][3] 3.000000 (end)
为什么array[]
的内容在array_buff[]
分配之前被更改?
注:我已经阅读到在某些MPI版本中,在发送完成之前不能使用array[]
(Isend()
缓冲区)。但我认为情况并非如此,因为即使在array[]
段之后根本不使用Isend() Irecv()
,程序的行为也仍然相同。
我已经尝试使用step value标识符值来设置tags以匹配每个特定的发送,但是却出现了死锁,但我不知道为什么。我希望对此做出解释。tag = step;
问题是array = array_buff
。当array[]
在第一步中获得正确的值时,此后array_buff
在第二,第三等步骤中更新时,它也会更改array[]
的内容,因为它们指向相同的内存。
问题已解决:
for(int i=0; i<n*d ; i++)
{
array[i] = array_buff[i];
}