Inheritance diagram for TestCUDAProducerGPUEW:

Public Member Functions
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, edm::WaitingTaskWithArenaHolder waitingTaskHolder) override

void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup) override

	TestCUDAProducerGPUEW (edm::ParameterSet const &iConfig)

	~TestCUDAProducerGPUEW () override=default

Public Member Functions inherited from edm::stream::EDProducer< edm::ExternalWork >
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
cms::cuda::ContextState	ctxState_

cms::cuda::device::unique_ptr < float[]>	devicePtr_

edm::EDPutTokenT < cms::cuda::Product < cms::cudatest::Thing > > const	dstToken_

TestCUDAProducerGPUKernel	gpuAlgo_

cms::cuda::host::noncached::unique_ptr < float >	hostData_

std::string const	label_

edm::EDGetTokenT < cms::cuda::Product < cms::cudatest::Thing > > const	srcToken_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::ExternalWork >
using	CacheTypes = CacheContexts< T...>

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T...>

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 19 of file TestCUDAProducerGPUEW.cc.

Constructor & Destructor Documentation

TestCUDAProducerGPUEW::TestCUDAProducerGPUEW ( edm::ParameterSet const & iConfig )

explicit

Definition at line 41 of file TestCUDAProducerGPUEW.cc.

References fwrapper::cs, dstToken_, CUDAService::enabled(), edm::ParameterSet::getParameter(), hostData_, srcToken_, and AlCaHLTBitMon_QueryRunRegistry::string.

     : label_{iConfig.getParameter<std::string>("@module_label")},
       srcToken_{consumes<cms::cuda::Product<cms::cudatest::Thing>>(iConfig.getParameter<edm::InputTag>("src"))},
       dstToken_{produces<cms::cuda::Product<cms::cudatest::Thing>>()} {
   edm::Service<CUDAService> cs;
   if (cs->enabled()) {
     hostData_ = cms::cuda::make_host_noncached_unique<float>();
   }
 }

TestCUDAProducerGPUEW::~TestCUDAProducerGPUEW ( )

overridedefault

Member Function Documentation

void TestCUDAProducerGPUEW::acquire	(	edm::Event const &	iEvent,
		edm::EventSetup const &	iSetup,
		edm::WaitingTaskWithArenaHolder	waitingTaskHolder
	)

override

Definition at line 62 of file TestCUDAProducerGPUEW.cc.

References ctxState_, cudaCheck, devicePtr_, edm::EventID::event(), cms::cudatest::Thing::get(), edm::Event::get(), gpuAlgo_, hostData_, edm::EventBase::id(), recoMuon::in, input, label_, eostools::move(), TestCUDAProducerGPUKernel::runAlgo(), srcToken_, and edm::Event::streamID().

                                                                                      {
   edm::LogVerbatim("TestCUDAProducerGPUEW") << label_ << " TestCUDAProducerGPUEW::acquire begin event "
                                             << iEvent.id().event() << " stream " << iEvent.streamID();
 
   auto const& in = iEvent.get(srcToken_);
   cms::cuda::ScopedContextAcquire ctx{in, std::move(waitingTaskHolder), ctxState_};
   cms::cudatest::Thing const& input = ctx.get(in);
 
   devicePtr_ = gpuAlgo_.runAlgo(label_, input.get(), ctx.stream());
   // Mimick the need to transfer some of the GPU data back to CPU to
   // be used for something within this module, or to be put in the
   // event.
   cudaCheck(
       cudaMemcpyAsync(hostData_.get(), devicePtr_.get() + 10, sizeof(float), cudaMemcpyDeviceToHost, ctx.stream()));
   edm::LogVerbatim("TestCUDAProducerGPUEW") << label_ << " TestCUDAProducerGPUEW::acquire end event "
                                             << iEvent.id().event() << " stream " << iEvent.streamID();
 }

void TestCUDAProducerGPUEW::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 51 of file TestCUDAProducerGPUEW.cc.

References edm::ParameterSetDescription::add(), edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, HLT_FULL_cff::InputTag, and edm::ConfigurationDescriptions::setComment().

                                                                                        {
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>("src", edm::InputTag());
   descriptions.addWithDefaultLabel(desc);
   descriptions.setComment(
       "This EDProducer is part of the TestCUDAProducer* family. It models a GPU algorithm this is not the first "
       "algorithm in the chain of the GPU EDProducers, and that transfers some data from GPU to CPU and thus needs to "
       "synchronize GPU and CPU. The synchronization is implemented with the ExternalWork extension. Produces "
       "cms::cuda::Product<cms::cuda::Thing>.");
 }

void TestCUDAProducerGPUEW::produce	(	edm::Event &	iEvent,
		edm::EventSetup const &	iSetup
	)

override

Definition at line 82 of file TestCUDAProducerGPUEW.cc.

References ctxState_, devicePtr_, dstToken_, edm::EventID::event(), hostData_, edm::EventBase::id(), label_, eostools::move(), and edm::Event::streamID().

                                                                                  {
   edm::LogVerbatim("TestCUDAProducerGPUEW")
       << label_ << " TestCUDAProducerGPUEW::produce begin event " << iEvent.id().event() << " stream "
       << iEvent.streamID() << " 10th element " << *hostData_;
 
   cms::cuda::ScopedContextProduce ctx{ctxState_};
 
   ctx.emplace(iEvent, dstToken_, std::move(devicePtr_));
 
   edm::LogVerbatim("TestCUDAProducerGPUEW") << label_ << " TestCUDAProducerGPUEW::produce end event "
                                             << iEvent.id().event() << " stream " << iEvent.streamID();
 }