PixelTrackSoAFromCUDA Class Reference

Inheritance diagram for PixelTrackSoAFromCUDA:

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

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
cms::cuda::host::unique_ptr< pixelTrack::TrackSoA >	soa_
edm::EDGetTokenT< cms::cuda::Product< PixelTrackHeterogeneous > >	tokenCUDA_
edm::EDPutTokenT< PixelTrackHeterogeneous >	tokenSOA_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::ExternalWork >
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 22 of file PixelTrackSoAFromCUDA.cc.

Constructor & Destructor Documentation

PixelTrackSoAFromCUDA()

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

explicit

Definition at line 41 of file PixelTrackSoAFromCUDA.cc.

    : tokenCUDA_(consumes<cms::cuda::Product<PixelTrackHeterogeneous>>(iConfig.getParameter<edm::InputTag>("src"))),
      tokenSOA_(produces<PixelTrackHeterogeneous>()) {}

~PixelTrackSoAFromCUDA()

PixelTrackSoAFromCUDA::~PixelTrackSoAFromCUDA ( )

overridedefault

Member Function Documentation

acquire()

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

overrideprivate

Definition at line 52 of file PixelTrackSoAFromCUDA.cc.

References iEvent, eostools::move(), soa_, and tokenCUDA_.

                                                                                     {
  cms::cuda::Product<PixelTrackHeterogeneous> const& inputDataWrapped = iEvent.get(tokenCUDA_);
  cms::cuda::ScopedContextAcquire ctx{inputDataWrapped, std::move(waitingTaskHolder)};
  auto const& inputData = ctx.get(inputDataWrapped);

  soa_ = inputData.toHostAsync(ctx.stream());
}

fillDescriptions()

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

static

Definition at line 45 of file PixelTrackSoAFromCUDA.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and HLT_2022v15_cff::InputTag.

                                                                                       {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("src", edm::InputTag("pixelTracksCUDA"));
  descriptions.add("pixelTracksSoA", desc);
}

produce()

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

overrideprivate

Definition at line 62 of file PixelTrackSoAFromCUDA.cc.

References cms::cuda::assert(), gather_cfg::cout, iEvent, DMR_cfg::maxTracks, eostools::move(), nHits, nt, BeamSpotPI::nTracks, soa_, and tokenSOA_.

                                                                                 {
  // check that the fixed-size SoA does not overflow
  auto const& tsoa = *soa_;
  auto maxTracks = tsoa.stride();
  auto nTracks = tsoa.nTracks();
  assert(nTracks < maxTracks);
  if (nTracks == maxTracks - 1) {
    edm::LogWarning("PixelTracks") << "Unsorted reconstructed pixel tracks truncated to " << maxTracks - 1
                                   << " candidates";
  }

#ifdef PIXEL_DEBUG_PRODUCE
  std::cout << "size of SoA " << sizeof(tsoa) << " stride " << maxTracks << std::endl;
  std::cout << "found " << nTracks << " tracks in cpu SoA at " << &tsoa << std::endl;

  int32_t nt = 0;
  for (int32_t it = 0; it < maxTracks; ++it) {
    auto nHits = tsoa.nHits(it);
    assert(nHits == int(tsoa.hitIndices.size(it)));
    if (nHits == 0)
      break;  // this is a guard: maybe we need to move to nTracks...
    nt++;
  }
  assert(nTracks == nt);
#endif

  // DO NOT  make a copy  (actually TWO....)
  iEvent.emplace(tokenSOA_, std::move(soa_));

  assert(!soa_);
}

Member Data Documentation

soa_

cms::cuda::host::unique_ptr<pixelTrack::TrackSoA> PixelTrackSoAFromCUDA::soa_

private

Definition at line 38 of file PixelTrackSoAFromCUDA.cc.

Referenced by acquire(), and produce().

tokenCUDA_

edm::EDGetTokenT<cms::cuda::Product<PixelTrackHeterogeneous> > PixelTrackSoAFromCUDA::tokenCUDA_

private

Definition at line 35 of file PixelTrackSoAFromCUDA.cc.

Referenced by acquire().

tokenSOA_

edm::EDPutTokenT<PixelTrackHeterogeneous> PixelTrackSoAFromCUDA::tokenSOA_

private

Definition at line 36 of file PixelTrackSoAFromCUDA.cc.

Referenced by produce().