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	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () 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::ParameterSetDescription::add(), edm::ConfigurationDescriptions::addWithDefaultLabel(), HLT_2018_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;
 }