HcalCPURecHitsProducer Class Reference

Inheritance diagram for HcalCPURecHitsProducer:

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

Private Types
using	IProductType = cms::cuda::Product< hcal::RecHitCollection< calo::common::DevStoragePolicy > >
using	OProductType = hcal::RecHitCollection< 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
const bool	produceLegacy_
const bool	produceSoA_
const edm::EDPutTokenT< HBHERecHitCollection >	recHitsLegacyTokenOut_
const edm::EDGetTokenT< IProductType >	recHitsM0TokenIn_
const edm::EDPutTokenT< OProductType >	recHitsM0TokenOut_
OProductType	tmpRecHits_

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 15 of file HcalCPURecHitsProducer.cc.

Member Typedef Documentation

IProductType

using HcalCPURecHitsProducer::IProductType = cms::cuda::Product<hcal::RecHitCollection<calo::common::DevStoragePolicy> >

private

Definition at line 30 of file HcalCPURecHitsProducer.cc.

OProductType

using HcalCPURecHitsProducer::OProductType = hcal::RecHitCollection<calo::common::VecStoragePolicy<calo::common::CUDAHostAllocatorAlias> >

private

Definition at line 33 of file HcalCPURecHitsProducer.cc.

Constructor & Destructor Documentation

HcalCPURecHitsProducer()

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

explicit

Definition at line 53 of file HcalCPURecHitsProducer.cc.

    : produceSoA_{ps.getParameter<bool>("produceSoA")},
      produceLegacy_{ps.getParameter<bool>("produceLegacy")},
      recHitsM0TokenIn_{consumes<IProductType>(ps.getParameter<edm::InputTag>("recHitsM0LabelIn"))},
      recHitsM0TokenOut_{produceSoA_ ? produces<OProductType>(ps.getParameter<std::string>("recHitsM0LabelOut"))
                                     : edm::EDPutTokenT<OProductType>{}},  // empty token if disabled
      recHitsLegacyTokenOut_{produceLegacy_
                                 ? produces<HBHERecHitCollection>(ps.getParameter<std::string>("recHitsLegacyLabelOut"))
                                 : edm::EDPutTokenT<HBHERecHitCollection>{}}  // empty token if disabled
{}

References edm::ParameterSet::getParameter().

~HcalCPURecHitsProducer()

HcalCPURecHitsProducer::~HcalCPURecHitsProducer ( )

overridedefault

Member Function Documentation

acquire()

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

overrideprivate

Definition at line 64 of file HcalCPURecHitsProducer.cc.

                                                                               {
  // retrieve data/ctx
  auto const& recHitsProduct = event.get(recHitsM0TokenIn_);
  cms::cuda::ScopedContextAcquire ctx{recHitsProduct, std::move(taskHolder)};
  auto const& recHits = ctx.get(recHitsProduct);
 
  // resize tmp buffers
  tmpRecHits_.resize(recHits.size);
 
#ifdef HCAL_MAHI_CPUDEBUG
  std::cout << "num rec Hits = " << recHits.size << std::endl;
#endif
 
  // do not try to copy the rechits if they are empty
  if (recHits.size == 0) {
    return;
  }
 
  auto lambdaToTransfer = [&ctx](auto& dest, auto* src) {
    using vector_type = typename std::remove_reference<decltype(dest)>::type;
    using src_data_type = typename std::remove_pointer<decltype(src)>::type;
    using type = typename vector_type::value_type;
    static_assert(std::is_same<src_data_type, type>::value && "Dest and Src data types do not match");
    cudaCheck(cudaMemcpyAsync(dest.data(), src, dest.size() * sizeof(type), cudaMemcpyDeviceToHost, ctx.stream()));
  };
 
  lambdaToTransfer(tmpRecHits_.energy, recHits.energy.get());
  lambdaToTransfer(tmpRecHits_.chi2, recHits.chi2.get());
  lambdaToTransfer(tmpRecHits_.energyM0, recHits.energyM0.get());
  lambdaToTransfer(tmpRecHits_.timeM0, recHits.timeM0.get());
  lambdaToTransfer(tmpRecHits_.did, recHits.did.get());
}

