EcalCPURecHitProducer Class Reference

Inheritance diagram for EcalCPURecHitProducer:

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

Private Types
using	InputProduct = cms::cuda::Product< ecal::RecHit< calo::common::DevStoragePolicy >>
using	OutputProduct = ecal::RecHit< calo::common::VecStoragePolicy< calo::common::CUDAHostAllocatorAlias >>

Private Member Functions
void	acquire (edm::Event const &, edm::EventSetup const &, edm::WaitingTaskWithArenaHolder) override
void	produce (edm::Event &, edm::EventSetup const &) override

Private Attributes
bool	containsTimingInformation_
OutputProduct	recHitsEB_
OutputProduct	recHitsEE_
edm::EDGetTokenT< InputProduct >	recHitsInEBToken_
edm::EDGetTokenT< InputProduct >	recHitsInEEToken_
edm::EDPutTokenT< OutputProduct >	recHitsOutEBToken_
edm::EDPutTokenT< OutputProduct >	recHitsOutEEToken_

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 24 of file EcalCPURecHitProducer.cc.

Member Typedef Documentation

using EcalCPURecHitProducer::InputProduct = cms::cuda::Product<ecal::RecHit<calo::common::DevStoragePolicy>>

private

Definition at line 35 of file EcalCPURecHitProducer.cc.

using EcalCPURecHitProducer::OutputProduct = ecal::RecHit<calo::common::VecStoragePolicy<calo::common::CUDAHostAllocatorAlias>>

private

Definition at line 37 of file EcalCPURecHitProducer.cc.

Constructor & Destructor Documentation

EcalCPURecHitProducer::EcalCPURecHitProducer ( edm::ParameterSet const &  ps )

explicit

Definition at line 56 of file EcalCPURecHitProducer.cc.

References edm::ParameterSet::getParameter().

    : recHitsInEBToken_{consumes<InputProduct>(ps.getParameter<edm::InputTag>("recHitsInLabelEB"))},
      recHitsInEEToken_{consumes<InputProduct>(ps.getParameter<edm::InputTag>("recHitsInLabelEE"))},
      recHitsOutEBToken_{produces<OutputProduct>(ps.getParameter<std::string>("recHitsOutLabelEB"))},
      recHitsOutEEToken_{produces<OutputProduct>(ps.getParameter<std::string>("recHitsOutLabelEE"))},
      containsTimingInformation_{ps.getParameter<bool>("containsTimingInformation")} {}

EcalCPURecHitProducer::~EcalCPURecHitProducer ( )

overridedefault

Member Function Documentation

void EcalCPURecHitProducer::acquire ( edm::Event const &  event, edm::EventSetup const &  setup, edm::WaitingTaskWithArenaHolder  taskHolder  )

overrideprivate

Definition at line 63 of file EcalCPURecHitProducer.cc.

