执行策略

C++17标准中定义了三种执行策略:

  • std::execution::sequenced_policy
  • std::execution::parallel_policy
  • std::execution::parallel_unsequenced_policy

(译者注:C++20中添加了unsequenced_policy策略)

相应的策略标定了程序应该串行、并行,还是与向量化并行。

  • std::execution::seq: 串行执行
  • std::execution::par: 多线程并行执行
  • std::execution::par_unseq: 多个线程上并行,可以循环交叉,也能使用SIMD(单指令多数据)

std::execution::parstd::execution::par_unseq允许算法并行或向量化并行。

下面的代码片段展示了所有执行策略的使用方式。

  1. #include <execution>
  2. #include <vector>
  3. #include <algorithm>
  4. int main() {
  5. std::vector<int> v = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
  6. // standard sequential sort
  7. std::sort(v.begin(), v.end());
  8. // sequential execution
  9. std::sort(std::execution::seq, v.begin(), v.end());
  10. // permitting parallel execution
  11. std::sort(std::execution::par, v.begin(), v.end());
  12. //permitting parallel and vectorized execution
  13. std::sort(std::execution::par_unseq, v.begin(), v.end());
  14. }

示例中,可以使用经典的std::sort(第11行)。C++17中,可以明确指定使用方式:串行(第14行)、并行(第17行),还是向量化并行(第20行)。

std::is_execution_policy可以检查模板参数T是标准数据类型,还是执行策略类型:std::is_execution_policy<T>::value。如果Tstd::execution::sequenced_policy, std::execution::parallel_policy, std::execution::parallel_unsequenced_policy,或已定义的执行策略类型,则表达式结果为true;否则,为false。

并行和向量化执行

算法是否以并行和向量化的方式运行,取决于许多因素。例如:CPU和编译器是否支持SIMD指令,还取决于编译器实现和代码的优化级别。

下面的示例使用循环填充数组。

  1. #include <iostream>
  2. const int SIZE = 8;
  3. int vec[] = { 1, 2, 3, 4, 5, 6, 7, 8 };
  4. int res[] = { 0, 0, 0, 0, 0, 0, 0, 0 };
  5. int main() {
  6. for (int i = 0; i < SIZE; ++i) {
  7. res[i] = vec[i] + 5;
  8. }
  9. for (int i = 0; i < SIZE; ++i) {
  10. std::cout << res[i] << " ";
  11. }
  12. std::cout << std::endl;
  13. }

第12行是这个示例中的关键。我们可以在compiler explorer看一下clang 3.6生成的相应汇编指令。

无优化

汇编指令中,每个加法都是串行进行的。

执行策略 - 图1

使用最高优化级别

通过使用最高的优化级别-O3,寄存器(如:xmm0)可以容纳128位,或者说是4个整型数字。这样,加法就可以同时在四个元素进行了。

执行策略 - 图2

无执行策略算法的重载,与具有串行执行策略std::execution::seq算法的重载在异常处理方面有所不同。

异常

如果执行策略的算法发生异常,将调用std::terminatestd::terminate调用std::terminate_handler,之后使用std::abort,让异常程序终止。执行策略的算法与调用std::execution::seq执行策略的算法之间没有区别。无执行策略的算法会传播异常,因此可以对异常进行处理。exceptionExecutionPolicy.cpp可以佐证我的观点。

  1. // exceptionExecutionPolicy.cpp
  2. #include <algorithm>
  3. #include <execution>
  4. #include <iostream>
  5. #include <stdexcept>
  6. #include <string>
  7. #include <vector>
  8. int main() {
  9. std::cout << std::endl;
  10. std::vector<int> myVec{ 1,2,3,4,5 };
  11. try {
  12. std::for_each(myVec.begin(), myVec.end(),
  13. [](int) {throw std::runtime_error("Without execution policy"); }
  14. );
  15. }
  16. catch (const std::runtime_error & e) {
  17. std::cout << e.what() << std::endl;
  18. }
  19. try {
  20. std::for_each(std::execution::seq, myVec.begin(), myVec.end(),
  21. [](int) {throw std::runtime_error("With execution policy"); }
  22. );
  23. }
  24. catch (const std::runtime_error & e) {
  25. std::cout << e.what() << std::endl;
  26. }
  27. catch (...) {
  28. std::cout << "Catch-all exceptions" << std::endl;
  29. }
  30. }

第21行可以捕获异常std::runtime_error,但不能捕获第30行中的异常,甚至在第33行中的捕获全部异常也无法捕获相应的异常。

使用新的MSVC编译器,并开启std:c++latest选项,程序会给出期望的输出。

执行策略 - 图3

只有第一个异常顺利捕获。

数据竞争和死锁的风险

并行算法无法避免数据竞争和死锁。

下面的并行代码中,就存在数据竞争。

  1. #include <execution>
  2. #include <vector>
  3. int main() {
  4. std::vector<int> v = { 1, 2, 3 };
  5. int sum = 0;
  6. std::for_each(std::execution::par, v.begin(), v.end(), [&sum](int i) {
  7. sum += i + i;
  8. });
  9. }

代码段中,sum有数据竞争。sum上累加了i + i的和,并且是并发修改的,所以必须保护sum

  1. #include <execution>
  2. #include <vector>
  3. #include <mutex>
  4. std::mutex m;
  5. int main() {
  6. std::vector<int> v = { 1, 2, 3 };
  7. int sum = 0;
  8. std::for_each(std::execution::par, v.begin(), v.end(), [&sum](int i) {
  9. std::lock_guard<std::mutex> lock(m);
  10. sum += i + i;
  11. });
  12. }

将执行策略更改为std::execution::par_unseq时,会出现条件竞争,并导致死锁。

  1. #include <execution>
  2. #include <vector>
  3. #include <mutex>
  4. std::mutex m;
  5. int main() {
  6. std::vector<int> v = { 1, 2, 3 };
  7. int sum = 0;
  8. std::for_each(std::execution::par_unseq, v.begin(), v.end(), [&sum](int i) {
  9. std::lock_guard<std::mutex> lock(m);
  10. sum += i + i;
  11. });
  12. }

同一个线程上,Lambda函数可能连续两次调用m.lock,这会产生未定义行为,大多数情况下会导致死锁。这里,可以使用原子来避免死锁。

  1. #include <execution>
  2. #include <vector>
  3. #include <mutex>
  4. #include <atomic>
  5. std::mutex m;
  6. int main() {
  7. std::vector<int> v = { 1, 2, 3 };
  8. std::atomic<int> sum = 0;
  9. std::for_each(std::execution::par_unseq, v.begin(), v.end(), [&sum](int i) {
  10. std::lock_guard<std::mutex> lock(m);
  11. sum += i + i;
  12. });
  13. }

因为sum是一个原子计数器,所以将语义放宽也没关系:sum.fetch_add(i * i, std::memory_order_relaxed) .

执行策略可以作为参数传入69个STL重载算法中,以及C++17添加的8个新算法中。