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 23 of file PixelTrackSoAFromCUDA.cc.

Constructor & Destructor Documentation

PixelTrackSoAFromCUDA()

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

explicit

Definition at line 42 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 53 of file PixelTrackSoAFromCUDA.cc.

                                                                                     {
  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());
}

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

fillDescriptions()

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

static

Definition at line 46 of file PixelTrackSoAFromCUDA.cc.

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

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

produce()

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

overrideprivate

Definition at line 63 of file PixelTrackSoAFromCUDA.cc.

                                                                                 {
#ifdef PIXEL_DEBUG_PRODUCE
  auto const& tsoa = *soa_;
  auto maxTracks = tsoa.stride();
  std::cout << "size of SoA" << sizeof(tsoa) << " stride " << maxTracks << 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++;
  }
  std::cout << "found " << nt << " tracks in cpu SoA at " << &tsoa << std::endl;
#endif
 
  // DO NOT  make a copy  (actually TWO....)
  iEvent.emplace(tokenSOA_, PixelTrackHeterogeneous(std::move(soa_)));
 
  assert(!soa_);
}

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

Member Data Documentation

soa_

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

private

Definition at line 39 of file PixelTrackSoAFromCUDA.cc.

Referenced by acquire(), and produce().

tokenCUDA_

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

private

Definition at line 36 of file PixelTrackSoAFromCUDA.cc.

Referenced by acquire().

tokenSOA_

edm::EDPutTokenT<PixelTrackHeterogeneous> PixelTrackSoAFromCUDA::tokenSOA_

private

Definition at line 37 of file PixelTrackSoAFromCUDA.cc.

Referenced by produce().