ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer Class Reference

Inheritance diagram for ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer:

Public Member Functions
void	produce (device::Event &iEvent, device::EventSetup const &iSetup) override
	TestAlpakaStreamProducer (edm::ParameterSet const &config)
Public Member Functions inherited from ALPAKA_ACCELERATOR_NAMESPACE::stream::EDProducer<>
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
TestAlgo	algo_
device::EDPutToken< portabletest::TestDeviceCollection >	devicePutToken_
device::EDPutToken< portabletest::TestDeviceMultiCollection2 >	devicePutTokenMulti2_
device::EDPutToken< portabletest::TestDeviceMultiCollection3 >	devicePutTokenMulti3_
device::ESGetToken< cms::alpakatest::AlpakaESTestDataB< Device >, AlpakaESTestRecordB >	esToken_
edm::EDGetTokenT< edmtest::IntProduct >	getToken_
const int32_t	size2_
const int32_t	size3_
const int32_t	size_

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

This class demonstrates a stream EDProducer that

• consumes a host EDProduct
• consumes a device ESProduct
• produces a device EDProduct (that gets transferred to host automatically if needed)
• optionally uses a product instance label

Definition at line 24 of file TestAlpakaStreamProducer.cc.

Constructor & Destructor Documentation

TestAlpakaStreamProducer()

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

inline

Definition at line 26 of file TestAlpakaStreamProducer.cc.

References EDM_STRINGIZE.

        : 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))} {
      getToken_ = consumes(config.getParameter<edm::InputTag>("source"));
      esToken_ = esConsumes(config.getParameter<edm::ESInputTag>("eventSetupSource"));
      devicePutToken_ = produces(config.getParameter<std::string>("productInstanceName"));
      devicePutTokenMulti2_ = produces(config.getParameter<std::string>("productInstanceName"));
      devicePutTokenMulti3_ = produces(config.getParameter<std::string>("productInstanceName"));
    }

Member Function Documentation

fillDescriptions()

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

inlinestatic

Definition at line 60 of file TestAlpakaStreamProducer.cc.

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

                                                                             {
      edm::ParameterSetDescription desc;
      desc.add<edm::InputTag>("source");
      desc.add("eventSetupSource", edm::ESInputTag{});
      desc.add<std::string>("productInstanceName", "");

      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::TestAlpakaStreamProducer::produce ( device::Event &  iEvent, device::EventSetup const &  iSetup  )

inlineoverridevirtual

Implements ALPAKA_ACCELERATOR_NAMESPACE::stream::EDProducer<>.

Definition at line 40 of file TestAlpakaStreamProducer.cc.

References algo_, devicePutToken_, devicePutTokenMulti2_, devicePutTokenMulti3_, esToken_, ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fill(), ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fillMulti2(), ALPAKA_ACCELERATOR_NAMESPACE::TestAlgo::fillMulti3(), ALPAKA_ACCELERATOR_NAMESPACE::device::EventSetup::getData(), getToken_, iEvent, eostools::move(), size2_, size3_, and size_.

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

      auto deviceProduct = std::make_unique<portabletest::TestDeviceCollection>(size_, iEvent.queue());
      auto deviceProductMulti2 = std::make_unique<portabletest::TestDeviceMultiCollection2>(
          portabletest::TestDeviceMultiCollection2::SizesArray{{size_, size2_}}, iEvent.queue());
      auto deviceProductMulti3 = std::make_unique<portabletest::TestDeviceMultiCollection3>(
          portabletest::TestDeviceMultiCollection3::SizesArray{{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.put(devicePutToken_, std::move(deviceProduct));
      iEvent.put(devicePutTokenMulti2_, std::move(deviceProductMulti2));
      iEvent.put(devicePutTokenMulti3_, std::move(deviceProductMulti3));
    }

Member Data Documentation

algo_

TestAlgo ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::algo_

private

Definition at line 86 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

devicePutToken_

device::EDPutToken<portabletest::TestDeviceCollection> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::devicePutToken_

private

Definition at line 78 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

devicePutTokenMulti2_

device::EDPutToken<portabletest::TestDeviceMultiCollection2> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::devicePutTokenMulti2_

private

Definition at line 79 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

devicePutTokenMulti3_

device::EDPutToken<portabletest::TestDeviceMultiCollection3> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::devicePutTokenMulti3_

private

Definition at line 80 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

esToken_

device::ESGetToken<cms::alpakatest::AlpakaESTestDataB<Device>, AlpakaESTestRecordB> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::esToken_

private

Definition at line 77 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

getToken_

edm::EDGetTokenT<edmtest::IntProduct> ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::getToken_

private

Definition at line 76 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

size2_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::size2_

private

Definition at line 82 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

size3_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::size3_

private

Definition at line 83 of file TestAlpakaStreamProducer.cc.

Referenced by produce().

size_

const int32_t ALPAKA_ACCELERATOR_NAMESPACE::TestAlpakaStreamProducer::size_

private

Definition at line 81 of file TestAlpakaStreamProducer.cc.

Referenced by produce().