ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer Class Reference

Inheritance diagram for ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer:

Public Member Functions
void	produce (edm::StreamID, device::Event &iEvent, device::EventSetup const &iSetup) const override
	TestAlpakaGlobalProducer (edm::ParameterSet const &config)
Public Member Functions inherited from ALPAKA_ACCELERATOR_NAMESPACE::global::EDProducer<>
void	produce (edm::StreamID sid, edm::Event &iEvent, edm::EventSetup const &iSetup) const final
Public Member Functions inherited from edm::ProducerBase< edm::global::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
TestAlgo	algo_
const device::EDPutToken< portabletest::TestDeviceCollection >	deviceToken_
const device::EDPutToken< portabletest::TestDeviceMultiCollection2 >	deviceTokenMulti2_
const device::EDPutToken< portabletest::TestDeviceMultiCollection3 >	deviceTokenMulti3_
const device::ESGetToken< AlpakaESTestDataADevice, AlpakaESTestRecordA >	esToken_
const int32_t	size2_
const int32_t	size3_
const int32_t	size_

Additional Inherited Members
Public Types inherited from edm::ProducerBase< edm::global::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::global::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

This class demonstrates a global EDProducer that

• consumes a device ESProduct
• produces a device EDProduct (that can get transferred to host automatically)

Definition at line 21 of file TestAlpakaGlobalProducer.cc.

Constructor & Destructor Documentation

TestAlpakaGlobalProducer()

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

inline

Definition at line 23 of file TestAlpakaGlobalProducer.cc.

References edm::ProducerBase< edm::global::EDProducer, Args... >::produces().

        : esToken_(esConsumes(config.getParameter<edm::ESInputTag>("eventSetupSource"))),
          deviceToken_{produces()},
          deviceTokenMulti2_{produces()},
          deviceTokenMulti3_{produces()},
          size_{config.getParameter<edm::ParameterSet>("size").getParameter<int32_t>(
              EDM_STRINGIZE(ALPAKA_ACCELERATOR_NAMESPACE))},
          size2_{config.getParameter<edm::ParameterSet>("size").getParameter<int32_t>(
              EDM_STRINGIZE(ALPAKA_ACCELERATOR_NAMESPACE))},
          size3_{config.getParameter<edm::ParameterSet>("size").getParameter<int32_t>(
              EDM_STRINGIZE(ALPAKA_ACCELERATOR_NAMESPACE))} {}

Member Function Documentation

fillDescriptions()

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

inlinestatic

Definition at line 52 of file TestAlpakaGlobalProducer.cc.

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

                                                                             {
      edm::ParameterSetDescription desc;
      desc.add("eventSetupSource", edm::ESInputTag{});

      edm::ParameterSetDescription psetSize;
      psetSize.add<int32_t>("alpaka_serial_sync");
      psetSize.add<int32_t>("alpaka_cuda_async");
      psetSize.add<int32_t>("alpaka_rocm_async");
      desc.add("size", psetSize);

      descriptions.addWithDefaultLabel(desc);
    }

produce()

void ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::produce ( edm::StreamID  , device::Event &  iEvent, device::EventSetup const &  iSetup  ) const

inlineoverridevirtual

Implements ALPAKA_ACCELERATOR_NAMESPACE::global::EDProducer<>.

Definition at line 35 of file TestAlpakaGlobalProducer.cc.

References algo_, deviceToken_, deviceTokenMulti2_, deviceTokenMulti3_, esToken_, ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fill(), ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fillMulti2(), ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fillMulti3(), ALPAKA_ACCELERATOR_NAMESPACE::device::EventSetup::getData(), iEvent, eostools::move(), size2_, size3_, and size_.

                                                                                                  {
      [[maybe_unused]] auto const& esData = iSetup.getData(esToken_);

      portabletest::TestDeviceCollection deviceProduct{size_, iEvent.queue()};
      portabletest::TestDeviceMultiCollection2 deviceProductMulti2{{{size_, size2_}}, iEvent.queue()};
      portabletest::TestDeviceMultiCollection3 deviceProductMulti3{{{size_, size2_, size3_}}, iEvent.queue()};

      // run the algorithm, potentially asynchronously
      algo_.fill(iEvent.queue(), deviceProduct);
      algo_.fillMulti2(iEvent.queue(), deviceProductMulti2);
      algo_.fillMulti3(iEvent.queue(), deviceProductMulti3);

      iEvent.emplace(deviceToken_, std::move(deviceProduct));
      iEvent.emplace(deviceTokenMulti2_, std::move(deviceProductMulti2));
      iEvent.emplace(deviceTokenMulti3_, std::move(deviceProductMulti3));
    }

Member Data Documentation

algo_

TestAlgo ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::algo_

private

Definition at line 75 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

deviceToken_

const device::EDPutToken<portabletest::TestDeviceCollection> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::deviceToken_

private

Definition at line 67 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

deviceTokenMulti2_

const device::EDPutToken<portabletest::TestDeviceMultiCollection2> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::deviceTokenMulti2_

private

Definition at line 68 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

deviceTokenMulti3_

const device::EDPutToken<portabletest::TestDeviceMultiCollection3> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::deviceTokenMulti3_

private

Definition at line 69 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

esToken_

const device::ESGetToken<AlpakaESTestDataADevice, AlpakaESTestRecordA> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::esToken_

private

Definition at line 66 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

size2_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::size2_

private

Definition at line 71 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

size3_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::size3_

private

Definition at line 72 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().

size_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaGlobalProducer::size_

private

Definition at line 70 of file TestAlpakaGlobalProducer.cc.

Referenced by produce().