Можете ли вы транспонировать массив при отправке с использованием MPI_Type_create_subarray?

Я пытаюсь транспонировать матрицу, используя MPI в C. Каждый процесс имеет квадратная подматрица, и я хочу отправить ее в правильный процесс («противоположный» в сетке), транспонировав как часть коммуникации.

Я использую MPI_Type_create_subarray , у которого есть аргумент для порядка: MPI_ORDER_C или MPI_ORDER_FORTRAN для основных строк и столбцов соответственно. Я думал, что если я отправлю как один из них, а получу как другой, то моя матрица будет транспонирована как часть сообщения. Однако, похоже, этого не происходит - он просто остается не транспонированным.

Важная часть кода приведена ниже, и весь файл кода доступен по адресу this gist . Есть ли у кого-нибудь идеи, почему это не работает? Должен ли этот подход к выполнению работы транспонирования? Я бы подумал, что это так, прочитав описания MPI_ORDER_C и MPI_ORDER_FORTRAN , но, может быть, и нет.

/* ----------- DO TRANSPOSE ----------- */
/* Find the opposite co-ordinates (as we know it's a square) */
coords2[0] = coords[1];
coords2[1] = coords[0];

/* Get the rank for this process */
MPI_Cart_rank(cart_comm, coords2, &rank2);

/* Send to these new coordinates */

tag = (coords[0] + 1) * (coords[1] + 1);

/* Create new derived type to receive as */
/* MPI_Type_vector(rows_in_core, cols_in_core, cols_in_core, MPI_DOUBLE, &vector_type); */
sizes[0] = rows_in_core;
sizes[1] = cols_in_core;

subsizes[0] = rows_in_core;
subsizes[1] = cols_in_core;

starts[0] = 0;
starts[1] = 0;

MPI_Type_create_subarray(2, sizes, subsizes, starts, MPI_ORDER_FORTRAN, MPI_DOUBLE, &send_type);
MPI_Type_commit(&send_type);

MPI_Type_create_subarray(2, sizes, subsizes, starts, MPI_ORDER_C, MPI_DOUBLE, &recv_type);
MPI_Type_commit(&recv_type);


/* We're sending in row-major form, so it's just rows_in_core * cols_in_core lots of MPI_DOUBLE */
MPI_Send(&array[0][0], 1, send_type, rank2, tag ,cart_comm);

/* Receive from these new coordinates */
MPI_Recv(&new_array[0][0], 1, recv_type, rank2, tag, cart_comm, &status);

6
задан robintw 11 April 2011 в 20:43
поделиться