循环通过 C 中的 MPI 环

Question

我无法弄清楚如何在我的 MPI 代码中进行多次迭代。仅通过在不同位置测试多个 for 循环就告诉我我做错了什么。我们的部分指令是在命令行参数中指定的环上多次传递令牌，但我似乎无法通过对迭代进行硬编码来使其正常工作。我将顶部的 for 循环留在了（这不起作用），但注释掉了其他的。

这可能是一个比我预期的更复杂的任务。 任何帮助表示赞赏

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

int main(int argc, char* argv[])
{
    double starttime, endtime;
    MPI_Init(&argc, &argv);
    starttime = MPI_Wtime();
    int world_rank;
    int world_size;

    for (int i = 0; i < 3; i++)

        MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
        MPI_Comm_size(MPI_COMM_WORLD, &world_size);

//pass the token

        int token =1000;

//for (int i = 0; i < 3; i++)
//{
    if (world_rank == 0)
    {
        token = 0;
        MPI_Send(&token, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);

        MPI_Recv(&token, 1, MPI_INT, world_size - 1, 0, MPI_COMM_WORLD, 
        MPI_STATUS_IGNORE);

        printf("FINAL TOKEN %d\n", token);
    }
    else if(world_rank < world_size - 1 )
    {
        MPI_Recv(&token, 1, MPI_INT, world_rank-1, 0, MPI_COMM_WORLD, 
        MPI_STATUS_IGNORE);

        printf("%d has token %d sending to %d\n", world_rank, token, 
        world_rank+1);
        token = token + 2;

        MPI_Send(&token, 1, MPI_INT, world_rank+1, 0, MPI_COMM_WORLD);
        }
    else
        {
        MPI_Recv(&token, 1, MPI_INT, world_rank-1, 0, MPI_COMM_WORLD, 
        MPI_STATUS_IGNORE);
        printf("%d has token %d sending to %d\n", world_rank, token, 
        world_rank+1);
        token = token + 2;
        MPI_Send(&token, 1, MPI_INT, 0, 0, MPI_COMM_WORLD);
        }
    endtime = MPI_Wtime();
    printf("That took %f seconds\n",endtime-starttime);
    }
    MPI_Finalize();

    return 0;
}

这是编译和运行的输出：

$ mpicc -std=c99 test.c
$ mpirun -np 3 ./a.out
1 has token 0 sending to 2
That took 0.000187 seconds
2 has token 2 sending to 3
That took 0.000145 seconds
FINAL TOKEN 4
That took 0.000180 seconds
FINAL TOKEN 4
That took 0.000212 seconds
2 has token 2 sending to 3
That took 0.000202 seconds
2 has token 2 sending to 3
That took 0.000233 seconds
1 has token 0 sending to 2
That took 0.000291 seconds
1 has token 0 sending to 2
That took 0.000322 seconds
FINAL TOKEN 4
That took 0.000244 seconds

Answer 1

如this comment 中所述，for 循环缺少大括号{}，并且每次循环迭代都会重置令牌。

此外，您可以删除其中一个案例，并使用模运算符 % 包含 world_rank==world_size。

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

int main(int argc, char* argv[])
{
    double starttime, endtime;
    MPI_Init(&argc, &argv);
    starttime = MPI_Wtime();
    int world_rank;
    int world_size;

    MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);

    //pass the token

    int token =1000;

    for (int i = 0; i < 3; i++)
    {
        if (world_rank == 0)
        {
            // token = 0; removed
            MPI_Send(&token, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);

            MPI_Recv(&token, 1, MPI_INT, world_size - 1, 0, MPI_COMM_WORLD, 
            MPI_STATUS_IGNORE);

            printf("FINAL TOKEN %d\n", token);
        }
        else
        {
            MPI_Recv(&token, 1, MPI_INT, world_rank-1, 0, MPI_COMM_WORLD, 
            MPI_STATUS_IGNORE);

            printf("%d has token %d sending to %d\n", world_rank, token, 
            (world_rank+1)%world_size); // modified
            token = token + 2;

            MPI_Send(&token, 1, MPI_INT, (world_rank+1)%world_size, 0, MPI_COMM_WORLD); // modified
        }

        endtime = MPI_Wtime();
        printf("That took %f seconds\n",endtime-starttime);
    }
    MPI_Finalize();

    return 0;
}