PixelVertexSoAFromCUDA Class Reference

Inheritance diagram for PixelVertexSoAFromCUDA:

Public Member Functions
	PixelVertexSoAFromCUDA (const edm::ParameterSet &iConfig)
	~PixelVertexSoAFromCUDA () 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 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

Private Attributes
edm::EDGetTokenT< cms::cuda::Product< ZVertexSoADevice > >	tokenCUDA_
edm::EDPutTokenT< ZVertexSoAHost >	tokenSOA_
ZVertexSoAHost	zvertex_h

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 21 of file PixelVertexSoAFromCUDA.cc.

Constructor & Destructor Documentation

PixelVertexSoAFromCUDA()

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

explicit

Definition at line 40 of file PixelVertexSoAFromCUDA.cc.

    : tokenCUDA_(consumes<cms::cuda::Product<ZVertexSoADevice>>(iConfig.getParameter<edm::InputTag>("src"))),
      tokenSOA_(produces<ZVertexSoAHost>()) {}

~PixelVertexSoAFromCUDA()

PixelVertexSoAFromCUDA::~PixelVertexSoAFromCUDA ( )

overridedefault

Member Function Documentation

acquire()

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

overrideprivate

Definition at line 51 of file PixelVertexSoAFromCUDA.cc.

References cms::cuda::PortableHostCollection< T >::buffer(), cudaCheck, iEvent, eostools::move(), tokenCUDA_, and zvertex_h.

                                                                                      {
  cms::cuda::Product<ZVertexSoADevice> const& inputDataWrapped = iEvent.get(tokenCUDA_);
  cms::cuda::ScopedContextAcquire ctx{inputDataWrapped, std::move(waitingTaskHolder)};
  auto const& zvertex_d = ctx.get(inputDataWrapped);  // Tracks on device
  zvertex_h = ZVertexSoAHost(ctx.stream());           // Create an instance of Tracks on Host, using the stream
  cudaCheck(cudaMemcpyAsync(zvertex_h.buffer().get(),
                            zvertex_d.const_buffer().get(),
                            zvertex_d.bufferSize(),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));  // Copy data from Device to Host
}

fillDescriptions()

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

static

Definition at line 44 of file PixelVertexSoAFromCUDA.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

                                                                                        {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("src", edm::InputTag("pixelVerticesCUDA"));
  descriptions.add("pixelVerticesSoA", desc);
}

produce()

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

overrideprivate

Definition at line 65 of file PixelVertexSoAFromCUDA.cc.

References iEvent, eostools::move(), tokenSOA_, and zvertex_h.

                                                                                  {
  // No copies....
  iEvent.emplace(tokenSOA_, std::move(zvertex_h));
}

Member Data Documentation

tokenCUDA_

edm::EDGetTokenT<cms::cuda::Product<ZVertexSoADevice> > PixelVertexSoAFromCUDA::tokenCUDA_

private

Definition at line 34 of file PixelVertexSoAFromCUDA.cc.

Referenced by acquire().

tokenSOA_

edm::EDPutTokenT<ZVertexSoAHost> PixelVertexSoAFromCUDA::tokenSOA_

private

Definition at line 35 of file PixelVertexSoAFromCUDA.cc.

Referenced by produce().

zvertex_h

ZVertexSoAHost PixelVertexSoAFromCUDA::zvertex_h

private

Definition at line 37 of file PixelVertexSoAFromCUDA.cc.

Referenced by acquire(), and produce().