c - MPI 上的并行数组加法

Question

MPI_Init(&argc, &argv);


    int size,rank,i,*a,*b,*c,N;
    double t1,t2;
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    srand(time(NULL)); 



    if(rank==0)
    {
        printf("Input an integer for arrays size\n");
        scanf("%d",&N);
        a = (int*) malloc((N)* sizeof(int));
        b = (int*) malloc((N) * sizeof(int));
        c = (int*) malloc((N)* sizeof(int));
        for(i=0;i<N;i++)
        {
            a[i]=rand()%100+1;
            b[i]=rand()%100+1;
            c[i]=0;


            printf("c[%d]= %d , a[i]= %d ,  b[i] = %d  \n",i, c[i],a[i],b[i]);

        }
    }
    t1=MPI_Wtime();


    MPI_Bcast(&N,1,MPI_INT,0,MPI_COMM_WORLD);

    MPI_Bcast(&a,N,MPI_INT,0,MPI_COMM_WORLD);
    MPI_Bcast(&b,N,MPI_INT,0,MPI_COMM_WORLD);
    MPI_Bcast(&c,N,MPI_INT,0,MPI_COMM_WORLD);

        printf("\n\n\n\n");


    for(i=rank;i<N;i+=size)
    {   
        printf("entered rank %d \n",rank);
        c[i]=a[i]+b[i];
        printf("c[%d]= %d , a[i]= %d ,  b[i] = %d  \n",i, c[i],a[i],b[i]);
    }


    t2=MPI_Wtime();




    if(rank==0)
        printf("time elapsed %.8f \n" ,t2-t1);



MPI_Finalize();

}

我想用 MPI 库编写一个并行程序。当我只用一个核心运行这段代码时没有问题。但除了核心之外，我还遇到了一个运行时错误，如下所示。等级 0 运行得很好。其他等级不起作用。

Answer 1

您需要对all 上的所有数组a、b 和c 进行malloc() 排名(例如，MPI_Bcast() 不会在非根排名上为您执行此操作)。

从性能和内存占用的角度来看，您更愿意使用 MPI_Scatterv() 数组，以便可以对操作进行矢量化，而无需跨步访问。