3 回答

TA貢獻1876條經驗 獲得超5個贊
只是為了擴大Joel的觀點:
如果分配數組以使它們連續(這樣的話,C的“多維數組”不會自動給您:),則這樣做會容易得多。
int **alloc_2d_int(int rows, int cols) {
int *data = (int *)malloc(rows*cols*sizeof(int));
int **array= (int **)malloc(rows*sizeof(int*));
for (int i=0; i<rows; i++)
array[i] = &(data[cols*i]);
return array;
}
/*...*/
int **A;
/*...*/
A = alloc_2d_init(N,M);
然后,您可以使用以下命令發送和接收整個NxM陣列
MPI_Send(&(A[0][0]), N*M, MPI_INT, destination, tag, MPI_COMM_WORLD);
完成后,釋放內存
free(A[0]);
free(A);
另外,MPI_Recv是阻止接收,并且MPI_Send可以是阻止發送。根據Joel的觀點,一件事意味著您絕對不需要障礙。此外,這意味著,如果您具有上述的發送/接收模式,則可能陷入僵局-每個人都在發送,沒有人在接收。更安全的是:
if (myrank == 0) {
MPI_Send(&(A[0][0]), N*M, MPI_INT, 1, tagA, MPI_COMM_WORLD);
MPI_Recv(&(B[0][0]), N*M, MPI_INT, 1, tagB, MPI_COMM_WORLD, &status);
} else if (myrank == 1) {
MPI_Recv(&(A[0][0]), N*M, MPI_INT, 0, tagA, MPI_COMM_WORLD, &status);
MPI_Send(&(B[0][0]), N*M, MPI_INT, 0, tagB, MPI_COMM_WORLD);
}
另一種更通用的方法是使用MPI_Sendrecv:
int *sendptr, *recvptr;
int neigh = MPI_PROC_NULL;
if (myrank == 0) {
sendptr = &(A[0][0]);
recvptr = &(B[0][0]);
neigh = 1;
} else {
sendptr = &(B[0][0]);
recvptr = &(A[0][0]);
neigh = 0;
}
MPI_Sendrecv(sendptr, N*M, MPI_INT, neigh, tagA, recvptr, N*M, MPI_INT, neigh, tagB, MPI_COMM_WORLD, &status);
或非阻塞發送和/或接收。

TA貢獻1784條經驗 獲得超9個贊
C的多維數組是一個連續的內存塊;如果您分配double d[50][50];,那是一個連續的內存塊。處理動態分配的“偽2D”數組的一種方法是分配一個指針塊和一組單獨的(通常是不連續的)存儲塊,每行一個。
- 3 回答
- 0 關注
- 635 瀏覽
添加回答
舉報