SiPixelRecHitFromCUDA Class Reference

Inheritance diagram for SiPixelRecHitFromCUDA:

Public Types
using	HMSstorage = HostProduct< uint32_t[]>
Public Types inherited from edm::stream::EDProducer< edm::ExternalWork >
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Member Functions
	SiPixelRecHitFromCUDA (const edm::ParameterSet &iConfig)
	~SiPixelRecHitFromCUDA () override=default
Public Member Functions inherited from edm::stream::EDProducer< edm::ExternalWork >
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, edm::WaitingTaskWithArenaHolder waitingTaskHolder) override
void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup) override

Private Attributes
const edm::EDGetTokenT< SiPixelClusterCollectionNew >	clusterToken_
const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >	geomToken_
cms::cuda::host::unique_ptr< uint32_t[]>	hitsModuleStart_
const edm::EDGetTokenT< cms::cuda::Product< TrackingRecHit2DCUDA > >	hitsToken_
const edm::EDPutTokenT< HMSstorage >	hostPutToken_
uint32_t	nHits_
const edm::EDPutTokenT< SiPixelRecHitCollection >	rechitsPutToken_
cms::cuda::host::unique_ptr< float[]>	store32_

Detailed Description

Definition at line 27 of file SiPixelRecHitFromCUDA.cc.

Member Typedef Documentation

HMSstorage

using SiPixelRecHitFromCUDA::HMSstorage = HostProduct<uint32_t[]>

Definition at line 34 of file SiPixelRecHitFromCUDA.cc.

Constructor & Destructor Documentation

SiPixelRecHitFromCUDA()

SiPixelRecHitFromCUDA::SiPixelRecHitFromCUDA ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 53 of file SiPixelRecHitFromCUDA.cc.

    : geomToken_(esConsumes()),
      hitsToken_(
          consumes<cms::cuda::Product<TrackingRecHit2DCUDA>>(iConfig.getParameter<edm::InputTag>("pixelRecHitSrc"))),
      clusterToken_(consumes<SiPixelClusterCollectionNew>(iConfig.getParameter<edm::InputTag>("src"))),
      rechitsPutToken_(produces<SiPixelRecHitCollection>()),
      hostPutToken_(produces<HMSstorage>()) {}

~SiPixelRecHitFromCUDA()

SiPixelRecHitFromCUDA::~SiPixelRecHitFromCUDA ( )

overridedefault

Member Function Documentation

acquire()

void SiPixelRecHitFromCUDA::acquire ( edm::Event const &  iEvent, edm::EventSetup const &  iSetup, edm::WaitingTaskWithArenaHolder  waitingTaskHolder  )

overrideprivate

Definition at line 68 of file SiPixelRecHitFromCUDA.cc.

                                                                                     {
  cms::cuda::Product<TrackingRecHit2DCUDA> const& inputDataWrapped = iEvent.get(hitsToken_);
  cms::cuda::ScopedContextAcquire ctx{inputDataWrapped, std::move(waitingTaskHolder)};
  auto const& inputData = ctx.get(inputDataWrapped);
 
  nHits_ = inputData.nHits();
 
  LogDebug("SiPixelRecHitFromCUDA") << "converting " << nHits_ << " Hits";
 
  if (0 == nHits_)
    return;
  store32_ = inputData.localCoordToHostAsync(ctx.stream());
  hitsModuleStart_ = inputData.hitsModuleStartToHostAsync(ctx.stream());
}

References hitsModuleStart_, hitsToken_, iEvent, LogDebug, eostools::move(), nHits_, and store32_.

fillDescriptions()

void SiPixelRecHitFromCUDA::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 61 of file SiPixelRecHitFromCUDA.cc.

                                                                                       {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("pixelRecHitSrc", edm::InputTag("siPixelRecHitsPreSplittingCUDA"));
  desc.add<edm::InputTag>("src", edm::InputTag("siPixelClustersPreSplitting"));
  descriptions.addWithDefaultLabel(desc);
}

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, and HLT_FULL_cff::InputTag.

