tensorflow::TBBThreadPool Class Reference

#include <TBBThreadPool.h>

Inheritance diagram for tensorflow::TBBThreadPool:

Public Member Functions
void	Cancel () override
int	CurrentThreadId () const override
int	GetNumScheduleCalled ()
int	NumThreads () const override
void	Schedule (std::function< void()> fn) override
void	ScheduleWithHint (std::function< void()> fn, int start, int end) override
	TBBThreadPool (int nThreads=-1)

Static Public Member Functions
static TBBThreadPool &	instance (int nThreads=-1)

Private Attributes
const int	nThreads_
std::atomic< int >	numScheduleCalled_

Detailed Description

Definition at line 22 of file TBBThreadPool.h.

Constructor & Destructor Documentation

tensorflow::TBBThreadPool::TBBThreadPool ( int  nThreads = -1 )

inlineexplicit

Definition at line 29 of file TBBThreadPool.h.

        : nThreads_(nThreads > 0 ? nThreads
                                 : tbb::global_control::active_value(tbb::global_control::max_allowed_parallelism)),
          numScheduleCalled_(0) {
      // when nThreads is zero or smaller, use the default value determined by tbb
    }

Member Function Documentation

void tensorflow::TBBThreadPool::Cancel ( )

inlineoverride

Definition at line 59 of file TBBThreadPool.h.

{}

int tensorflow::TBBThreadPool::CurrentThreadId ( ) const

inlineoverride

Definition at line 63 of file TBBThreadPool.h.

References gpuClustering::id.

                                         {
      static std::atomic<int> idCounter{0};
      thread_local const int id = idCounter++;
      return id;
    }

int tensorflow::TBBThreadPool::GetNumScheduleCalled ( )

inline

Definition at line 69 of file TBBThreadPool.h.

References numScheduleCalled_.

{ return numScheduleCalled_; }

static TBBThreadPool& tensorflow::TBBThreadPool::instance ( int  nThreads = -1 )

inlinestatic

Definition at line 24 of file TBBThreadPool.h.

References CMS_THREAD_SAFE, and submitPVResolutionJobs::pool.

Referenced by tensorflow::run().

                                                      {
      CMS_THREAD_SAFE static TBBThreadPool pool(nThreads);
      return pool;
    }

int tensorflow::TBBThreadPool::NumThreads ( ) const

inlineoverride

Definition at line 61 of file TBBThreadPool.h.

References nThreads_.

{ return nThreads_; }

void tensorflow::TBBThreadPool::Schedule ( std::function< void()>  fn )

inlineoverride

Definition at line 36 of file TBBThreadPool.h.

References personalPlayback::fn, and numScheduleCalled_.

Referenced by ScheduleWithHint().

                                                   {
      numScheduleCalled_ += 1;

      // use a task arena to avoid having unrelated tasks start
      // running on this thread, which could potentially start deadlocks
      tbb::task_arena taskArena;
      tbb::task_group taskGroup;

      // we are required to always call wait before destructor
      auto doneWithTaskGroup = [&taskArena, &taskGroup](void*) {
        taskArena.execute([&taskGroup]() { taskGroup.wait(); });
      };
      std::unique_ptr<tbb::task_group, decltype(doneWithTaskGroup)> taskGuard(&taskGroup, doneWithTaskGroup);

      // schedule the task
      taskArena.execute([&taskGroup, &fn] { taskGroup.run(fn); });

      // reset the task guard which will call wait
      taskGuard.reset();
    }

void tensorflow::TBBThreadPool::ScheduleWithHint ( std::function< void()>  fn, int  start, int  end  )

inlineoverride

Definition at line 57 of file TBBThreadPool.h.

References personalPlayback::fn, and Schedule().

{ Schedule(fn); }

Member Data Documentation

const int tensorflow::TBBThreadPool::nThreads_

private

Definition at line 72 of file TBBThreadPool.h.

Referenced by NumThreads().

std::atomic<int> tensorflow::TBBThreadPool::numScheduleCalled_

private

Definition at line 73 of file TBBThreadPool.h.

Referenced by GetNumScheduleCalled(), and Schedule().