File indexing completed on 2025-02-05 03:15:03
0001 #ifndef HeterogeneousCore_AlpakaCore_interface_alpaka_stream_EDProducer_h
0002 #define HeterogeneousCore_AlpakaCore_interface_alpaka_stream_EDProducer_h
0003
0004 #include "FWCore/Framework/interface/stream/EDProducer.h"
0005 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/EDMetadataSentry.h"
0006 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/Event.h"
0007 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/EventSetup.h"
0008 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/ProducerBase.h"
0009 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
0010
0011 namespace ALPAKA_ACCELERATOR_NAMESPACE {
0012 namespace stream {
0013 template <typename... Args>
0014 class EDProducer : public ProducerBase<edm::stream::EDProducer, Args...> {
0015 static_assert(not edm::CheckAbility<edm::module::Abilities::kExternalWork, Args...>::kHasIt,
0016 "ALPAKA_ACCELERATOR_NAMESPACE::stream::EDProducer may not be used with ExternalWork ability. "
0017 "Please use ALPAKA_ACCELERATOR_NAMESPACE::stream::SynchronizingEDProducer instead.");
0018 using Base = ProducerBase<edm::stream::EDProducer, Args...>;
0019
0020 protected:
0021 EDProducer(edm::ParameterSet const iConfig) : Base(iConfig) {}
0022
0023 public:
0024 void produce(edm::Event& iEvent, edm::EventSetup const& iSetup) final {
0025 detail::EDMetadataSentry sentry(iEvent.streamID(), this->synchronize());
0026 device::Event ev(iEvent, sentry.metadata());
0027 device::EventSetup const es(iSetup, ev.device());
0028 produce(ev, es);
0029 this->putBackend(iEvent);
0030 sentry.finish(ev.wasQueueUsed());
0031 }
0032
0033 virtual void produce(device::Event& iEvent, device::EventSetup const& iSetup) = 0;
0034 };
0035 }
0036 }
0037
0038 #endif