References hcal::RecHitCollection< StoragePolicy >::chi2, gather_cfg::cout, cudaCheck, mps_fire::dest, hcal::RecHitCollection< StoragePolicy >::did, hcal::RecHitCollection< StoragePolicy >::energy, hcal::RecHitCollection< StoragePolicy >::energyM0, eostools::move(), FastTrackerRecHitMaskProducer_cfi::recHits, recHitsM0TokenIn_, hcal::RecHitCollection< StoragePolicy >::resize(), TrackRefitter_38T_cff::src, hcal::RecHitCollection< StoragePolicy >::timeM0, tmpRecHits_, and relativeConstraints::value.

fillDescriptions()

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

static

Definition at line 41 of file HcalCPURecHitsProducer.cc.

                                                                                    {
  edm::ParameterSetDescription desc;
 
  desc.add<edm::InputTag>("recHitsM0LabelIn", edm::InputTag{"hbheRecHitProducerGPU"});
  desc.add<std::string>("recHitsM0LabelOut", "");
  desc.add<std::string>("recHitsLegacyLabelOut", "");
  desc.add<bool>("produceSoA", true);
  desc.add<bool>("produceLegacy", true);
 
  confDesc.addWithDefaultLabel(desc);
}

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

produce()

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

overrideprivate

Definition at line 99 of file HcalCPURecHitsProducer.cc.

                                                                                {
  if (produceLegacy_) {
    // populate the legacy collection
    auto recHitsLegacy = std::make_unique<HBHERecHitCollection>();
    // did not set size with ctor as there is no setter for did
    recHitsLegacy->reserve(tmpRecHits_.did.size());
    for (uint32_t i = 0; i < tmpRecHits_.did.size(); i++) {
      recHitsLegacy->emplace_back(HcalDetId{tmpRecHits_.did[i]},
                                  tmpRecHits_.energy[i],
                                  0  // timeRising
      );
 
      // update newly pushed guy
      (*recHitsLegacy)[i].setChiSquared(tmpRecHits_.chi2[i]);
      (*recHitsLegacy)[i].setRawEnergy(tmpRecHits_.energyM0[i]);
    }
 
    // put the legacy collection
    event.put(recHitsLegacyTokenOut_, std::move(recHitsLegacy));
  }
 
  if (produceSoA_) {
    // put the SoA collection
    event.emplace(recHitsM0TokenOut_, std::move(tmpRecHits_));
  }
  // clear the temporary collection for the next event
  tmpRecHits_.resize(0);
}

References hcal::RecHitCollection< StoragePolicy >::chi2, hcal::RecHitCollection< StoragePolicy >::did, hcal::RecHitCollection< StoragePolicy >::energy, hcal::RecHitCollection< StoragePolicy >::energyM0, mps_fire::i, eostools::move(), produceLegacy_, produceSoA_, recHitsLegacyTokenOut_, recHitsM0TokenOut_, hcal::RecHitCollection< StoragePolicy >::resize(), and tmpRecHits_.

Member Data Documentation

produceLegacy_

const bool HcalCPURecHitsProducer::produceLegacy_

private

Definition at line 28 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

produceSoA_

const bool HcalCPURecHitsProducer::produceSoA_

private

Definition at line 27 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

recHitsLegacyTokenOut_

const edm::EDPutTokenT<HBHERecHitCollection> HcalCPURecHitsProducer::recHitsLegacyTokenOut_

private

Definition at line 35 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

recHitsM0TokenIn_

const edm::EDGetTokenT<IProductType> HcalCPURecHitsProducer::recHitsM0TokenIn_

private

Definition at line 31 of file HcalCPURecHitsProducer.cc.

Referenced by acquire().

recHitsM0TokenOut_

const edm::EDPutTokenT<OProductType> HcalCPURecHitsProducer::recHitsM0TokenOut_

private

Definition at line 34 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

tmpRecHits_

OProductType HcalCPURecHitsProducer::tmpRecHits_

private

Definition at line 38 of file HcalCPURecHitsProducer.cc.

Referenced by acquire(), and produce().