ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer Class Reference

Inheritance diagram for ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer:

Public Member Functions
	PFClusterSoAProducer (edm::ParameterSet const &config)
void	produce (device::Event &event, device::EventSetup const &setup) override
Public Member Functions inherited from ALPAKA_ACCELERATOR_NAMESPACE::stream::EDProducer< Args >
void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup) final
Public Member Functions inherited from edm::ProducerBase< edm::stream::EDProducer, Args... >
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const
	ProducerBase ()
std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const
std::vector< bool > const &	recordProvenanceList () const
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...
	~ProducerBase () noexcept(false) override

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
const edm::EDGetTokenT< reco::PFRecHitHostCollection >	inputPFRecHitSoA_Token_
const device::EDPutToken< reco::PFClusterDeviceCollection >	outputPFClusterSoA_Token_
const device::EDPutToken< reco::PFRecHitFractionDeviceCollection >	outputPFRHFractionSoA_Token_
const device::ESGetToken< reco::PFClusterParamsDeviceCollection, JobConfigurationGPURecord >	pfClusParamsToken
const int	pfRecHitFractionAllocation_
const bool	synchronise_
const device::ESGetToken< reco::PFRecHitHCALTopologyDeviceCollection, PFRecHitHCALTopologyRecord >	topologyToken_

Additional Inherited Members
Public Types inherited from edm::ProducerBase< edm::stream::EDProducer, Args... >
using	BranchAliasSetterT = ProductRegistryHelper::BranchAliasSetterT< T >
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex > >
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Protected Member Functions inherited from edm::ProducerBase< edm::stream::EDProducer, Args... >
auto	produces (std::string instanceName) noexcept
	declare what type of product will make and with which optional label More...
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
BranchAliasSetterT< ProductType >	produces ()
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
BranchAliasSetterT< ProductType >	produces ()
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
BranchAliasSetterT< ProductType >	produces ()
auto	produces () noexcept
ProducesCollector	producesCollector ()

Detailed Description

Definition at line 19 of file PFClusterSoAProducer.cc.

Constructor & Destructor Documentation

PFClusterSoAProducer()

ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::PFClusterSoAProducer ( edm::ParameterSet const &  config )

inline

Definition at line 21 of file PFClusterSoAProducer.cc.

        : pfClusParamsToken(esConsumes(config.getParameter<edm::ESInputTag>("pfClusterParams"))),
          topologyToken_(esConsumes(config.getParameter<edm::ESInputTag>("topology"))),
          inputPFRecHitSoA_Token_{consumes(config.getParameter<edm::InputTag>("pfRecHits"))},
          outputPFClusterSoA_Token_{produces()},
          outputPFRHFractionSoA_Token_{produces()},
          synchronise_(config.getParameter<bool>("synchronise")),
          pfRecHitFractionAllocation_(config.getParameter<int>("pfRecHitFractionAllocation")) {
      // Workaround until the ProductID problem in issue https://github.com/cms-sw/cmssw/issues/44643 is fixed
#ifdef ALPAKA_ACC_CPU_B_SEQ_T_SEQ_ENABLED
      producesTemporarily("edm::DeviceProduct<alpaka_cuda_async::reco::PFClusterDeviceCollection>");
      producesTemporarily("edm::DeviceProduct<alpaka_cuda_async::reco::PFRecHitFractionDeviceCollection>");
#endif
    }

Member Function Documentation

fillDescriptions()

static void ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

inlinestatic

Definition at line 68 of file PFClusterSoAProducer.cc.

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

                                                                             {
      edm::ParameterSetDescription desc;
      desc.add<edm::InputTag>("pfRecHits", edm::InputTag(""));
      desc.add<edm::ESInputTag>("pfClusterParams", edm::ESInputTag(""));
      desc.add<edm::ESInputTag>("topology", edm::ESInputTag(""));
      desc.add<bool>("synchronise", false);
      desc.add<int>("pfRecHitFractionAllocation", 120);
      descriptions.addWithDefaultLabel(desc);
    }

