#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::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

◆ Cancel()

void tensorflow::TBBThreadPool::Cancel ( )

inlineoverride

Definition at line 59 of file TBBThreadPool.h.

59 {}

◆ CurrentThreadId()

int tensorflow::TBBThreadPool::CurrentThreadId ( ) const

inlineoverride

Definition at line 63 of file TBBThreadPool.h.

References triggerObjects_cff::id.

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

◆ GetNumScheduleCalled()

int tensorflow::TBBThreadPool::GetNumScheduleCalled ( )

inline

Definition at line 69 of file TBBThreadPool.h.

References numScheduleCalled_.

69 { return numScheduleCalled_; }

tensorflow::TBBThreadPool::numScheduleCalled_

std::atomic< int > numScheduleCalled_

Definition: TBBThreadPool.h:73

◆ instance()

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;
     }

◆ NumThreads()

int tensorflow::TBBThreadPool::NumThreads ( ) const

inlineoverride

Definition at line 61 of file TBBThreadPool.h.

References nThreads_.

61 { return nThreads_; }

tensorflow::TBBThreadPool::nThreads_

const int nThreads_

Definition: TBBThreadPool.h:72

◆ Schedule()

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