如何将动态数组从从节点发送到主节点

时间:2014-08-04 02:46:33

标签: c++ mpi openmpi

我完成了一个简单的MPI计划,并且我在项目的最后部分挣扎。 我将包含起点和终点的2 int发送到从节点。使用这些我需要创建一个数组并填充它。我需要将它发送回主节点。下面的奴隶代码:

printf("Client waiting for start point and endpoint array\n");fflush(stdout);

int startEnd [2];

MPI_Recv(startEnd, 2, MPI_INT, 0, 100, MPI_COMM_WORLD, &status);
int end = startEnd[1];
int start = startEnd[0];
printf("Recieved Start End of %d \t %d\n", startEnd[0], startEnd[1]);fflush(stdout);

unsigned char TargetHash[MAX_HASH_LEN];
MPI_Recv(TargetHash, MAX_HASH_LEN, MPI_CHAR, 0, 100, MPI_COMM_WORLD, &status);

int sizeToCompute = (end - start);
uint64* pStartPosIndexE = new uint64[sizeToCompute];
int iterator = 0;
for (int nPos = end; nPos >= start; nPos--)
{
    cwc.SetHash(TargetHash);
    cwc.HashToIndex(nPos);
    int i;
    for (i = nPos + 1; i <= cwc.GetRainbowChainLength() - 2; i++)
    {
        cwc.IndexToPlain();
        cwc.PlainToHash();
        cwc.HashToIndex(i);
    }

    pStartPosIndexE[iterator] = cwc.GetIndex();                 
}

这是创建动态长度数组的正确方法吗?如何将此数组发送回主节点?

1 个答案:

答案 0 :(得分:3)

发送动态分配的数组与发送静态数组没什么不同。当数组大小变化时,接收代码会变得更复杂,但不会那么复杂:

// ---------- Sender code ----------

MPI_Send(pStartPosIndexE, sizeToCompute, MPI_UINT64, 99, ...);

// --------- Receiver code ---------

// Wait for a message with tag 99
MPI_Status status;
MPI_Probe(MPI_ANY_SOURCE, 99, MPI_COMM_WORLD, &status);

// Get the number of elements in the message
int nElems;
MPI_Get_elements(&status, MPI_UINT64_T, &nElems);

// Allocate buffer of appropriate size
uint64 *result = new uint64[nElems];

// Receive the message
MPI_Recv(result, nElems, MPI_UINT64_T, status.MPI_SOURCE, 99, ...);

使用源代码为MPI_Probe的{​​{1}}通常在主/工作者应用程序中进行,其中工作人员以先到先得的方式处理。