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

TBBThreadPool()

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

inlineexplicit

Definition at line 29 of file TBBThreadPool.h.

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

Member Function Documentation

Cancel()

void tensorflow::TBBThreadPool::Cancel ( )

inlineoverride

Definition at line 57 of file TBBThreadPool.h.

{}

CurrentThreadId()

int tensorflow::TBBThreadPool::CurrentThreadId ( ) const

inlineoverride

Definition at line 61 of file TBBThreadPool.h.

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

References triggerObjects_cff::id.

GetNumScheduleCalled()

int tensorflow::TBBThreadPool::GetNumScheduleCalled ( )

inline

Definition at line 67 of file TBBThreadPool.h.

{ return numScheduleCalled_; }

References numScheduleCalled_.

instance()

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

inlinestatic

Definition at line 24 of file TBBThreadPool.h.

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

References CMS_THREAD_SAFE, runTheMatrix::nThreads, and dqmiodatasetharvest::pool.

Referenced by tensorflow::run().

NumThreads()

int tensorflow::TBBThreadPool::NumThreads ( ) const

inlineoverride

Definition at line 59 of file TBBThreadPool.h.

{ return nThreads_; }

References nThreads_.

Schedule()

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

inlineoverride

Definition at line 34 of file TBBThreadPool.h.

                                                   {
      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();
    }

References personalPlayback::fn, and numScheduleCalled_.

Referenced by ScheduleWithHint().

ScheduleWithHint()

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

inlineoverride

Definition at line 55 of file TBBThreadPool.h.

{ Schedule(fn); }

References personalPlayback::fn, and Schedule().

Member Data Documentation

nThreads_

const int tensorflow::TBBThreadPool::nThreads_

private

Definition at line 70 of file TBBThreadPool.h.

Referenced by NumThreads().

numScheduleCalled_

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

private

Definition at line 71 of file TBBThreadPool.h.

Referenced by GetNumScheduleCalled(), and Schedule().