EcalUncalibRecHitPhase2WeightsProducerGPU Class Reference

Inheritance diagram for EcalUncalibRecHitPhase2WeightsProducerGPU:

Public Member Functions
	EcalUncalibRecHitPhase2WeightsProducerGPU (edm::ParameterSet const &ps)
	~EcalUncalibRecHitPhase2WeightsProducerGPU () 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::DigisCollection< calo::common::DevStoragePolicy > >
using	OutputProduct = cms::cuda::Product< ecal::UncalibratedRecHit< calo::common::DevStoragePolicy > >

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

Private Attributes
cms::cuda::ContextState	cudaState_
const edm::EDGetTokenT< InputProduct >	digisToken_
ecal::weights::EventOutputDataGPU	eventOutputDataGPU_
const edm::EDPutTokenT< OutputProduct >	recHitsToken_
uint32_t	size_
const std::vector< double, cms::cuda::HostAllocator< double > >	weights_

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

Member Typedef Documentation

InputProduct

using EcalUncalibRecHitPhase2WeightsProducerGPU::InputProduct = cms::cuda::Product<ecal::DigisCollection<calo::common::DevStoragePolicy> >

private

Definition at line 27 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

OutputProduct

using EcalUncalibRecHitPhase2WeightsProducerGPU::OutputProduct = cms::cuda::Product<ecal::UncalibratedRecHit<calo::common::DevStoragePolicy> >

private

Definition at line 29 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Constructor & Destructor Documentation

EcalUncalibRecHitPhase2WeightsProducerGPU()

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

explicit

Definition at line 41 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

References createJobs::tmp.

    :  // use lambda to initialise the vector with CUDA::HostAllocator from a normal vector
      weights_([tmp = ps.getParameter<std::vector<double>>("weights")] {
        return std::vector<double, cms::cuda::HostAllocator<double>>(tmp.begin(), tmp.end());
      }()),
      digisToken_{consumes<InputProduct>(ps.getParameter<edm::InputTag>("digisLabelEB"))},
      recHitsToken_{produces<OutputProduct>(ps.getParameter<std::string>("recHitsLabelEB"))} {}

~EcalUncalibRecHitPhase2WeightsProducerGPU()

EcalUncalibRecHitPhase2WeightsProducerGPU::~EcalUncalibRecHitPhase2WeightsProducerGPU ( )

overridedefault

Member Function Documentation

acquire()

void EcalUncalibRecHitPhase2WeightsProducerGPU::acquire ( edm::Event const &  event, edm::EventSetup const &  setup, edm::WaitingTaskWithArenaHolder  holder  )

overrideprivate

Definition at line 78 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

References ecal::weights::EventOutputDataGPU::allocate(), cudaCheck, cudaState_, digisToken_, eventOutputDataGPU_, EcalDataFrame_Ph2::MAXSAMPLES, eostools::move(), ecal::weights::phase2Weights(), size_, and weights_.

                                                                                              {
  // cuda products
  auto const &digisProduct = event.get(digisToken_);
  // raii
  cms::cuda::ScopedContextAcquire ctx{digisProduct, std::move(holder), cudaState_};

  // get actual obj
  auto const &digis = ctx.get(digisProduct);

  size_ = digis.size;

  // if no digis stop here
  if (size_ == 0)
    return;

  auto weights_d = cms::cuda::make_device_unique<double[]>(EcalDataFrame_Ph2::MAXSAMPLES, ctx.stream());

  cudaCheck(cudaMemcpyAsync(weights_d.get(),
                            weights_.data(),
                            EcalDataFrame_Ph2::MAXSAMPLES * sizeof(double),
                            cudaMemcpyHostToDevice,
                            ctx.stream()));

  // output on GPU
  eventOutputDataGPU_.allocate(size_, ctx.stream());

  ecal::weights::phase2Weights(digis, eventOutputDataGPU_, weights_d, ctx.stream());
}

fillDescriptions()

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

static

Definition at line 49 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

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

                                                                                                           {
  edm::ParameterSetDescription desc;

  desc.add<std::string>("recHitsLabelEB", "EcalUncalibRecHitsEB");
  //The below weights values should be kept up to date with those on the CPU version of this module
  desc.add<std::vector<double>>("weights",
                                {-0.121016,
                                 -0.119899,
                                 -0.120923,
                                 -0.0848959,
                                 0.261041,
                                 0.509881,
                                 0.373591,
                                 0.134899,
                                 -0.0233605,
                                 -0.0913195,
                                 -0.112452,
                                 -0.118596,
                                 -0.121737,
                                 -0.121737,
                                 -0.121737,
                                 -0.121737});

  desc.add<edm::InputTag>("digisLabelEB", edm::InputTag("simEcalUnsuppressedDigis", ""));

  descriptions.addWithDefaultLabel(desc);
}

produce()

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

overrideprivate

Definition at line 109 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

References cudaState_, eventOutputDataGPU_, eostools::move(), ecal::weights::EventOutputDataGPU::recHits, recHitsToken_, and size_.

                                                                                                   {
  cms::cuda::ScopedContextProduce ctx{cudaState_};

  // set the size of digis
  eventOutputDataGPU_.recHits.size = size_;

  // put into the event
  ctx.emplace(event, recHitsToken_, std::move(eventOutputDataGPU_.recHits));
}

Member Data Documentation

cudaState_

cms::cuda::ContextState EcalUncalibRecHitPhase2WeightsProducerGPU::cudaState_

private

Definition at line 35 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by acquire(), and produce().

digisToken_

const edm::EDGetTokenT<InputProduct> EcalUncalibRecHitPhase2WeightsProducerGPU::digisToken_

private

Definition at line 28 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by acquire().

eventOutputDataGPU_

ecal::weights::EventOutputDataGPU EcalUncalibRecHitPhase2WeightsProducerGPU::eventOutputDataGPU_

private

Definition at line 33 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by acquire(), and produce().

recHitsToken_

const edm::EDPutTokenT<OutputProduct> EcalUncalibRecHitPhase2WeightsProducerGPU::recHitsToken_

private

Definition at line 30 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by produce().

size_

uint32_t EcalUncalibRecHitPhase2WeightsProducerGPU::size_

private

Definition at line 37 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by acquire(), and produce().

weights_

const std::vector<double, cms::cuda::HostAllocator<double> > EcalUncalibRecHitPhase2WeightsProducerGPU::weights_

private

Definition at line 25 of file EcalUncalibRecHitPhase2WeightsProducerGPU.cc.

Referenced by acquire().