c++ - 取消来自辅助线程的请求以强制 MPI_Wait 返回

Question

我正在尝试取消使用 MPI_Irecv 启动的异步操作。在一个线程中，我有一个循环，通过连续调用 MPI_Irecv 和 MPI_Wait 来不断监听请求。要干净地退出循环，我想取消请求，例如从另一个线程调用 MPI_Cancel。

据我了解，MPI_Cancel 标记取消通信，并引用规范:

If a communication is marked for cancellation, then a MPI_WAIT call for that communication is guaranteed to return.

下面的代码表明这并没有像我预期的那样工作，因为 MPI_WAIT 永远不会返回 - 在 Windows 上使用最新的 MS-MPI 和在 Linux 上使用 MPICH2 进行测试。

#include <mpi.h>

#include <iostream>
#include <future>


using namespace std::literals::chrono_literals;

void async_cancel(MPI_Request *request)
{
    std::this_thread::sleep_for(1s);

    std::cout << "Before MPI_Cancel" << std::endl;

    int res = MPI_Cancel(request);
    if (res != MPI_SUCCESS)
        std::cerr << "MPI_Cancel failed" << std::endl;

    std::cout << "After MPI_Cancel" << std::endl;
}


int main(int argc, char* argv[])
{
    int provided;
    MPI_Init_thread(&argc, &argv, MPI_THREAD_MULTIPLE, &provided);

    if (provided != MPI_THREAD_MULTIPLE)
        std::cout << "MPI_Init_thread could not provide MPI_THREAD_MULTIPLE" << std::endl;

    int rank, numprocs;
    MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);


    MPI_Request request;
    MPI_Status status;

    int buffer;

    if (rank == 0)
    {
        MPI_Irecv(&buffer, 1, MPI_INT, 1, 123, MPI_COMM_WORLD, &request);

        auto res = std::async(std::launch::async, &async_cancel, &request);

        std::cout << "Before MPI_Wait" << std::endl;

        MPI_Wait(&request, &status);

        std::cout << "After MPI_Wait " << std::endl;
    }
    else
        std::this_thread::sleep_for(2s);

    MPI_Finalize();

    return 0;
}

这是一个实现问题吗？有没有更好的方法来强制 MPI_Wait 返回？

---

编辑:代码(使用@Zulan 和@amlucas 评论更新)实际上适用于除 MS-MPI 之外的大多数实现。

Answer 1

MPI_Init 为单线程调用初始化 MPI。尝试改用 MPI_Init_thread，它允许您使用线程安全的 MPI 调用。