使用 TBB parallel_for_each() 并减少到原子计数器

Question

这是用于 TBB 代码的原子计数器类。

#include <atomic>
template <typename T = int>
struct AtomicCounter {

private:
std::atomic<T> value;
std::atomic<T> num_inter;

public:
void put_inter(T niter) {
T x = num_inter.fetch_add(niter);
}

T get_inter() { return num_inter.load(); }

void increment() {  ++value; }
void put(T data) {  value.fetch_add(data) ; // ignore the old value returned }
void decrement() {  --value; }
T    get()       { return value.load(); }

};

我正在使用并行 foreach 循环，其中每个线程都有自己的本地计数器值，并使用 AtomicCounter 类的 put 函数更新全局计数器对象。并行 foreach 的函数对象将此全局计数器作为参考。

template<typename counter_t>
struct my_functor {
 public:
  my_functor(){}
  my_functor(counter_t &c):m_c(c){ } 

  template<typename label_t>
  void operator()(label_t label) const {

  // do some work --> local counter

   m_c.put(local_value); // put the local counter value in the global counter 
  } 

  private:
    counter_t& m_c;
    mutable int local_value; //local counter value  

  }

   // for TBB 
 //declare an object of this class as a global counter.

 AtmoicCounter<int> c;
 my_functor<atomicCounter<int>>  func(c) //functor object
 parallel_for_each( mywork.begin(), mywork.end(), func)    //TBB parallel for each

所以基本上，每个线程都会更新全局计数器。我声明 std::atomic 成员的 atomicCounter 类有什么问题吗？我正在使用具有 C++11 功能和英特尔 TBB 4.2 的 GCC 4.7.2

谢谢！

Answer 1

我认为声明中没有任何问题。如果由于某种原因您不确定 GCC atomics，请使用我确定的 TBB atomics。

但我发现问题在于使用 mutable 关键字来访问来自 const 仿函数的字段。 mutable 通常是糟糕设计的标志。在并行编程中，它的使用特别容易出错。如果一个被并行调用的方法标记为 const，这通常意味着它将在仿函数类的同一个实例上被并行调用。因此，您在可变字段上存在数据竞争。

您可以将 local_value 移动到 operator() 的范围内，而不是打破 const 限制，或者尝试线程本地容器 tbb::combinable 或 tbb::enumerable_thread_specific 缓存特定于线程的部分值，或者应用并行缩减算法tbb::parallel_reduce 您可以安全地将部分结果存储在仿函数体中。