c++ - boost.mpi 中的自定义 reduce 操作

Question

我在使用 boost.mpi 库编译自定义 reduce 操作时遇到了麻烦。我正在尝试对两个 std::vector 进行元素相加。当我用来将两个 vector 相加时，该操作工作正常，但是当我尝试使用 mpi::reduce 时，我收到错误消息“第一次没有从‘int’到‘const std::vector’的已知转换”争论”。

#include <boost/mpi.hpp>

namespace mpi = boost::mpi;

template <typename T>
struct elementwise_add {
    std::vector<T> operator()(const std::vector<T>& a, const std::vector<T>& b) const {
        std::vector<T> result(a.size());
        std::transform(a.begin(), a.end(), b.begin(), result.begin(), std::plus<T>());
        return result;
    }
};

int main() {
    mpi::environment env;
    mpi::communicator world;
    int N = 10;

    std::vector<int> a(N,1), b(N,1);
    auto op = elementwise_add<int>();
    auto c = op(a, b); // works fine

    if (world.rank() == 0) {
        std::vector<int> sum(N,0);
        mpi::reduce(world, a, sum, op, 0); // errors
    } else {
        mpi::reduce(world, a, op, 0);
    }
}

编译:mpic++ -O3 -std=c++11 -I/usr/local/include/-I. -lboost_mpi-mt -lboost_serialization-mt test.cc -o test

Answer 1

您的代码可以在 boost 1.54 上正常编译。但是，在 1.55 版中有两个重载 reduce()介绍:

template<typename T, typename Op> 
void 
reduce(const communicator & comm, std::vector<T> const & in_values, Op op, 
       int root) 
{
  reduce(comm, &in_values.front(), in_values.size(), op, root);
}

template<typename T, typename Op> 
void 
reduce(const communicator & comm, std::vector<T> const & in_values, 
       std::vector<T> & out_values, Op op, int root) 
{
  out_values.resize(in_values.size());
  reduce(comm, &in_values.front(), in_values.size(), &out_values.front(), op, 
         root);
}

这意味着当您调用 reduce()并提供std::vector作为输入(就像您在代码中所做的那样)，指向数据的指针将被传递。请注意，如果 reduce()给定指针，默认情况下它将在每个元素的基础上运行。因此，您不应传递用于组合 vector 的运算符，而应传递用于组合 ints 的运算符。 (例如 std::plus<int>() )。

以下代码应该适合您( boost 版本 >= 1.55):

#include <boost/mpi.hpp>

namespace mpi = boost::mpi;

int main() {
    mpi::environment env;
    mpi::communicator world;
    int N = 10;

    std::vector<int> a(N,1);

    if (world.rank() == 0) {
        std::vector<int> sum(N,0);
        mpi::reduce(world, a, sum, std::plus<int>(), 0);
    } else {
        mpi::reduce(world, a, std::plus<int>(), 0);
    }
}

如果您希望您的代码也能与较早版本的 boost (< 1.55) 一起使用，您必须自己将指针传递给数据。像这样:

    if (world.rank() == 0) {
        std::vector<int> sum(N,0);
        mpi::reduce(world, &a.front(), N, &sum.front(), std::plus<int>(), 0);
    } else {
        mpi::reduce(world, &a.front(), N, std::plus<int>(), 0);
    }