References ecal::RecHit< StoragePolicy >::chi2, gather_cfg::cout, cudaCheck, ecal::RecHit< StoragePolicy >::did, ecal::RecHit< StoragePolicy >::energy, ecal::RecHit< StoragePolicy >::extra, ecal::RecHit< StoragePolicy >::flagBits, eostools::move(), recHitsEB_, recHitsEE_, recHitsInEBToken_, recHitsInEEToken_, and ecal::RecHit< StoragePolicy >::resize().

                                                                              {
  // retrieve data/ctx
  auto const& ebRecHitsProduct = event.get(recHitsInEBToken_);
  auto const& eeRecHitsProduct = event.get(recHitsInEEToken_);
  cms::cuda::ScopedContextAcquire ctx{ebRecHitsProduct, std::move(taskHolder)};
  auto const& ebRecHits = ctx.get(ebRecHitsProduct);
  auto const& eeRecHits = ctx.get(eeRecHitsProduct);

  // resize the output buffers
  recHitsEB_.resize(ebRecHits.size);
  recHitsEE_.resize(eeRecHits.size);

#ifdef ECAL_RECO_CUDA_DEBUG
  std::cout << " [EcalCPURecHitProducer::acquire] ebRecHits.size = " << ebRecHits.size << std::endl;
  std::cout << " [EcalCPURecHitProducer::acquire] eeRecHits.size = " << eeRecHits.size << std::endl;
#endif

  // enqeue transfers
  cudaCheck(cudaMemcpyAsync(recHitsEB_.did.data(),
                            ebRecHits.did.get(),
                            recHitsEB_.did.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  cudaCheck(cudaMemcpyAsync(recHitsEE_.did.data(),
                            eeRecHits.did.get(),
                            recHitsEE_.did.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  //
  //     ./CUDADataFormats/EcalRecHitSoA/interface/RecoTypes.h:using StorageScalarType = float;
  //

  cudaCheck(cudaMemcpyAsync(recHitsEB_.energy.data(),
                            ebRecHits.energy.get(),
                            recHitsEB_.energy.size() * sizeof(::ecal::reco::StorageScalarType),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  cudaCheck(cudaMemcpyAsync(recHitsEE_.energy.data(),
                            eeRecHits.energy.get(),
                            recHitsEE_.energy.size() * sizeof(::ecal::reco::StorageScalarType),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));

  cudaCheck(cudaMemcpyAsync(recHitsEB_.chi2.data(),
                            ebRecHits.chi2.get(),
                            recHitsEB_.chi2.size() * sizeof(::ecal::reco::StorageScalarType),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  cudaCheck(cudaMemcpyAsync(recHitsEE_.chi2.data(),
                            eeRecHits.chi2.get(),
                            recHitsEE_.chi2.size() * sizeof(::ecal::reco::StorageScalarType),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));

  cudaCheck(cudaMemcpyAsync(recHitsEB_.extra.data(),
                            ebRecHits.extra.get(),
                            recHitsEB_.extra.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  cudaCheck(cudaMemcpyAsync(recHitsEE_.extra.data(),
                            eeRecHits.extra.get(),
                            recHitsEE_.extra.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));

  cudaCheck(cudaMemcpyAsync(recHitsEB_.flagBits.data(),
                            ebRecHits.flagBits.get(),
                            recHitsEB_.flagBits.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));
  cudaCheck(cudaMemcpyAsync(recHitsEE_.flagBits.data(),
                            eeRecHits.flagBits.get(),
                            recHitsEE_.flagBits.size() * sizeof(uint32_t),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));

#ifdef ECAL_RECO_CUDA_DEBUG
  for (unsigned int ieb = 0; ieb < ebRecHits.size; ieb++) {
    if (recHitsEB_.extra[ieb] != 0)
      std::cout << " [ " << ieb << " :: " << ebRecHits.size << " ] [ " << recHitsEB_.did[ieb]
                << " ] eb extra = " << recHitsEB_.extra[ieb] << std::endl;
  }

  for (unsigned int ieb = 0; ieb < ebRecHits.size; ieb++) {
    if (recHitsEB_.energy[ieb] != 0)
      std::cout << " [ " << ieb << " :: " << ebRecHits.size << " ] [ " << recHitsEB_.did[ieb]
                << " ] eb energy = " << recHitsEB_.energy[ieb] << std::endl;
  }

  for (unsigned int iee = 0; iee < eeRecHits.size; iee++) {
    if (recHitsEE_.energy[iee] != 0)
      std::cout << " [ " << iee << " :: " << eeRecHits.size << " ] [ " << recHitsEE_.did[iee]
                << " ] ee energy = " << recHitsEE_.energy[iee] << std::endl;
  }
#endif
}

void EcalCPURecHitProducer::fillDescriptions ( edm::ConfigurationDescriptions &  confDesc )

static

Definition at line 44 of file EcalCPURecHitProducer.cc.

References edm::ParameterSetDescription::add(), edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                   {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("recHitsInLabelEB", edm::InputTag{"ecalRecHitProducerGPU", "EcalRecHitsEB"});
  desc.add<edm::InputTag>("recHitsInLabelEE", edm::InputTag{"ecalRecHitProducerGPU", "EcalRecHitsEE"});
  desc.add<std::string>("recHitsOutLabelEB", "EcalRecHitsEB");
  desc.add<std::string>("recHitsOutLabelEE", "EcalRecHitsEE");
  desc.add<bool>("containsTimingInformation", false);

  confDesc.addWithDefaultLabel(desc);
}

void EcalCPURecHitProducer::produce ( edm::Event &  event, edm::EventSetup const &  setup  )

overrideprivate

Definition at line 162 of file EcalCPURecHitProducer.cc.

References eostools::move(), recHitsEB_, recHitsEE_, recHitsOutEBToken_, and recHitsOutEEToken_.

                                                                               {
  // put into event
  event.emplace(recHitsOutEBToken_, std::move(recHitsEB_));
  event.emplace(recHitsOutEEToken_, std::move(recHitsEE_));
}

Member Data Documentation

bool EcalCPURecHitProducer::containsTimingInformation_

private

Definition at line 41 of file EcalCPURecHitProducer.cc.

OutputProduct EcalCPURecHitProducer::recHitsEB_

private

Definition at line 40 of file EcalCPURecHitProducer.cc.

Referenced by acquire(), and produce().

OutputProduct EcalCPURecHitProducer::recHitsEE_

private

Definition at line 40 of file EcalCPURecHitProducer.cc.

Referenced by acquire(), and produce().

edm::EDGetTokenT<InputProduct> EcalCPURecHitProducer::recHitsInEBToken_

private

Definition at line 36 of file EcalCPURecHitProducer.cc.

Referenced by acquire().

edm::EDGetTokenT<InputProduct> EcalCPURecHitProducer::recHitsInEEToken_

private

Definition at line 36 of file EcalCPURecHitProducer.cc.

Referenced by acquire().

edm::EDPutTokenT<OutputProduct> EcalCPURecHitProducer::recHitsOutEBToken_

private

Definition at line 38 of file EcalCPURecHitProducer.cc.

Referenced by produce().

edm::EDPutTokenT<OutputProduct> EcalCPURecHitProducer::recHitsOutEEToken_

private

Definition at line 38 of file EcalCPURecHitProducer.cc.

Referenced by produce().