运行时执行策略不同

Different execution policies at runtime

本文关键字:策略 执行 运行时      更新时间:2023-10-16

在 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);

还有我忽略的更优雅的解决方案吗?

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

这里的标准方法是将类型的选择与类型的使用分开:后者采用函数模板的形式,由前一个非模板函数(或具有较少模板参数的函数模板(多次实例化。

为避免在这两个层之间重复正常参数,请使用通用 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);
}

另一种方法是为此目的使用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
);
}