PixelTrackSoAFromCUDAT< TrackerTraits > Class Template Reference

Inheritance diagram for PixelTrackSoAFromCUDAT< TrackerTraits >:

Public Member Functions
	PixelTrackSoAFromCUDAT (const edm::ParameterSet &iConfig)
	~PixelTrackSoAFromCUDAT () 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 Types
using	TrackSoADevice = TrackSoAHeterogeneousDevice< TrackerTraits >
using	TrackSoAHost = TrackSoAHeterogeneousHost< TrackerTraits >

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
edm::EDGetTokenT< cms::cuda::Product< TrackSoADevice > >	tokenCUDA_
edm::EDPutTokenT< TrackSoAHost >	tokenSOA_
TrackSoAHost	tracks_h_

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

template<typename TrackerTraits>
class PixelTrackSoAFromCUDAT< TrackerTraits >

Definition at line 26 of file PixelTrackSoAFromCUDA.cc.

Member Typedef Documentation

TrackSoADevice

template<typename TrackerTraits >

using PixelTrackSoAFromCUDAT< TrackerTraits >::TrackSoADevice = TrackSoAHeterogeneousDevice<TrackerTraits>

private

Definition at line 28 of file PixelTrackSoAFromCUDA.cc.

TrackSoAHost

template<typename TrackerTraits >

using PixelTrackSoAFromCUDAT< TrackerTraits >::TrackSoAHost = TrackSoAHeterogeneousHost<TrackerTraits>

private

Definition at line 27 of file PixelTrackSoAFromCUDA.cc.

Constructor & Destructor Documentation

PixelTrackSoAFromCUDAT()

template<typename TrackerTraits >

PixelTrackSoAFromCUDAT< TrackerTraits >::PixelTrackSoAFromCUDAT ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 49 of file PixelTrackSoAFromCUDA.cc.

    : tokenCUDA_(consumes(iConfig.getParameter<edm::InputTag>("src"))), tokenSOA_(produces<TrackSoAHost>()) {}

~PixelTrackSoAFromCUDAT()

template<typename TrackerTraits >

PixelTrackSoAFromCUDAT< TrackerTraits >::~PixelTrackSoAFromCUDAT ( )

overridedefault

Member Function Documentation

acquire()

template<typename TrackerTraits >

void PixelTrackSoAFromCUDAT< TrackerTraits >::acquire ( edm::Event const &  iEvent, edm::EventSetup const &  iSetup, edm::WaitingTaskWithArenaHolder  waitingTaskHolder  )

overrideprivate

Definition at line 61 of file PixelTrackSoAFromCUDA.cc.

References cudaCheck, iEvent, and eostools::move().

                                                                                                     {
  cms::cuda::Product<TrackSoADevice> const& inputDataWrapped = iEvent.get(tokenCUDA_);
  cms::cuda::ScopedContextAcquire ctx{inputDataWrapped, std::move(waitingTaskHolder)};
  auto const& tracks_d = ctx.get(inputDataWrapped);  // Tracks on device
  tracks_h_ = TrackSoAHost(ctx.stream());            // Create an instance of Tracks on Host, using the stream
  cudaCheck(cudaMemcpyAsync(tracks_h_.buffer().get(),
                            tracks_d.const_buffer().get(),
                            tracks_d.bufferSize(),
                            cudaMemcpyDeviceToHost,
                            ctx.stream()));  // Copy data from Device to Host
}

fillDescriptions()

template<typename TrackerTraits >

void PixelTrackSoAFromCUDAT< TrackerTraits >::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 53 of file PixelTrackSoAFromCUDA.cc.

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

                                                                                                       {
  edm::ParameterSetDescription desc;

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

produce()

template<typename TrackerTraits >

void PixelTrackSoAFromCUDAT< TrackerTraits >::produce ( edm::Event &  iEvent, edm::EventSetup const &  iSetup  )

overrideprivate

Definition at line 76 of file PixelTrackSoAFromCUDA.cc.

References cms::cuda::assert(), gather_cfg::cout, iEvent, DMR_cfg::maxTracks, eostools::move(), nHits, TracksUtilities< TrackerTraits >::nHits(), nt, and BeamSpotPI::nTracks.

                                                                                                 {
  auto maxTracks = tracks_h_.view().metadata().size();
  auto nTracks = tracks_h_.view().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 = TracksUtilities<TrackerTraits>::nHits(tracks_h_.view(), it);
    assert(nHits == int(tracks_h_.view().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(tracks_h_));
  assert(!tracks_h_.buffer());
}

Member Data Documentation

tokenCUDA_

template<typename TrackerTraits >

edm::EDGetTokenT<cms::cuda::Product<TrackSoADevice> > PixelTrackSoAFromCUDAT< TrackerTraits >::tokenCUDA_

private

Definition at line 42 of file PixelTrackSoAFromCUDA.cc.

tokenSOA_

template<typename TrackerTraits >

edm::EDPutTokenT<TrackSoAHost> PixelTrackSoAFromCUDAT< TrackerTraits >::tokenSOA_

private

Definition at line 43 of file PixelTrackSoAFromCUDA.cc.

tracks_h_

template<typename TrackerTraits >

TrackSoAHost PixelTrackSoAFromCUDAT< TrackerTraits >::tracks_h_

private

Definition at line 45 of file PixelTrackSoAFromCUDA.cc.