ALPAKA_ACCELERATOR_NAMESPACE::FastCluster Class Reference

Public Member Functions
template<bool debug = false, typename TAcc , typename = std::enable_if<!std::is_same_v<Device, alpaka::DevCpu>>>
ALPAKA_FN_ACC void	operator() (const TAcc &acc, const reco::PFRecHitHostCollection::ConstView pfRecHits, const reco::PFClusterParamsDeviceCollection::ConstView pfClusParams, const reco::PFRecHitHCALTopologyDeviceCollection::ConstView topology, reco::PFClusteringVarsDeviceCollection::View pfClusteringVars, reco::PFClusterDeviceCollection::View clusterView, reco::PFRecHitFractionDeviceCollection::View fracView) const

Detailed Description

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

Member Function Documentation

operator()()

template<bool debug = false, typename TAcc , typename = std::enable_if<!std::is_same_v<Device, alpaka::DevCpu>>>

ALPAKA_FN_ACC void ALPAKA_ACCELERATOR_NAMESPACE::FastCluster::operator() ( const TAcc &  acc, const reco::PFRecHitHostCollection::ConstView  pfRecHits, const reco::PFClusterParamsDeviceCollection::ConstView  pfClusParams, const reco::PFRecHitHCALTopologyDeviceCollection::ConstView  topology, reco::PFClusteringVarsDeviceCollection::View  pfClusteringVars, reco::PFClusterDeviceCollection::View  clusterView, reco::PFRecHitFractionDeviceCollection::View  fracView  ) const

inline

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

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), debug, ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_multiSeedIterative(), ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_multiSeedParallel(), ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_singleSeed(), cms::alpakatools::once_per_block(), pfClusterHBHEAlpaka_cff::pfRecHits, ALPAKA_ACCELERATOR_NAMESPACE::threadsPerBlockForClustering, and pfClusterHBHEAlpaka_cff::topology.

                                                                                           {
      const int nRH = pfRecHits.size();
      int& topoId = alpaka::declareSharedVar<int, __COUNTER__>(acc);
      int& nRHTopo = alpaka::declareSharedVar<int, __COUNTER__>(acc);
      int& nSeeds = alpaka::declareSharedVar<int, __COUNTER__>(acc);

      if (once_per_block(acc)) {
        topoId = alpaka::getIdx<alpaka::Grid, alpaka::Blocks>(acc)[0u];
        nRHTopo = pfClusteringVars[topoId].topoRHCount();
        nSeeds = pfClusteringVars[topoId].topoSeedCount();
      }

      alpaka::syncBlockThreads(acc);  // all threads call sync

      if (topoId < nRH && nRHTopo > 0 && nSeeds > 0) {
        if (nRHTopo == nSeeds) {
          // PF cluster is isolated seed. No iterations needed
          if (once_per_block(acc)) {
            // Fill PFCluster-level information
            int rhIdx = pfClusteringVars[pfClusteringVars[topoId].topoSeedOffsets()].topoSeedList();
            int seedIdx = pfClusteringVars[rhIdx].rhIdxToSeedIdx();
            clusterView[seedIdx].energy() = pfRecHits[rhIdx].energy();
            clusterView[seedIdx].x() = pfRecHits[rhIdx].x();
            clusterView[seedIdx].y() = pfRecHits[rhIdx].y();
            clusterView[seedIdx].z() = pfRecHits[rhIdx].z();
          }
          // singleSeed and multiSeedParallel functions work only for GPU backend
        } else if ((not std::is_same_v<Device, alpaka::DevCpu>)&&nSeeds == 1) {
          // Single seed cluster
          hcalFastCluster_singleSeed(
              acc, pfClusParams, topology, topoId, nRHTopo, pfRecHits, pfClusteringVars, clusterView, fracView);
        } else if ((not std::is_same_v<Device, alpaka::DevCpu>)&&nSeeds <= 100 &&
                   nRHTopo - nSeeds < threadsPerBlockForClustering) {
          hcalFastCluster_multiSeedParallel(
              acc, pfClusParams, topology, topoId, nSeeds, nRHTopo, pfRecHits, pfClusteringVars, clusterView, fracView);
        } else if (nSeeds <= 400 && nRHTopo - nSeeds <= 1500) {
          // nSeeds value must match exotic in FastClusterExotic
          hcalFastCluster_multiSeedIterative(
              acc, pfClusParams, topology, topoId, nSeeds, nRHTopo, pfRecHits, pfClusteringVars, clusterView, fracView);
        } else {
          if constexpr (debug) {
            if (once_per_block(acc))
              printf("Topo cluster %d has %d seeds and %d rechits. Will be processed in next kernel.\n",
                     topoId,
                     nSeeds,
                     nRHTopo);
          }
        }
      }
    }