produce()

void ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::produce ( device::Event &  event, device::EventSetup const &  setup  )

inlineoverridevirtual

Implements ALPAKA_ACCELERATOR_NAMESPACE::stream::EDProducer< Args >.

Definition at line 36 of file PFClusterSoAProducer.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::PFClusterProducerKernel::execute(), inputPFRecHitSoA_Token_, eostools::move(), outputPFClusterSoA_Token_, outputPFRHFractionSoA_Token_, submitPVValidationJobs::params, pfClusParamsToken, pfRecHitFractionAllocation_, HLT_2024v12_cff::pfRecHits, singleTopDQM_cfi::setup, synchronise_, HLT_2024v12_cff::topology, topologyToken_, and SequenceTypes::wait().

                                                                             {
      const reco::PFClusterParamsDeviceCollection& params = setup.getData(pfClusParamsToken);
      const reco::PFRecHitHCALTopologyDeviceCollection& topology = setup.getData(topologyToken_);
      const reco::PFRecHitHostCollection& pfRecHits = event.get(inputPFRecHitSoA_Token_);
      int nRH = 0;
      if (pfRecHits->metadata().size() != 0)
        nRH = pfRecHits->size();

      reco::PFClusteringVarsDeviceCollection pfClusteringVars{nRH, event.queue()};
      reco::PFClusteringEdgeVarsDeviceCollection pfClusteringEdgeVars{(nRH * 8), event.queue()};
      reco::PFClusterDeviceCollection pfClusters{nRH, event.queue()};
      reco::PFRecHitFractionDeviceCollection pfrhFractions{nRH * pfRecHitFractionAllocation_, event.queue()};

      if (nRH != 0) {
        PFClusterProducerKernel kernel(event.queue(), pfRecHits);
        kernel.execute(event.queue(),
                       params,
                       topology,
                       pfClusteringVars,
                       pfClusteringEdgeVars,
                       pfRecHits,
                       pfClusters,
                       pfrhFractions);
      }

      if (synchronise_)
        alpaka::wait(event.queue());

      event.emplace(outputPFClusterSoA_Token_, std::move(pfClusters));
      event.emplace(outputPFRHFractionSoA_Token_, std::move(pfrhFractions));
    }

Member Data Documentation

inputPFRecHitSoA_Token_

const edm::EDGetTokenT<reco::PFRecHitHostCollection> ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::inputPFRecHitSoA_Token_

private

Definition at line 81 of file PFClusterSoAProducer.cc.

Referenced by produce().

outputPFClusterSoA_Token_

const device::EDPutToken<reco::PFClusterDeviceCollection> ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::outputPFClusterSoA_Token_

private

Definition at line 82 of file PFClusterSoAProducer.cc.

Referenced by produce().

outputPFRHFractionSoA_Token_

const device::EDPutToken<reco::PFRecHitFractionDeviceCollection> ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::outputPFRHFractionSoA_Token_

private

Definition at line 83 of file PFClusterSoAProducer.cc.

Referenced by produce().

pfClusParamsToken

const device::ESGetToken<reco::PFClusterParamsDeviceCollection, JobConfigurationGPURecord> ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::pfClusParamsToken

private

Definition at line 79 of file PFClusterSoAProducer.cc.

Referenced by produce().

pfRecHitFractionAllocation_

const int ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::pfRecHitFractionAllocation_

private

Definition at line 85 of file PFClusterSoAProducer.cc.

Referenced by produce().

synchronise_

const bool ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::synchronise_

private

Definition at line 84 of file PFClusterSoAProducer.cc.

Referenced by produce().

topologyToken_

const device::ESGetToken<reco::PFRecHitHCALTopologyDeviceCollection, PFRecHitHCALTopologyRecord> ALPAKA_ACCELERATOR_NAMESPACE::PFClusterSoAProducer::topologyToken_

private

Definition at line 80 of file PFClusterSoAProducer.cc.

Referenced by produce().