boost.mpi 中的自定义减少操作

Question

我在使用 boost.mpi 库编译自定义归约操作时遇到问题。我正在尝试对两个 std::vector 进行元素相加。当我使用将两个向量加在一起时操作正常，但是当我尝试使用 mpi::reduce 时出现错误 "no known conversion from 'int' to 'const std::vector' for 1st argument".

#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() 被赋予指针，它将默认在每个元素的基础上运行。因此，您不应传递用于组合向量的运算符，而应传递用于组合 ints（例如 std::plus<int>()）的运算符。

以下代码应该适合您（提升版本 >= 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);
    }

boost.mpi 中的自定义减少操作

Custom reduce operation in boost.mpi

c++

c++11

boost

mpi