gpt4 book ai didi

c++ - boost::mpi 1.53 到位 all_reduce

转载 作者:行者123 更新时间:2023-11-30 05:14:33 25 4
gpt4 key购买 nike

如何使用 boost::mpi 1.53 就地全部减少? (这是 CentOS 7 的版本)

boost 1.61 有 boost::mpi::inplace_t ( doc ) 但 boost 1.53 没有 ( doc )。

对于 1.61,我可以使用:

boost::mpi::all_reduce(
comm,
boost::mpi::inplace_t<int*>(ptr_int_array),
n_elements,
op);

最佳答案

如果您不想有一个额外的out_value 字段,您可以通过设置为函数的返回值来覆盖in_value

#include <boost/mpi.hpp>

int main()
{
boost::mpi::environment env;
boost::mpi::communicator comm;

// set in_value to whatever you want.
int in_value = comm.rank()
// overwrite.
in_value = all_reduce(comm, in_value, std::plus<double>());

return 0;
}

关于c++ - boost::mpi 1.53 到位 all_reduce,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43407722/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com