cms::cuda::StreamCache Class Reference

#include <StreamCache.h>

Classes
class	Deleter

Public Types
using	BareStream = SharedStreamPtr::element_type

Public Member Functions
SharedStreamPtr	get ()
	StreamCache ()

Private Member Functions
void	clear ()

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

Friends
class	::CUDAService

Detailed Description

Definition at line 15 of file StreamCache.h.

Member Typedef Documentation

BareStream

using cms::cuda::StreamCache::BareStream = SharedStreamPtr::element_type

Definition at line 17 of file StreamCache.h.

Constructor & Destructor Documentation

StreamCache()

cms::cuda::StreamCache::StreamCache

(

)

Definition at line 18 of file StreamCache.cc.

: cache_(deviceCount()) {}

Member Function Documentation

clear()

void cms::cuda::StreamCache::clear ( void  )

private

Definition at line 29 of file StreamCache.cc.

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

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

                          {
    // 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
    // StreamCache lives through multiple tests (and go through
    // multiple shutdowns of the framework).
    cache_.clear();
    cache_.resize(deviceCount());
  }

get()

SharedStreamPtr cms::cuda::StreamCache::get

(

)

Definition at line 20 of file StreamCache.cc.

References cache_, cudaCheck, cms::cuda::currentDevice(), and cms::cuda::stream.

Referenced by Options.Options::__getitem__(), betterConfigParser.BetterConfigParser::__updateDict(), submitPVValidationJobs.BetterConfigParser::__updateDict(), rrapi.RRApi::columns(), rrapi.RRApi::count(), rrapi.RRApi::data(), TestCUDAAnalyzerGPU::endJob(), betterConfigParser.BetterConfigParser::getCompares(), betterConfigParser.BetterConfigParser::getGeneral(), betterConfigParser.BetterConfigParser::getResultingSection(), submitPVValidationJobs.BetterConfigParser::getResultingSection(), rrapi.RRApi::report(), rrapi.RRApi::reports(), cms::cuda::impl::ScopedContextBase::ScopedContextBase(), rrapi.RRApi::tables(), rrapi.RRApi::tags(), rrapi.RRApi::templates(), TestCUDAAnalyzerGPU::TestCUDAAnalyzerGPU(), and rrapi.RRApi::workspaces().

                                   {
    const auto dev = currentDevice();
    return cache_[dev].makeOrGet([dev]() {
      cudaStream_t stream;
      cudaCheck(cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking));
      return std::unique_ptr<BareStream, Deleter>(stream, Deleter{dev});
    });
  }

Friends And Related Function Documentation

::CUDAService

friend class ::CUDAService

friend

Definition at line 27 of file StreamCache.h.

Member Data Documentation

cache_

std::vector<edm::ReusableObjectHolder<BareStream, Deleter> > cms::cuda::StreamCache::cache_

private

Definition at line 41 of file StreamCache.h.

Referenced by clear(), and get().