|
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
79
|
#include <type_traits>
#include <alpaka/alpaka.hpp>
#include "DataFormats/TrackSoA/interface/TrackDefinitions.h"
#include "DataFormats/TrackSoA/interface/TracksDevice.h"
#include "DataFormats/TrackSoA/interface/TracksHost.h"
#include "DataFormats/TrackSoA/interface/alpaka/TracksSoACollection.h"
#include "Geometry/CommonTopologies/interface/SimplePixelTopology.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
#include "TrackSoAHeterogeneous_test.h"
using namespace reco;
using Quality = pixelTrack::Quality;
namespace ALPAKA_ACCELERATOR_NAMESPACE {
using namespace cms::alpakatools;
namespace testTrackSoA {
// Kernel which fills the TrackSoAView with data
// to test writing to it
template <typename TrackerTraits>
class TestFillKernel {
public:
ALPAKA_FN_ACC void operator()(Acc1D const& acc, TrackSoAView<TrackerTraits> tracks_view, int32_t nTracks) const {
if (cms::alpakatools::once_per_grid(acc)) {
tracks_view.nTracks() = nTracks;
}
for (int32_t j : uniform_elements(acc, nTracks)) {
tracks_view[j].pt() = (float)j;
tracks_view[j].eta() = (float)j;
tracks_view[j].chi2() = (float)j;
tracks_view[j].quality() = (Quality)(j % 256);
tracks_view[j].nLayers() = j % 128;
tracks_view.hitIndices().off[j] = j;
}
}
};
// Kernel which reads from the TrackSoAView to verify
// that it was written correctly from the fill kernel
template <typename TrackerTraits>
class TestVerifyKernel {
public:
ALPAKA_FN_ACC void operator()(Acc1D const& acc,
TrackSoAConstView<TrackerTraits> tracks_view,
int32_t nTracks) const {
if (cms::alpakatools::once_per_grid(acc)) {
ALPAKA_ASSERT(tracks_view.nTracks() == nTracks);
}
for (int32_t j : uniform_elements(acc, tracks_view.nTracks())) {
ALPAKA_ASSERT(abs(tracks_view[j].pt() - (float)j) < .0001);
ALPAKA_ASSERT(abs(tracks_view[j].eta() - (float)j) < .0001);
ALPAKA_ASSERT(abs(tracks_view[j].chi2() - (float)j) < .0001);
ALPAKA_ASSERT(tracks_view[j].quality() == (Quality)(j % 256));
ALPAKA_ASSERT(tracks_view[j].nLayers() == j % 128);
ALPAKA_ASSERT(tracks_view.hitIndices().off[j] == uint32_t(j));
}
}
};
// Host function which invokes the two kernels above
template <typename TrackerTraits>
void runKernels(TrackSoAView<TrackerTraits> tracks_view, Queue& queue) {
int32_t tracks = 420;
uint32_t items = 64;
uint32_t groups = divide_up_by(tracks, items);
auto workDiv = make_workdiv<Acc1D>(groups, items);
alpaka::exec<Acc1D>(queue, workDiv, TestFillKernel<TrackerTraits>{}, tracks_view, tracks);
alpaka::exec<Acc1D>(queue, workDiv, TestVerifyKernel<TrackerTraits>{}, tracks_view, tracks);
}
template void runKernels<pixelTopology::Phase1>(TrackSoAView<pixelTopology::Phase1> tracks_view, Queue& queue);
template void runKernels<pixelTopology::Phase2>(TrackSoAView<pixelTopology::Phase2> tracks_view, Queue& queue);
} // namespace testTrackSoA
} // namespace ALPAKA_ACCELERATOR_NAMESPACE
|