SiPixelDigisSoAFromCUDA Class Reference

Inheritance diagram for SiPixelDigisSoAFromCUDA:

Public Member Functions
	SiPixelDigisSoAFromCUDA (const edm::ParameterSet &iConfig)
	~SiPixelDigisSoAFromCUDA () 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 Member Functions
void	acquire (const edm::Event &iEvent, const edm::EventSetup &iSetup, edm::WaitingTaskWithArenaHolder waitingTaskHolder) override
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

Private Attributes
cms::cuda::host::unique_ptr< uint16_t[]>	adc_
cms::cuda::host::unique_ptr< int32_t[]>	clus_
edm::EDGetTokenT< cms::cuda::Product< SiPixelDigisCUDA > >	digiGetToken_
edm::EDPutTokenT< SiPixelDigisSoA >	digiPutToken_
int	nDigis_
cms::cuda::host::unique_ptr< uint32_t[]>	pdigi_
cms::cuda::host::unique_ptr< uint32_t[]>	rawIdArr_

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 14 of file SiPixelDigisSoAFromCUDA.cc.

Constructor & Destructor Documentation

SiPixelDigisSoAFromCUDA()

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

explicit

Definition at line 38 of file SiPixelDigisSoAFromCUDA.cc.

    : digiGetToken_(consumes<cms::cuda::Product<SiPixelDigisCUDA>>(iConfig.getParameter<edm::InputTag>("src"))),
      digiPutToken_(produces<SiPixelDigisSoA>()) {}

~SiPixelDigisSoAFromCUDA()

SiPixelDigisSoAFromCUDA::~SiPixelDigisSoAFromCUDA ( )

overridedefault

Member Function Documentation

acquire()

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

overrideprivate

Definition at line 48 of file SiPixelDigisSoAFromCUDA.cc.

                                                                                       {
  // Do the transfer in a CUDA stream parallel to the computation CUDA stream
  cms::cuda::ScopedContextAcquire ctx{iEvent.streamID(), std::move(waitingTaskHolder)};
 
  const auto& gpuDigis = ctx.get(iEvent, digiGetToken_);
 
  nDigis_ = gpuDigis.nDigis();
  pdigi_ = gpuDigis.pdigiToHostAsync(ctx.stream());
  rawIdArr_ = gpuDigis.rawIdArrToHostAsync(ctx.stream());
  adc_ = gpuDigis.adcToHostAsync(ctx.stream());
  clus_ = gpuDigis.clusToHostAsync(ctx.stream());
}

References adc_, clus_, digiGetToken_, iEvent, eostools::move(), nDigis_, pdigi_, and rawIdArr_.

fillDescriptions()

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

static

Definition at line 42 of file SiPixelDigisSoAFromCUDA.cc.

                                                                                         {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("src", edm::InputTag("siPixelClustersCUDA"));
  descriptions.addWithDefaultLabel(desc);
}

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

produce()

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

overrideprivate

Definition at line 63 of file SiPixelDigisSoAFromCUDA.cc.

                                                                                   {
  // The following line copies the data from the pinned host memory to
  // regular host memory. In principle that feels unnecessary (why not
  // just use the pinned host memory?). There are a few arguments for
  // doing it though
  // - Now can release the pinned host memory back to the (caching) allocator
  //   * if we'd like to keep the pinned memory, we'd need to also
  //     keep the CUDA stream around as long as that, or allow pinned
  //     host memory to be allocated without a CUDA stream
  // - What if a CPU algorithm would produce the same SoA? We can't
  //   use cudaMallocHost without a GPU...
  iEvent.emplace(digiPutToken_, nDigis_, pdigi_.get(), rawIdArr_.get(), adc_.get(), clus_.get());
 
  pdigi_.reset();
  rawIdArr_.reset();
  adc_.reset();
  clus_.reset();
}

References adc_, clus_, digiPutToken_, iEvent, nDigis_, pdigi_, and rawIdArr_.

Member Data Documentation

adc_

cms::cuda::host::unique_ptr<uint16_t[]> SiPixelDigisSoAFromCUDA::adc_

private

Definition at line 32 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire(), and produce().

clus_

cms::cuda::host::unique_ptr<int32_t[]> SiPixelDigisSoAFromCUDA::clus_

private

Definition at line 33 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire(), and produce().

digiGetToken_

edm::EDGetTokenT<cms::cuda::Product<SiPixelDigisCUDA> > SiPixelDigisSoAFromCUDA::digiGetToken_

private

Definition at line 27 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire().

digiPutToken_

edm::EDPutTokenT<SiPixelDigisSoA> SiPixelDigisSoAFromCUDA::digiPutToken_

private

Definition at line 28 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by produce().

nDigis_

int SiPixelDigisSoAFromCUDA::nDigis_

private

Definition at line 35 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire(), and produce().

pdigi_

cms::cuda::host::unique_ptr<uint32_t[]> SiPixelDigisSoAFromCUDA::pdigi_

private

Definition at line 30 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire(), and produce().

rawIdArr_

cms::cuda::host::unique_ptr<uint32_t[]> SiPixelDigisSoAFromCUDA::rawIdArr_

private

Definition at line 31 of file SiPixelDigisSoAFromCUDA.cc.

Referenced by acquire(), and produce().