CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
EERecHitGPUtoSoA Class Reference
Inheritance diagram for EERecHitGPUtoSoA:
edm::stream::EDProducer< edm::ExternalWork >

Public Member Functions

void acquire (edm::Event const &, edm::EventSetup const &, edm::WaitingTaskWithArenaHolder) override
 
 EERecHitGPUtoSoA (const edm::ParameterSet &ps)
 
void produce (edm::Event &, const edm::EventSetup &) override
 
 ~EERecHitGPUtoSoA () override
 
- 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 EDProduceroperator= (const EDProducer &)=delete
 

Private Attributes

cms::cuda::ContextState ctxState_
 
std::unique_ptr< HGCRecHitCPUProductprodPtr_
 
edm::EDPutTokenT< HGCRecHitCPUProductrecHitCPUSoAToken_
 
edm::EDGetTokenT< cms::cuda::Product< HGCRecHitGPUProduct > > recHitGPUToken_
 

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 35 of file EERecHitGPUtoSoA.cc.

Constructor & Destructor Documentation

◆ EERecHitGPUtoSoA()

EERecHitGPUtoSoA::EERecHitGPUtoSoA ( const edm::ParameterSet ps)
explicit

Definition at line 51 of file EERecHitGPUtoSoA.cc.

References edm::ParameterSet::getParameter().

52  : recHitGPUToken_{consumes<cms::cuda::Product<HGCRecHitGPUProduct>>(
53  ps.getParameter<edm::InputTag>("EERecHitGPUTok"))},
54  recHitCPUSoAToken_(produces<HGCRecHitCPUProduct>()) {}
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
edm::EDGetTokenT< cms::cuda::Product< HGCRecHitGPUProduct > > recHitGPUToken_
edm::EDPutTokenT< HGCRecHitCPUProduct > recHitCPUSoAToken_

◆ ~EERecHitGPUtoSoA()

EERecHitGPUtoSoA::~EERecHitGPUtoSoA ( )
override

Definition at line 56 of file EERecHitGPUtoSoA.cc.

56 {}

Member Function Documentation

◆ acquire()

void EERecHitGPUtoSoA::acquire ( edm::Event const &  event,
edm::EventSetup const &  setup,
edm::WaitingTaskWithArenaHolder  w 
)
override

Definition at line 58 of file EERecHitGPUtoSoA.cc.

References eostools::move(), prodPtr_, recHitGPUToken_, KernelManagerHGCalRecHit::transfer_soa_to_host(), and w().

60  {
61  cms::cuda::ScopedContextAcquire ctx{event.streamID(), std::move(w)};
62  const auto& gpuRecHits = ctx.get(event, recHitGPUToken_);
63 
64  prodPtr_ = std::make_unique<HGCRecHitCPUProduct>(gpuRecHits.nHits(), ctx.stream());
65 
66  HGCRecHitCPUProduct& prod_ = *prodPtr_;
67  KernelManagerHGCalRecHit km(prod_.get(), gpuRecHits.get());
68  km.transfer_soa_to_host(ctx.stream());
69 }
T w() const
edm::EDGetTokenT< cms::cuda::Product< HGCRecHitGPUProduct > > recHitGPUToken_
std::unique_ptr< HGCRecHitCPUProduct > prodPtr_
void transfer_soa_to_host(const cudaStream_t &)
def move(src, dest)
Definition: eostools.py:511
Definition: event.py:1

◆ produce()

void EERecHitGPUtoSoA::produce ( edm::Event event,
const edm::EventSetup setup 
)
override

Definition at line 71 of file EERecHitGPUtoSoA.cc.

References eostools::move(), and prodPtr_.

71 { event.put(std::move(prodPtr_)); }
std::unique_ptr< HGCRecHitCPUProduct > prodPtr_
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

◆ ctxState_

cms::cuda::ContextState EERecHitGPUtoSoA::ctxState_
private

Definition at line 44 of file EERecHitGPUtoSoA.cc.

◆ prodPtr_

std::unique_ptr<HGCRecHitCPUProduct> EERecHitGPUtoSoA::prodPtr_
private

Definition at line 48 of file EERecHitGPUtoSoA.cc.

Referenced by acquire(), and produce().

◆ recHitCPUSoAToken_

edm::EDPutTokenT<HGCRecHitCPUProduct> EERecHitGPUtoSoA::recHitCPUSoAToken_
private

Definition at line 46 of file EERecHitGPUtoSoA.cc.

◆ recHitGPUToken_

edm::EDGetTokenT<cms::cuda::Product<HGCRecHitGPUProduct> > EERecHitGPUtoSoA::recHitGPUToken_
private

Definition at line 45 of file EERecHitGPUtoSoA.cc.

Referenced by acquire().