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

时间:2015-03-04 02:48:51

标签: c++ c++11 boost mpi

我在使用boost.mpi库进行自定义reduce操作编译时遇到了麻烦。我正在尝试按元素添加两个std::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

1 个答案:

答案 0 :(得分:2)

使用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>())。

以下代码适合您(升级版&gt; = 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(&lt; 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);
    }