cms::cuda::EventCache Class Reference

#include <EventCache.h>

Classes
class	Deleter

Public Types
using	BareEvent = SharedEventPtr::element_type

Public Member Functions
	EventCache ()
SharedEventPtr	get ()

Private Member Functions
void	clear ()
SharedEventPtr	makeOrGet (int dev)

Private Attributes
std::vector< edm::ReusableObjectHolder< BareEvent, Deleter > >	cache_

Friends
class	::CUDAService

Detailed Description

Definition at line 15 of file EventCache.h.

Member Typedef Documentation

BareEvent

using cms::cuda::EventCache::BareEvent = SharedEventPtr::element_type

Definition at line 17 of file EventCache.h.

Constructor & Destructor Documentation

EventCache()

cms::cuda::EventCache::EventCache

(

)

Definition at line 19 of file EventCache.cc.

: cache_(deviceCount()) {}

Member Function Documentation

clear()

void cms::cuda::EventCache::clear ( void  )

private

Definition at line 54 of file EventCache.cc.

                         {
    // Reset the contents of the caches, but leave an
    // edm::ReusableObjectHolder alive for each device. This is needed
    // mostly for the unit tests, where the function-static
    // EventCache lives through multiple tests (and go through
    // multiple shutdowns of the framework).
    cache_.clear();
    cache_.resize(deviceCount());
  }

References cache_, and cms::cuda::deviceCount().

Referenced by CUDAService::CUDAService(), and CUDAService::~CUDAService().

get()

SharedEventPtr cms::cuda::EventCache::get

(

)

Definition at line 21 of file EventCache.cc.

                                 {
    const auto dev = currentDevice();
    auto event = makeOrGet(dev);
    // captured work has completed, or a just-created event
    if (eventWorkHasCompleted(event.get())) {
      return event;
    }
 
    // Got an event with incomplete captured work. Try again until we
    // get a completed (or a just-created) event. Need to keep all
    // incomplete events until a completed event is found in order to
    // avoid ping-pong with an incomplete event.
    std::vector<SharedEventPtr> ptrs{std::move(event)};
    bool completed;
    do {
      event = makeOrGet(dev);
      completed = eventWorkHasCompleted(event.get());
      if (not completed) {
        ptrs.emplace_back(std::move(event));
      }
    } while (not completed);
    return event;
  }

References cms::cuda::currentDevice(), event(), cms::cuda::eventWorkHasCompleted(), makeOrGet(), and eostools::move().

Referenced by Options.Options::__getitem__(), betterConfigParser.BetterConfigParser::__updateDict(), rrapi.RRApi::columns(), rrapi.RRApi::count(), rrapi.RRApi::data(), cms::cuda::ESProduct< T >::ESProduct(), betterConfigParser.BetterConfigParser::getCompares(), betterConfigParser.BetterConfigParser::getGeneral(), betterConfigParser.BetterConfigParser::getResultingSection(), rrapi.RRApi::report(), rrapi.RRApi::reports(), rrapi.RRApi::tables(), rrapi.RRApi::tags(), rrapi.RRApi::templates(), and rrapi.RRApi::workspaces().

makeOrGet()

SharedEventPtr cms::cuda::EventCache::makeOrGet ( int  dev )

private

Definition at line 45 of file EventCache.cc.

                                              {
    return cache_[dev].makeOrGet([dev]() {
      cudaEvent_t event;
      // it should be a bit faster to ignore timings
      cudaCheck(cudaEventCreateWithFlags(&event, cudaEventDisableTiming));
      return std::unique_ptr<BareEvent, Deleter>(event, Deleter{dev});
    });
  }

References cache_, cudaCheck, and event().

Referenced by get().

Friends And Related Function Documentation

::CUDAService

friend class ::CUDAService

friend

Definition at line 30 of file EventCache.h.

Member Data Documentation

cache_

std::vector<edm::ReusableObjectHolder<BareEvent, Deleter> > cms::cuda::EventCache::cache_

private

Definition at line 48 of file EventCache.h.

Referenced by clear(), and makeOrGet().