TestCUDAProducerGPUEW Class Reference

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::TestCUDAProducerGPUEW ( edm::ParameterSet const &  iConfig )

explicit

Definition at line 41 of file TestCUDAProducerGPUEW.cc.

References edm::ParameterSet::getParameter(), 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::~TestCUDAProducerGPUEW ( )

overridedefault

Member Function Documentation

acquire()

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_, dqmMemoryStats::float, gpuAlgo_, hostData_, iEvent, recoMuon::in, input, label_, eostools::move(), TestCUDAProducerGPUKernel::runAlgo(), and srcToken_.

                                                                                     {
  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();
}

fillDescriptions()

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

static

Definition at line 51 of file TestCUDAProducerGPUEW.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, HLT_2022v12_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>.");
}

produce()

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

override

Definition at line 82 of file TestCUDAProducerGPUEW.cc.

References ctxState_, devicePtr_, dstToken_, hostData_, iEvent, label_, and eostools::move().

                                                                                 {
  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();
}

Member Data Documentation

ctxState_

cms::cuda::ContextState TestCUDAProducerGPUEW::ctxState_

private

Definition at line 36 of file TestCUDAProducerGPUEW.cc.

Referenced by acquire(), and produce().

devicePtr_

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

private

Definition at line 37 of file TestCUDAProducerGPUEW.cc.

Referenced by acquire(), and produce().

dstToken_

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

private

Definition at line 34 of file TestCUDAProducerGPUEW.cc.

Referenced by produce().

gpuAlgo_

TestCUDAProducerGPUKernel TestCUDAProducerGPUEW::gpuAlgo_

private

Definition at line 35 of file TestCUDAProducerGPUEW.cc.

Referenced by acquire().

hostData_

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

private

Definition at line 38 of file TestCUDAProducerGPUEW.cc.

Referenced by acquire(), and produce().

label_

std::string const TestCUDAProducerGPUEW::label_

private

Definition at line 32 of file TestCUDAProducerGPUEW.cc.

Referenced by Modules.SwitchProducer::__addParameter(), SequenceTypes.DummyModule::__repr__(), Modules.SwitchProducer::__setattr__(), Mixins._Labelable::_findDependencies(), acquire(), produce(), and Mixins._Labelable::setLabel().

srcToken_

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

private

Definition at line 33 of file TestCUDAProducerGPUEW.cc.

Referenced by acquire().