Skip to content

Commit 5e7b948

Browse files
committed
Make BeamSpotDeviceProducer to rely on implicit host-to-device copy
1 parent 24690bd commit 5e7b948

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

RecoVertex/BeamSpotProducer/plugins/alpaka/BeamSpotDeviceProducer.cc

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
1717
BeamSpotDeviceProducer(edm::ParameterSet const& config)
1818
: EDProducer(config),
1919
legacyToken_{consumes(config.getParameter<edm::InputTag>("src"))},
20-
deviceToken_{produces()} {}
20+
podToken_{produces()} {}
2121

2222
void produce(edm::StreamID, device::Event& event, device::EventSetup const& setup) const override {
2323
reco::BeamSpot const& beamspot = event.get(legacyToken_);
@@ -35,13 +35,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
3535
hostProduct->emittanceY = beamspot.emittanceY();
3636
hostProduct->betaStar = beamspot.betaStar();
3737

38-
if constexpr (std::is_same_v<Device, alpaka::DevCpu>) {
39-
event.emplace(deviceToken_, std::move(hostProduct));
40-
} else {
41-
BeamSpotDevice deviceProduct{event.queue()};
42-
alpaka::memcpy(event.queue(), deviceProduct.buffer(), hostProduct.const_buffer());
43-
event.emplace(deviceToken_, std::move(deviceProduct));
44-
}
38+
event.emplace(podToken_, std::move(hostProduct));
4539
}
4640

4741
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
@@ -52,7 +46,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
5246

5347
private:
5448
const edm::EDGetTokenT<reco::BeamSpot> legacyToken_;
55-
const device::EDPutToken<BeamSpotDevice> deviceToken_;
49+
const edm::EDPutTokenT<BeamSpotHost> podToken_;
5650
};
5751

5852
} // namespace ALPAKA_ACCELERATOR_NAMESPACE

0 commit comments

Comments
 (0)