我正在尝试取消使用 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 之外的大多数实现。
最佳答案
MPI_Init
为单线程调用初始化 MPI。
尝试改用 MPI_Init_thread
,它允许您使用线程安全的 MPI 调用。
关于c++ - 取消来自辅助线程的请求以强制 MPI_Wait 返回,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56966716/