运行时的不同执行策略

时间:2018-10-24 17:45:53

标签: c++ parallel-processing

在C ++ 17中,algorithm头中的许多函数现在可以采用执行策略。例如,我可以定义和调用这样的函数:

template <class ExecutionPolicy>
void f1(const std::vector<std::string>& vec, const std::string& elem, ExecutionPolicy&& policy) {
    const auto it = std::find(
        std::forward<ExecutionPolicy>(policy),
        vec.cbegin(), vec.cend(), elem
    );
}

std::vector<std::string> vec;
f1(vec, "test", std::execution::seq);

但是,我还没有找到在运行时使用不同策略的好方法。例如,当我想根据某些输入文件使用其他策略时。

我玩弄各种变体,但最终问题始终是std::execution::seqstd::execution::parstd::execution::par_unseq的不同类型。

一个可行但麻烦的解决方案如下:

void f2(const std::vector<std::string>& vec, const std::string& elem, const int policy) {
    const auto it = [&]() {
        if (policy == 0) {
            return std::find(
                std::execution::seq,
                vec.cbegin(), vec.cend(), elem
            );
        }
        else if (policy == 1) {
            return std::find(
                std::execution::par,
                vec.cbegin(), vec.cend(), elem
            );
        }
        else{
            return std::find(
                std::execution::par_unseq,
                vec.cbegin(), vec.cend(), elem
            );
        }
    };
}

f2(vec, "test", 0);

我还有其他更优雅的解决方案吗?

编辑:也许我应该更精确些。假设目标是将策略保存在可以具有三个策略之一的变量中。该变量应该是函数的参数。

2 个答案:

答案 0 :(得分:2)

这里的标准方法是将类型的选择与该类型的 use 分开:后者采用功能模板的形式,该模板被多次实例化。以前的非模板函数(或具有较少模板参数的函数模板)。

为避免在这两层之间复制常规参数,请使用通用lambda作为模板。为避免重复选择逻辑,请制作一个函数模板,以适当的策略调用任何lambda:

enum Policy {seq,par,par_unseq};

template<class F>
auto maybe_parallel(F f,Policy p) {
  switch(p) {
  case seq: return f(std::execution::seq);
  case par: return f(std::execution::par);
  default: return f(std::execution::par_unseq);
  }
}

auto f2(const std::vector<std::string>& vec,
        const std::string& elem,Policy p) {
  return maybe_parallel
    ([&](auto &pol) {return std::find(pol,vec.begin(),vec.end(),elem);},p);
}

答案 1 :(得分:1)

另一种方法是为此使用 std::variantstd::visit

using parallel_policy = std::variant
                        <
                         std::execution::sequenced_policy,
                         std::execution::parallel_policy,
                         std::execution::parallel_unsequenced_policy,
                         std::execution::unsequenced_policy
                        >;

void f2(const std::vector<std::string>& vec,
        const std::string&              elem,
        const parallel_policy           policy
       )
{
    std::visit
    (
        [&](auto policy_real)
        {
            f1(vec, elem, policy_real);
        },
        policy
    );
}