test/alpaka/TrackSoAHeterogeneous_test.dev.cc

0001 #include <type_traits>
0002
0003 #include <alpaka/alpaka.hpp>
0004
0005 #include "DataFormats/TrackSoA/interface/TrackDefinitions.h"
0006 #include "DataFormats/TrackSoA/interface/TracksDevice.h"
0007 #include "DataFormats/TrackSoA/interface/TracksHost.h"
0008 #include "DataFormats/TrackSoA/interface/alpaka/TracksSoACollection.h"
0009 #include "Geometry/CommonTopologies/interface/SimplePixelTopology.h"
0010 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
0011
0012 #include "TrackSoAHeterogeneous_test.h"
0013
0014 using namespace reco;
0015
0016 using Quality = pixelTrack::Quality;
0017 namespace ALPAKA_ACCELERATOR_NAMESPACE {
0018   using namespace cms::alpakatools;
0019   namespace testTrackSoA {
0020
0021     // Kernel which fills the TrackSoAView with data
0022     // to test writing to it
0023     class TestFillKernel {
0024     public:
0025       ALPAKA_FN_ACC void operator()(Acc1D const& acc, TrackSoAView tracks_view, int32_t nTracks) const {
0026         if (cms::alpakatools::once_per_grid(acc)) {
0027           tracks_view.nTracks() = nTracks;
0028         }
0029
0030         for (int32_t j : uniform_elements(acc, nTracks)) {
0031           tracks_view[j].pt() = (float)j;
0032           tracks_view[j].eta() = (float)j;
0033           tracks_view[j].chi2() = (float)j;
0034           tracks_view[j].quality() = (Quality)(j % 256);
0035           tracks_view[j].nLayers() = j % 128;
0036           tracks_view[j].hitOffsets() = j;
0037         }
0038       }
0039     };
0040
0041     // Kernel which reads from the TrackSoAView to verify
0042     // that it was written correctly from the fill kernel
0043     class TestVerifyKernel {
0044     public:
0045       ALPAKA_FN_ACC void operator()(Acc1D const& acc, TrackSoAConstView tracks_view, int32_t nTracks) const {
0046         if (cms::alpakatools::once_per_grid(acc)) {
0047           ALPAKA_ASSERT(tracks_view.nTracks() == nTracks);
0048         }
0049         for (int32_t j : uniform_elements(acc, tracks_view.nTracks())) {
0050           ALPAKA_ASSERT(abs(tracks_view[j].pt() - (float)j) < .0001);
0051           ALPAKA_ASSERT(abs(tracks_view[j].eta() - (float)j) < .0001);
0052           ALPAKA_ASSERT(abs(tracks_view[j].chi2() - (float)j) < .0001);
0053           ALPAKA_ASSERT(tracks_view[j].quality() == (Quality)(j % 256));
0054           ALPAKA_ASSERT(tracks_view[j].nLayers() == j % 128);
0055           ALPAKA_ASSERT(tracks_view[j].hitOffsets() == uint32_t(j));
0056         }
0057       }
0058     };
0059
0060     // Host function which invokes the two kernels above
0061     void runKernels(TrackSoAView tracks_view, Queue& queue) {
0062       int32_t tracks = 420;
0063       uint32_t items = 64;
0064       uint32_t groups = divide_up_by(tracks, items);
0065       auto workDiv = make_workdiv<Acc1D>(groups, items);
0066       alpaka::exec<Acc1D>(queue, workDiv, TestFillKernel{}, tracks_view, tracks);
0067       alpaka::exec<Acc1D>(queue, workDiv, TestVerifyKernel{}, tracks_view, tracks);
0068     }
0069
0070   }  // namespace testTrackSoA
0071
0072 }  // namespace ALPAKA_ACCELERATOR_NAMESPACE