Inheritance diagram for TestCUDAProducerGPUtoCPU:

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

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

	TestCUDAProducerGPUtoCPU (edm::ParameterSet const &iConfig)

	~TestCUDAProducerGPUtoCPU () override=default

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

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

Private Attributes
cms::cuda::host::unique_ptr< float[]>	buffer_

const edm::EDPutTokenT< int >	dstToken_

const std::string	label_

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

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

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 16 of file TestCUDAProducerGPUtoCPU.cc.

Constructor & Destructor Documentation

◆ TestCUDAProducerGPUtoCPU()

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

explicit

Definition at line 36 of file TestCUDAProducerGPUtoCPU.cc.

     : label_{iConfig.getParameter<std::string>("@module_label")},
       srcToken_{consumes<cms::cuda::Product<cms::cudatest::Thing>>(iConfig.getParameter<edm::InputTag>("src"))},
       dstToken_{produces<int>()} {}

References edm::ParameterSet::getParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

◆ ~TestCUDAProducerGPUtoCPU()

TestCUDAProducerGPUtoCPU::~TestCUDAProducerGPUtoCPU ( )

overridedefault

Member Function Documentation

◆ acquire()

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

override

Definition at line 50 of file TestCUDAProducerGPUtoCPU.cc.

                                                                                         {
   edm::LogVerbatim("TestCUDAProducerGPUtoCPU") << label_ << " TestCUDAProducerGPUtoCPU::acquire begin event "
                                                << iEvent.id().event() << " stream " << iEvent.streamID();
  
   auto const& in = iEvent.get(srcToken_);
   cms::cuda::ScopedContextAcquire ctx{in, std::move(waitingTaskHolder)};
   cms::cudatest::Thing const& device = ctx.get(in);
  
   buffer_ = cms::cuda::make_host_unique<float[]>(TestCUDAProducerGPUKernel::NUM_VALUES, ctx.stream());
   // Enqueue async copy, continue in produce once finished
   cudaCheck(cudaMemcpyAsync(buffer_.get(),
                             device.get(),
                             TestCUDAProducerGPUKernel::NUM_VALUES * sizeof(float),
                             cudaMemcpyDeviceToHost,
                             ctx.stream()));
  
   edm::LogVerbatim("TestCUDAProducerGPUtoCPU") << label_ << " TestCUDAProducerGPUtoCPU::acquire end event "
                                                << iEvent.id().event() << " stream " << iEvent.streamID();
 }

References buffer_, cudaCheck, dqmMemoryStats::float, cms::cudatest::Thing::get(), iEvent, recoMuon::in, label_, eostools::move(), TestCUDAProducerGPUKernel::NUM_VALUES, and srcToken_.

◆ fillDescriptions()

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

static

Definition at line 41 of file TestCUDAProducerGPUtoCPU.cc.

                                                                                           {
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>("src", edm::InputTag())->setComment("Source for cms::cuda::Product<cms::cudatest::Thing>.");
   descriptions.addWithDefaultLabel(desc);
   descriptions.setComment(
       "This EDProducer is part of the TestCUDAProducer* family. It models the GPU->CPU data transfer and formatting of "
       "the data to legacy data format. Produces int, to be compatible with TestCUDAProducerCPU.");
 }

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

◆ produce()

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

override

Definition at line 72 of file TestCUDAProducerGPUtoCPU.cc.

                                                                                     {
   edm::LogVerbatim("TestCUDAProducerGPUtoCPU") << label_ << " TestCUDAProducerGPUtoCPU::produce begin event "
                                                << iEvent.id().event() << " stream " << iEvent.streamID();
  
   int counter = 0;
   for (int i = 0; i < TestCUDAProducerGPUKernel::NUM_VALUES; ++i) {
     counter += buffer_[i];
   }
   buffer_.reset();  // not so nice, but no way around?
  
   iEvent.emplace(dstToken_, counter);
  
   edm::LogVerbatim("TestCUDAProducerGPUtoCPU")
       << label_ << " TestCUDAProducerGPUtoCPU::produce end event " << iEvent.id().event() << " stream "
       << iEvent.streamID() << " result " << counter;
 }