Question

我正在尝试解决一项家庭作业问题，以调试以下单元测试。

基本上，主进程会生成随机整数，并将其发送给子进程以检查素数，然后将结果传递回主进程，然后算法会结束。

我知道应该将循环替换为集体通信，但这是问题的另一部分。我想了解为什么我这里的这段代码会导致死锁。

通过阅读其他questions，我知道发送/接收的数量应该彼此匹配。但是，我看不出代码中的情况并非如此。

当前行为是找到素数，然后将其发送回主进程，此时程序将挂起-直到使用ctrl-C手动将其取消为止。

我知道这不是解决此问题的惯用方式，但我确实想确切地知道此方法中的错误所在。

谢谢！


TEST_CASE("3a - Finding prime numbers", "[MPI]" )
{
    int rank, size;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    // Random number generation
    std::minstd_rand generator;
    unsigned min(2342), max(12342340);
    std::uniform_int_distribution<> distribution(min, max);

    // candidates too big, one of the size values is the master node
    std::vector<unsigned> candidates(size - 1);

    // Main loop continues until a prime is found
    int found_prime(0);
    while (found_prime == 0) {


        if (rank == 0) {

            // Create some candidate numbers for each process to test
            std::generate(candidates.begin(), candidates.end(),
                          [&]() { return distribution(generator); });

            // Send one to each worker
            for (int worker(1); worker < size; ++worker) {
                int rc = MPI_Ssend(&candidates[worker - 1], 1, MPI_UNSIGNED,
                                   worker, 0, MPI_COMM_WORLD);
                REQUIRE(rc == MPI_SUCCESS);
            }

            // Receive whether it was prime
            for (int worker(1); worker < size; ++worker) {
                unsigned result;
                int rc = MPI_Recv(&result, 1, MPI_UNSIGNED, worker, 0,
                                  MPI_COMM_WORLD, MPI_STATUS_IGNORE);
                REQUIRE(rc == MPI_SUCCESS);
                if (result == 1) {
                    found_prime = candidates[worker - 1];
                    std::cout << "Worker " << worker << " found prime "
                              << found_prime << std::endl;
                }
            }
        } else {
            // Receive the candidate to check
            unsigned candidate;
            int rc = MPI_Recv(&candidate, 1, MPI_UNSIGNED, 0, 0, MPI_COMM_WORLD,
                              MPI_STATUS_IGNORE);
            REQUIRE(rc == MPI_SUCCESS);
            // Do the check
            unsigned is_prime = mp::IsPrime(candidate) ? 1 : 0;
            // Return the result
            rc = MPI_Ssend(&is_prime, 1, MPI_UNSIGNED, 0, 0, MPI_COMM_WORLD);
            REQUIRE(rc == MPI_SUCCESS);
        }
    }
    std::cout << "Finished" << rank << std::endl;
}

Answer 1

我对MPI一无所知，但是在您的代码中，如果rank != 0，则永远不会退出while循环，因为从未在found_prime分支中设置else（并且{{1 }}也从未更改过。）

编辑：

就像@DanielLangr所说的那样，奴隶将需要一种方法来发现没有更多的工作来退出（循环）。

MPI中使用点到点从主机发送到子进程的点对点通信引起的死锁

1 个答案: