Question

我正在尝试使用qsort对键值对进行排序。每个proc都读入文件名为proc的文件。 MPI_Gather将所有读取值发送到proc 0，proc 0对键进行排序并将key-val对存储在名为“Output”的文件中。然而，聚集似乎不起作用。任何帮助表示赞赏。谢谢！

我将代码作为

运行

  mpirun -np 3 ./a.out

我的输入文件是：文件“0”：

21 bbbb  
2119 iiii
120 hhhh

档案“1”：

40 dddd 
10 aaaa
100 gggg

档案“2”：

32 cccc
44 eeee
99 ffff

代码是：

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

#define BUFSIZE 3
#define BUFLEN 255

struct keyval{
int key;
char val[BUFLEN];
};

typedef struct keyval keyval_s;

typedef int (*compareptr)(const void*, const void*);

int compare (keyval_s * a, keyval_s * b)
{
  return ( a->key - b->key );
}

int main (int argc, char *argv[])
{    
  int values[BUFSIZE];
  keyval_s kv[BUFSIZE], *recv;
  int n, i=0, temp;
  FILE *in, *out;

  int rank, size;
  MPI_Init(&argc, &argv);
  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  MPI_Comm_size(MPI_COMM_WORLD, &size);
  char filename[20];
  char data[20];

  if(rank ==0){
recv = (keyval_s *) malloc (size*BUFSIZE*sizeof(keyval_s));
   }

  sprintf(filename, "%d", rank);
  in=fopen(filename,"r");
  while(fscanf(in,"%d %s",&kv[i].key, kv[i].val) != EOF){
printf("Read key %d, data %s from file\n", kv[i].key, kv[i].val);   
i++;
  }
  MPI_Gather(kv,BUFSIZE,MPI_BYTE,recv,BUFSIZE,MPI_BYTE,0,MPI_COMM_WORLD);
  if(rank==0){
 qsort ((void*)&kv, BUFSIZE, sizeof(keyval_s),(compareptr) compare);
 out=fopen("Output","w");
 for (n=0; n<BUFSIZE*size; n++)
    fprintf (out,"%d %s\n",recv[n].key, recv[n].val);
free(recv);
fclose(out);
   }
fclose(in);
return 0;   
}

Answer 1

MPI_Gather中的数据大小不正确。它应该是

MPI_Gather(kv, 
           sizeof(keyval_s)*BUFSIZE, 
           MPI_BYTE,
           recv, 
           sizeof(keyval_s)*BUFSIZE, 
           MPI_BYTE, 
           0, 
           MPI_COMM_WORLD);

请注意，MPI_Gather中的recvcount参数用于来自单个排名的邮件，而不是收集数据的总大小。

mpi_gather结构

1 个答案: