ALPAKA_ACCELERATOR_NAMESPACE::FillRhfIndex Class Reference

Public Member Functions
template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>
ALPAKA_FN_ACC void	operator() (const TAcc &acc, const reco::PFRecHitHostCollection::ConstView pfRecHits, reco::PFClusteringVarsDeviceCollection::View pfClusteringVars, reco::PFRecHitFractionDeviceCollection::View fracView) const

Detailed Description

Definition at line 1317 of file PFClusterSoAProducerKernel.dev.cc.

Member Function Documentation

operator()()

template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>

ALPAKA_FN_ACC void ALPAKA_ACCELERATOR_NAMESPACE::FillRhfIndex::operator() ( const TAcc &  acc, const reco::PFRecHitHostCollection::ConstView  pfRecHits, reco::PFClusteringVarsDeviceCollection::View  pfClusteringVars, reco::PFRecHitFractionDeviceCollection::View  fracView  ) const

inline

Definition at line 1320 of file PFClusterSoAProducerKernel.dev.cc.

References cms::cudacompat::atomicAdd(), HLT_2024v13_cff::fraction, mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, HLT_2024v13_cff::pfRecHits, fileCollector::seed, and cms::alpakatools::uniform_elements_nd().

                                                                                           {
      const int nRH = pfRecHits.size();

      for (auto index : uniform_elements_nd(acc, {nRH, nRH})) {
        const int i = index[0u];  // i is a seed index
        const int j = index[1u];  // j is NOT a seed
        int topoId = pfClusteringVars[i].pfrh_topoId();
        if (topoId > -1 && pfClusteringVars[i].pfrh_isSeed() && topoId == pfClusteringVars[j].pfrh_topoId()) {
          if (!pfClusteringVars[j].pfrh_isSeed()) {  // NOT a seed
            int k = alpaka::atomicAdd(
                acc, &pfClusteringVars[i].rhCount(), 1);  // Increment the number of rechit fractions for this seed
            auto fraction = fracView[pfClusteringVars[i].seedFracOffsets() + k];
            fraction.pfrhIdx() = j;
            fraction.pfcIdx() = pfClusteringVars[i].rhIdxToSeedIdx();
          } else if (i == j) {  // i==j is a seed rechit index
            auto seed = fracView[pfClusteringVars[i].seedFracOffsets()];
            seed.pfrhIdx() = j;
            seed.frac() = 1;
            seed.pfcIdx() = pfClusteringVars[i].rhIdxToSeedIdx();
          }
        }
      }
    }