Проблемы с реализацией MPI_Scatter с помощью MPI_Send и MPI_Recv - PullRequest
Новая
       7

Проблемы с реализацией MPI_Scatter с помощью MPI_Send и MPI_Recv

0 голосов
/ 13 октября 2018

Я пытаюсь реализовать функцию MPI MPI_Scatter, используя MPI_Send и MPI_Recv.

Я хотел бы использовать официальное объявление функции, которая использует указатели для векторов / массивов. 

MPI_Scatter(
    void* send_data,
    int send_count,
    MPI_Datatype send_datatype,
    void* recv_data,
    int recv_count,
    MPI_Datatype recv_datatype,
    int root,
    MPI_Comm communicator)

Я создал пример, который прекрасно работает с MPI MPI_Scatter.Он показывает правильный результат.

Мне нужно функционировать, чтобы реализовать этот материал, один с указателями, а другой со статическими массивами целых чисел.Второй работает нормально, но первый просто показывает первые три элемента созданного массива.Я думаю, что это проблема, связанная с выделенной памятью матрицы, но я не вижу способа исправить это.

ЗДЕСЬ КОД (MMPI_Scatter выдает ошибку): 

#include <stdio.h>
#include <mpi.h>
#include <stdlib.h>
#include <math.h>
#include <unistd.h>


#define ROOT 0
#define N 3

int main(int argc, char **argv) {

    // for storing this process' rank, and the number of processes
    int rank, np;  
    int *matrix;

    //MPI_Scatter
    int send_count, recv_count;
    int *recv_data;

    MPI_Status status, info;

    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD,&np);
    MPI_Comm_rank(MPI_COMM_WORLD,&rank);


    if (rank == ROOT) {
        matrix = createMatrix(np, np);
        printArray(matrix, np * np);

    }

    send_count = np;
    recv_count = np;
    recv_data = malloc(recv_count * sizeof(int));


    //The original function provided by MPI works great!!
    MPI_Scatter(matrix, send_count, MPI_INT, recv_data, recv_count, MPI_INT, ROOT, MPI_COMM_WORLD);

    //This function just show the first three elements of the matrix
    //MMPI_Scatter(matrix, send_count, MPI_INT, recv_data, recv_count, MPI_INT, ROOT, MPI_COMM_WORLD);

    //This function works great, but it not use the official declaration of the MPI_Scatter
    //MMPI_Scatter2(matrix, send_count, MPI_INT, recv_data, recv_count, MPI_INT, ROOT, MPI_COMM_WORLD);
    printArray(recv_data , recv_count);


    MPI_Finalize();
    return 0;
}


//http://mpitutorial.com/tutorials/mpi-scatter-gather-and-allgather/
void MMPI_Scatter(void* send_data, int send_count, MPI_Datatype send_datatype, 
                  void* recv_data, int recv_count, MPI_Datatype recv_datatype, 
                  int root, MPI_Comm communicator) {

    int np, rank;
    int i;
    MPI_Status status;
    MPI_Comm_size(communicator, &np);
    MPI_Comm_rank(communicator, &rank);

    printArray(send_data, np * np);

    if (rank == ROOT) {
        for (i = 0; i < np; i++) {
            MPI_Send(send_data + (i * send_count), send_count, send_datatype, i, 0, communicator);
        }
    }
    MPI_Recv(recv_data, recv_count, recv_datatype, root, 0, communicator, &status);
    printArray(send_data, np * np);

}

//Works great, but without pointer
void MMPI_Scatter2(int send_data[], int send_count, MPI_Datatype send_datatype, 
                   int recv_data[], int recv_count, MPI_Datatype recv_datatype, 
                   int root, MPI_Comm communicator) {

    int np, rank;
    int i;
    MPI_Status status;
    MPI_Comm_size(communicator, &np);
    MPI_Comm_rank(communicator, &rank);



    if (rank == ROOT) {
        for (i = 0; i < np; i++) {
            MPI_Send(send_data + (i * send_count), send_count, send_datatype, i, 0, communicator);
        }
    }
    MPI_Recv(recv_data, recv_count, recv_datatype, root, 0, communicator, &status); 
    printArray(recv_data, np);
}


int *createMatrix(int nRows, int nCols) {

  int *matrix;

  int h, i, j;

  if ((matrix = malloc(nRows * nCols * sizeof(int))) == NULL) {
    printf("Malloc error:");
    exit(1);
  }

  //Test values 
  for (h = 0; h < nRows * nCols; h++) {
    matrix[h] = h + 1;
  }

  return matrix;
}

ОБНОВЛЕНИЕ 1:

Я думаю, что это связано с информацией по этой ссылке: https://www.mpi -forum.org / docs / mpi-1.1 / mpi-11-html / node71.html # Node71

Есть строка с: 

MPI_Send(sendbuf + i*sendcount*extent(sendtype), sendcount, sendtype, i.....)

, но я не знаю, как обрабатывать extension (sendtype)

ОБНОВЛЕНИЕ 2:

Теперь это работает, но на данный момент, потому что я сам знаю тип данных 

void MMPI_Scatter(void* send_data, int send_count, MPI_Datatype send_datatype, 
                  void* recv_data, int recv_count, MPI_Datatype recv_datatype, 
                  int root, MPI_Comm communicator) {

    int np, rank;
    int i;
    int size;

    MPI_Datatype type;
    type = MPI_INT;
    MPI_Type_size(type, &size);

    MPI_Status status;
    MPI_Comm_size(communicator, &np);
    MPI_Comm_rank(communicator, &rank);



    if (rank == ROOT) {
        for (i = 0; i < np; i++) {
            MPI_Send(send_data + ((i * send_count) * size), send_count, send_datatype, i, 0, communicator);
        }
    }

    MPI_Recv(recv_data, recv_count, recv_datatype, root, 0, communicator, &status);

}

ОБНОВЛЕНИЕ 3 (решено): 

void MMPI_Scatter(void* send_data, int send_count, MPI_Datatype send_datatype, 
                  void* recv_data, int recv_count, MPI_Datatype recv_datatype, 
                  int root, MPI_Comm communicator) {

    int np, rank;
    int i;
    int size;

    MPI_Datatype type;
    type = send_datatype;
    MPI_Type_size(type, &size);

    MPI_Status status;
    MPI_Comm_size(communicator, &np);
    MPI_Comm_rank(communicator, &rank);



    if (rank == ROOT) {
        for (i = 0; i < np; i++) {
            MPI_Send(send_data + ((i * send_count) * size), send_count, send_datatype, i, 0, communicator);

        }
    }
    MPI_Recv(recv_data, recv_count, recv_datatype, root, 0, communicator, &status);

}

ОБНОВЛЕНИЕ 4

Эта функция работает нормально, поскольку используется ROOT, но после вызова из коллектива ROOT должен быть заменен на root следующим образом: 

if (rank == root) {
}

1 Ответ

0 голосов
/ 14 октября 2018

Изменить send_data + (i * send_count) на: 

send_data + (i * send_count + MPI_Type_size(send_datatype)
...