produce()

void SiPixelRecHitFromCUDA::produce ( edm::Event &  iEvent, edm::EventSetup const &  iSetup  )

overrideprivate

Definition at line 85 of file SiPixelRecHitFromCUDA.cc.

                                                                             {
  // allocate a buffer for the indices of the clusters
  auto hmsp = std::make_unique<uint32_t[]>(gpuClustering::maxNumModules + 1);
  std::copy(hitsModuleStart_.get(), hitsModuleStart_.get() + gpuClustering::maxNumModules + 1, hmsp.get());
  // wrap the buffer in a HostProduct, and move it to the Event, without reallocating the buffer or affecting hitsModuleStart
  iEvent.emplace(hostPutToken_, std::move(hmsp));
 
  SiPixelRecHitCollection output;
  if (0 == nHits_) {
    iEvent.emplace(rechitsPutToken_, std::move(output));
    return;
  }
 
  auto xl = store32_.get();
  auto yl = xl + nHits_;
  auto xe = yl + nHits_;
  auto ye = xe + nHits_;
 
  const TrackerGeometry* geom = &es.getData(geomToken_);
 
  edm::Handle<SiPixelClusterCollectionNew> hclusters = iEvent.getHandle(clusterToken_);
  auto const& input = *hclusters;
 
  constexpr uint32_t maxHitsInModule = gpuClustering::maxHitsInModule();
 
  int numberOfDetUnits = 0;
  int numberOfClusters = 0;
  for (auto const& dsv : input) {
    numberOfDetUnits++;
    unsigned int detid = dsv.detId();
    DetId detIdObject(detid);
    const GeomDetUnit* genericDet = geom->idToDetUnit(detIdObject);
    auto gind = genericDet->index();
    const PixelGeomDetUnit* pixDet = dynamic_cast<const PixelGeomDetUnit*>(genericDet);
    assert(pixDet);
    SiPixelRecHitCollection::FastFiller recHitsOnDetUnit(output, detid);
    auto fc = hitsModuleStart_[gind];
    auto lc = hitsModuleStart_[gind + 1];
    auto nhits = lc - fc;
 
    assert(lc > fc);
    LogDebug("SiPixelRecHitFromCUDA") << "in det " << gind << ": conv " << nhits << " hits from " << dsv.size()
                                      << " legacy clusters" << ' ' << fc << ',' << lc;
    if (nhits > maxHitsInModule)
      edm::LogWarning("SiPixelRecHitFromCUDA") << fmt::sprintf(
          "Too many clusters %d in module %d. Only the first %d hits will be converted", nhits, gind, maxHitsInModule);
    nhits = std::min(nhits, maxHitsInModule);
 
    LogDebug("SiPixelRecHitFromCUDA") << "in det " << gind << "conv " << nhits << " hits from " << dsv.size()
                                      << " legacy clusters" << ' ' << lc << ',' << fc;
 
    if (0 == nhits)
      continue;
    auto jnd = [&](int k) { return fc + k; };
    assert(nhits <= dsv.size());
    if (nhits != dsv.size()) {
      edm::LogWarning("GPUHits2CPU") << "nhits!= nclus " << nhits << ' ' << dsv.size();
    }
    for (auto const& clust : dsv) {
      assert(clust.originalId() >= 0);
      assert(clust.originalId() < dsv.size());
      if (clust.originalId() >= nhits)
        continue;
      auto ij = jnd(clust.originalId());
      if (ij >= TrackingRecHit2DSOAView::maxHits())
        continue;  // overflow...
      LocalPoint lp(xl[ij], yl[ij]);
      LocalError le(xe[ij], 0, ye[ij]);
      SiPixelRecHitQuality::QualWordType rqw = 0;
 
      numberOfClusters++;
 
      /* cpu version....  (for reference)
      std::tuple<LocalPoint, LocalError, SiPixelRecHitQuality::QualWordType> tuple = cpe_->getParameters( clust, *genericDet );
      LocalPoint lp( std::get<0>(tuple) );
      LocalError le( std::get<1>(tuple) );
      SiPixelRecHitQuality::QualWordType rqw( std::get<2>(tuple) );
      */
 
      // Create a persistent edm::Ref to the cluster
      edm::Ref<edmNew::DetSetVector<SiPixelCluster>, SiPixelCluster> cluster = edmNew::makeRefTo(hclusters, &clust);
      // Make a RecHit and add it to the DetSet
      SiPixelRecHit hit(lp, le, rqw, *genericDet, cluster);
      //
      // Now save it =================
      recHitsOnDetUnit.push_back(hit);
      // =============================
 
      LogDebug("SiPixelRecHitFromCUDA") << "cluster " << numberOfClusters << " at " << lp << ' ' << le;
 
    }  //  <-- End loop on Clusters
 
    //  LogDebug("SiPixelRecHitGPU")
    LogDebug("SiPixelRecHitFromCUDA") << "found " << recHitsOnDetUnit.size() << " RecHits on " << detid;
 
  }  //    <-- End loop on DetUnits
 
  LogDebug("SiPixelRecHitFromCUDA") << "found " << numberOfDetUnits << " dets, " << numberOfClusters << " clusters";
 
  iEvent.emplace(rechitsPutToken_, std::move(output));
}

