|
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#ifndef DataFormats_TrackingRecHitSoA_interface_alpaka_TrackingRecHitsSoACollection_h
#define DataFormats_TrackingRecHitSoA_interface_alpaka_TrackingRecHitsSoACollection_h
#include <cstdint>
#include <type_traits>
#include <alpaka/alpaka.hpp>
#include "DataFormats/Portable/interface/alpaka/PortableCollection.h"
#include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsDevice.h"
#include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsHost.h"
#include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsSoA.h"
#include "Geometry/CommonTopologies/interface/SimplePixelTopology.h"
#include "HeterogeneousCore/AlpakaInterface/interface/CopyToHost.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
namespace ALPAKA_ACCELERATOR_NAMESPACE {
template <typename TrackerTraits>
using TrackingRecHitsSoACollection = std::conditional_t<std::is_same_v<Device, alpaka::DevCpu>,
TrackingRecHitHost<TrackerTraits>,
TrackingRecHitDevice<TrackerTraits, Device>>;
// Classes definition for Phase1/Phase2, to make the classes_def lighter. Not actually used in the code.
using TrackingRecHitSoAPhase1 = TrackingRecHitsSoACollection<pixelTopology::Phase1>;
using TrackingRecHitSoAPhase2 = TrackingRecHitsSoACollection<pixelTopology::Phase2>;
using TrackingRecHitSoAHIonPhase1 = TrackingRecHitsSoACollection<pixelTopology::HIonPhase1>;
} // namespace ALPAKA_ACCELERATOR_NAMESPACE
namespace cms::alpakatools {
template <typename TrackerTraits, typename TDevice>
struct CopyToHost<TrackingRecHitDevice<TrackerTraits, TDevice>> {
template <typename TQueue>
static auto copyAsync(TQueue& queue, TrackingRecHitDevice<TrackerTraits, TDevice> const& deviceData) {
TrackingRecHitHost<TrackerTraits> hostData(queue, deviceData.view().metadata().size());
// Don't bother if zero hits
if (deviceData.view().metadata().size() == 0) {
std::memset(hostData.buffer().data(),
0,
alpaka::getExtentProduct(hostData.buffer()) *
sizeof(alpaka::Elem<typename TrackingRecHitHost<TrackerTraits>::Buffer>));
return hostData;
}
alpaka::memcpy(queue, hostData.buffer(), deviceData.buffer());
#ifdef GPU_DEBUG
printf("TrackingRecHitsSoACollection: I'm copying to host.\n");
alpaka::wait(queue);
assert(deviceData.nHits() == hostData.nHits());
assert(deviceData.offsetBPIX2() == hostData.offsetBPIX2());
#endif
return hostData;
}
// Update the contents address of the phiBinner histo container after the copy from device happened
static void postCopy(TrackingRecHitHost<TrackerTraits>& hostData) {
// Don't bother if zero hits
if (hostData.view().metadata().size() == 0) {
return;
}
typename TrackingRecHitSoA<TrackerTraits>::PhiBinnerView pbv;
pbv.assoc = &(hostData.view().phiBinner());
pbv.offSize = -1;
pbv.offStorage = nullptr;
pbv.contentSize = hostData.nHits();
pbv.contentStorage = hostData.view().phiBinnerStorage();
hostData.view().phiBinner().initStorage(pbv);
}
};
} // namespace cms::alpakatools
ASSERT_DEVICE_MATCHES_HOST_COLLECTION(TrackingRecHitSoAPhase1, TrackingRecHitHostPhase1);
ASSERT_DEVICE_MATCHES_HOST_COLLECTION(TrackingRecHitSoAPhase2, TrackingRecHitHostPhase2);
ASSERT_DEVICE_MATCHES_HOST_COLLECTION(TrackingRecHitSoAHIonPhase1, TrackingRecHitHostHIonPhase1);
#endif // DataFormats_TrackingRecHitSoA_interface_alpaka_TrackingRecHitsSoACollection_h
|