test/alpaka/Hits_test.dev.cc

0001 #include <type_traits>
0002
0003 #include <alpaka/alpaka.hpp>
0004
0005 #include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsDevice.h"
0006 #include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsSoA.h"
0007 #include "DataFormats/TrackingRecHitSoA/interface/alpaka/TrackingRecHitsSoACollection.h"
0008 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
0009 #include "HeterogeneousCore/AlpakaInterface/interface/workdivision.h"
0010
0011 #include "Hits_test.h"
0012
0013 using namespace alpaka;
0014
0015 namespace ALPAKA_ACCELERATOR_NAMESPACE {
0016
0017   using namespace cms::alpakatools;
0018   using namespace ALPAKA_ACCELERATOR_NAMESPACE::reco;
0019   namespace testTrackingRecHitSoA {
0020
0021     struct TestFillKernel {
0022       ALPAKA_FN_ACC void operator()(Acc1D const& acc,
0023                                     ::reco::TrackingRecHitView soa,
0024                                     ::reco::HitModuleSoAView mods) const {
0025         const uint32_t i(alpaka::getIdx<alpaka::Grid, alpaka::Blocks>(acc)[0u]);
0026         const uint32_t j(alpaka::getIdx<alpaka::Block, alpaka::Threads>(acc)[0u]);
0027
0028         if (cms::alpakatools::once_per_grid(acc)) {
0029           soa.offsetBPIX2() = 22;
0030           soa[10].xLocal() = 1.11;
0031         }
0032
0033         soa[i].iphi() = i % 10;
0034         mods[j].moduleStart() = j;
0035       }
0036     };
0037
0038     struct ShowKernel {
0039       ALPAKA_FN_ACC void operator()(Acc1D const& acc,
0040                                     ::reco::TrackingRecHitConstView soa,
0041                                     ::reco::HitModuleSoAView mods) const {
0042         if (cms::alpakatools::once_per_grid(acc)) {
0043           printf("offsetBPIX = %d\n", soa.offsetBPIX2());
0044           printf("nHits = %d\n", soa.metadata().size());
0045           printf("hitsModuleStart[28] = %d\n", mods[28].moduleStart());
0046         }
0047
0048         // can be increased to soa.nHits() for debugging
0049         for (uint32_t i : cms::alpakatools::uniform_elements(acc, soa.metadata().size())) {
0050           printf("iPhi %d -> %d\n", i, soa[i].iphi());
0051           printf("x %d -> %.2f \n", i, soa[i].xLocal());
0052         }
0053       }
0054     };
0055
0056     void runKernels(::reco::TrackingRecHitView& view, ::reco::HitModuleSoAView& mods, Queue& queue) {
0057       uint32_t items = 64;
0058       uint32_t groups = divide_up_by(view.metadata().size(), items);
0059       auto workDiv = make_workdiv<Acc1D>(groups, items);
0060       alpaka::exec<Acc1D>(queue, workDiv, TestFillKernel{}, view, mods);
0061       alpaka::exec<Acc1D>(queue, workDiv, ShowKernel{}, view, mods);
0062     }
0063
0064   }  // namespace testTrackingRecHitSoA
0065 }  // namespace ALPAKA_ACCELERATOR_NAMESPACE