|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#ifndef HeterogeneousCore_AlpakaCore_interface_alpaka_stream_SynchronizingEDProducer_h
#define HeterogeneousCore_AlpakaCore_interface_alpaka_stream_SynchronizingEDProducer_h
#include "FWCore/Concurrency/interface/WaitingTaskWithArenaHolder.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/EDMetadataAcquireSentry.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/EDMetadataSentry.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/Event.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/EventSetup.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/ProducerBase.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
namespace ALPAKA_ACCELERATOR_NAMESPACE {
namespace stream {
template <typename... Args>
class SynchronizingEDProducer : public ProducerBase<edm::stream::EDProducer, edm::ExternalWork, Args...> {
static_assert(
not edm::CheckAbility<edm::module::Abilities::kExternalWork, Args...>::kHasIt,
"ExternalWork ability is redundant with ALPAKA_ACCELERATOR_NAMESPACE::stream::SynchronizingEDProducer."
"Please remove it.");
using Base = ProducerBase<edm::stream::EDProducer, edm::ExternalWork, Args...>;
protected:
SynchronizingEDProducer(edm::ParameterSet const iConfig) : Base(iConfig) {}
public:
void acquire(edm::Event const& iEvent,
edm::EventSetup const& iSetup,
edm::WaitingTaskWithArenaHolder holder) final {
detail::EDMetadataAcquireSentry sentry(iEvent.streamID(), std::move(holder), this->synchronize());
device::Event const ev(iEvent, sentry.metadata());
device::EventSetup const es(iSetup, ev.device());
acquire(ev, es);
metadata_ = sentry.finish();
}
void produce(edm::Event& iEvent, edm::EventSetup const& iSetup) final {
detail::EDMetadataSentry sentry(std::move(metadata_), this->synchronize());
device::Event ev(iEvent, sentry.metadata());
device::EventSetup const es(iSetup, ev.device());
produce(ev, es);
this->putBackend(iEvent);
sentry.finish(ev.wasQueueUsed());
}
virtual void acquire(device::Event const& iEvent, device::EventSetup const& iSetup) = 0;
virtual void produce(device::Event& iEvent, device::EventSetup const& iSetup) = 0;
private:
std::shared_ptr<EDMetadata> metadata_;
};
} // namespace stream
} // namespace ALPAKA_ACCELERATOR_NAMESPACE
#endif
|