MPI_File_read_at逐行

时间:2016-03-19 02:52:35

标签: c mpi mpi-io

我是MPI的新学习者。我想使用MPI_File_read_at()逐行读取txt文件中的数据。每行的长度是不同的,所以当我读一行(设置一个缓冲区的长度)时,有时它也会读取下一行,将下一行的某些部分发送到缓冲区,这确实会导致问题......所以,我想知道有什么方法可以使用MPI_File_read_at()逐行读取数据吗?在见面时让它停止" \ n"在每一行的结尾?或者您是否有更好的建议使用除MPI_File_read_at()以外的MPI函数逐行读取数据?

我想我的问题是如何使用MPI_File_read_at()来执行与fgets相同的操作

/*the traditional way to read a file line by line:*/
for(i=0;i<nlens;i++)
{
    fgets(line, 20, fp);
    sscanf(line,"%d%d",&e1,&e2);
}


/*the way I am doing this by MPI*/

int offset = 0;
int count = 15;
char line[15];
for(i=0;i<nlens;i++)
{
    offset=i*count;
    MPI_File_read_at(fp, offset, line, count, MPI_CHAR, &status);
    printf("line%d:/n%s/n",i,line);
}

/*So, if I have a file looks like:*/
0        2
0        44353
3        423
4        012312
5        2212
5        476

/*the output of mpi version  will be:*/
line0: 
0        2
0
line1:
      44353
3
line2:
      423
4
line3:
    012312
5
line4:
      2212
5
line5:
     476
2
5

void read_data(char address_data[], int num_edges, int link[][100])
{
    int i,e1,e2;
    FILE *fp;
    char line[30];

    int totalTaskNum, rankID;
    MPI_Comm_rank(MPI_COMM_WORLD, &rankID);
    MPI_Comm_size(MPI_COMM_WORLD, &totalTaskNum);
    MPI_Status status;
    if(rankID == 0)  /*to avoid different processors access to the file simultaneously*/
    {
        fp = fopen(address_data, "r");
        for(i=0;i<num_edges;i++)
        {
            fgets(line, 20, fp);
            sscanf(line,"%d%d",&e1,&e2);
            link[e1][e2]++;
        }
        fclose(fp);
    }
}

1 个答案:

答案 0 :(得分:0)

MPI I / O用于读取二进制文件,其中每个记录由结构体表示,因此这样您可以循环记录,或在给定位置查找记录。

但是对于文本文件,您无法提前知道每行结束/开始的时间。

如果你想读取一个文本文件,你需要将文件的数据传递给缓冲区,然后通过char读取数据char,每当c == '\n'时,你现在就知道当前的行结束。

如果你只需要查找一次这样的行就可以了,那么你就像我说的那样读取文件,每当一行开始时递增一个计数器,这样你就知道哪一行是你想要的。

但是如果你的程序需要多次查看(更有可能),一个想法是,你读取文件一次,建立行的位置索引,然后你就可以了知道每条线的位置。 当然,如果您之后更新文件,则需要相应地更新索引。

修改:您还可以查看此answer。它在过去帮助了我。

编辑2:你问的是什么。 MASTER进程将文件读入2D int数组。

我创建了一个文件matrix.txt,我放置了这个space separated值:

0 2
0 44353
3 423
4 012312
5 2212
5 476

以下是该计划:

#include <mpi.h>

#define FILENAME    "matrix.txt"
#define WIDTH       2
#define HEIGHT      6


int main() {

    int rank, world_size;

    MPI_Init(NULL, NULL);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);


    int matrix[HEIGHT][WIDTH];

    /**
     * The MASTER process reads the file
     */
    if (rank == 0) { 
        FILE *fp = fopen(FILENAME, "r");

        char line[100];
        int i = 0;
        while (fgets(line, 100, fp)) {
            sscanf(line, "%d %d", &matrix[i][0], &matrix[i][1]);
            i++;
        }
    }

    //Broadcast it to all processes
    MPI_Bcast(matrix, HEIGHT * WIDTH, MPI_INT, 0, MPI_COMM_WORLD);

    //just for demo purposes each process prints its array orderly
    int p = 0, i = 0, j = 0;
    for (p = 0; p < world_size; p++) {
        MPI_Barrier(MPI_COMM_WORLD);
        if (p == rank) {
            printf("----------\n proc:%d received:\n", rank);
            for (i = 0; i < HEIGHT; i++) {
                for (j = 0; j < WIDTH; j++) {
                    printf("%d\t", matrix[i][j]);
                }
                printf("\n");
            }
        }
    }


    MPI_Finalize();
    return 0;
}

我做了一些假设,比如你事先知道文件中矩阵的大小。如果你想让你的程序更具活力,我可以把它留给你。 在阅读文件后,我将矩阵广播到所有过程,以便您可以测试它是否有效。因为你说你想要进行矩阵乘法,你需要使用MPI_Scatter或其他东西来遵循分解策略,这样每个进程都会收到一个不同的块。

这是输出:

$ mpirun -np 4 ./program 
----------
 proc:0 received:
0       2       
0       44353   
3       423     
4       12312   
5       2212    
5       476     
----------
 proc:1 received:
0       2       
0       44353   
3       423     
4       12312   
5       2212    
5       476     
----------
 proc:2 received:
0       2       
0       44353   
3       423     
4       12312   
5       2212    
5       476     
----------
 proc:3 received:
0       2       
0       44353   
3       423     
4       12312   
5       2212    
5       476