ALPAKA_ACCELERATOR_NAMESPACE::PrepareTopoInputs 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::PFClusteringEdgeVarsDeviceCollection::View pfClusteringEdgeVars, uint32_t *__restrict__ nSeeds) const

Detailed Description

Definition at line 1165 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::PrepareTopoInputs::operator() ( const TAcc &  acc, const reco::PFRecHitHostCollection::ConstView  pfRecHits, reco::PFClusteringVarsDeviceCollection::View  pfClusteringVars, reco::PFClusteringEdgeVarsDeviceCollection::View  pfClusteringEdgeVars, uint32_t *__restrict__  nSeeds  ) const

inline

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

References cms::alpakatools::elements_with_stride(), mps_fire::i, dqmiolumiharvest::j, cms::alpakatools::once_per_grid(), and HLT_2024v10_cff::pfRecHits.

                                                                       {
      const int nRH = pfRecHits.size();

      if (once_per_grid(acc)) {
        pfClusteringVars.nEdges() = nRH * 8;
        pfClusteringEdgeVars[nRH].pfrh_edgeIdx() = nRH * 8;
      }
      for (uint32_t i : cms::alpakatools::elements_with_stride(acc, nRH)) {
        pfClusteringEdgeVars[i].pfrh_edgeIdx() = i * 8;
        pfClusteringVars[i].pfrh_topoId() = 0;
        for (int j = 0; j < 8; j++) {  // checking if neighbours exist and assigning neighbours as edges
          if (pfRecHits[i].neighbours()(j) == -1)
            pfClusteringEdgeVars[i * 8 + j].pfrh_edgeList() = i;
          else
            pfClusteringEdgeVars[i * 8 + j].pfrh_edgeList() = pfRecHits[i].neighbours()(j);
        }
      }

      return;
    }