References cms::cuda::assert(), clusterToken_, filterCSVwithJSON::copy, benchmark_cfg::fc, relativeConstraints::geom, geomToken_, edm::EventSetup::getData(), hitsModuleStart_, hostPutToken_, iEvent, GeomDet::index(), input, dqmdumpme::k, LogDebug, edmNew::makeRefTo(), TrackingRecHit2DSOAView::maxHits(), gpuClustering::maxHitsInModule(), gpuClustering::maxNumModules, min(), eostools::move(), nhits, nHits_, convertSQLitetoXML_cfg::output, edmNew::DetSetVector< T >::FastFiller::push_back(), rechitsPutToken_, edmNew::DetSetVector< T >::FastFiller::size(), and store32_.

Member Data Documentation

clusterToken_

const edm::EDGetTokenT<SiPixelClusterCollectionNew> SiPixelRecHitFromCUDA::clusterToken_

private

Definition at line 44 of file SiPixelRecHitFromCUDA.cc.

Referenced by produce().

geomToken_

const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> SiPixelRecHitFromCUDA::geomToken_

private

Definition at line 42 of file SiPixelRecHitFromCUDA.cc.

Referenced by produce().

hitsModuleStart_

cms::cuda::host::unique_ptr<uint32_t[]> SiPixelRecHitFromCUDA::hitsModuleStart_

private

Definition at line 50 of file SiPixelRecHitFromCUDA.cc.

Referenced by acquire(), and produce().

hitsToken_

const edm::EDGetTokenT<cms::cuda::Product<TrackingRecHit2DCUDA> > SiPixelRecHitFromCUDA::hitsToken_

private

Definition at line 43 of file SiPixelRecHitFromCUDA.cc.

Referenced by acquire().

hostPutToken_

const edm::EDPutTokenT<HMSstorage> SiPixelRecHitFromCUDA::hostPutToken_

private

Definition at line 46 of file SiPixelRecHitFromCUDA.cc.

Referenced by produce().

nHits_

uint32_t SiPixelRecHitFromCUDA::nHits_

private

Definition at line 48 of file SiPixelRecHitFromCUDA.cc.

Referenced by acquire(), and produce().

rechitsPutToken_

const edm::EDPutTokenT<SiPixelRecHitCollection> SiPixelRecHitFromCUDA::rechitsPutToken_

private

Definition at line 45 of file SiPixelRecHitFromCUDA.cc.

Referenced by produce().

store32_

cms::cuda::host::unique_ptr<float[]> SiPixelRecHitFromCUDA::store32_

private

Definition at line 49 of file SiPixelRecHitFromCUDA.cc.

Referenced by acquire(), and produce().