File indexing completed on 2024-04-06 12:25:53
0001 #include <iostream>
0002 #include <memory>
0003 #include <chrono>
0004 #include <cuda_runtime.h>
0005
0006 #include "FWCore/Framework/interface/ESProducer.h"
0007 #include "FWCore/Framework/interface/stream/EDProducer.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009 #include "FWCore/Utilities/interface/Exception.h"
0010 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
0011 #include "DataFormats/HGCRecHit/interface/HGCRecHit.h"
0012 #include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
0013 #include "DataFormats/ForwardDetId/interface/HGCSiliconDetId.h"
0014 #include "FWCore/Utilities/interface/EDGetToken.h"
0015 #include "FWCore/Utilities/interface/EDPutToken.h"
0016 #include "FWCore/Utilities/interface/InputTag.h"
0017
0018 #include "RecoLocalCalo/HGCalRecAlgos/interface/RecHitTools.h"
0019 #include "Geometry/HGCalGeometry/interface/HGCalGeometry.h"
0020 #include "Geometry/HGCalCommonData/interface/HGCalDDDConstants.h"
0021 #include "Geometry/HGCalCommonData/interface/HGCalWaferIndex.h"
0022
0023 #include "HeterogeneousCore/CUDACore/interface/ScopedContext.h"
0024 #include "HeterogeneousCore/CUDACore/interface/ContextState.h"
0025 #include "HeterogeneousCore/CUDAServices/interface/CUDAInterface.h"
0026 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
0027
0028 #include "FWCore/ServiceRegistry/interface/Service.h"
0029 #include "CommonTools/UtilAlgos/interface/TFileService.h"
0030 #include "RecoLocalCalo/HGCalRecProducers/plugins/KernelManagerHGCalRecHit.h"
0031
0032 #include "CUDADataFormats/HGCal/interface/HGCRecHitGPUProduct.h"
0033 #include "CUDADataFormats/HGCal/interface/HGCRecHitCPUProduct.h"
0034
0035 class EERecHitGPUtoSoA : public edm::stream::EDProducer<edm::ExternalWork> {
0036 public:
0037 explicit EERecHitGPUtoSoA(const edm::ParameterSet& ps);
0038 ~EERecHitGPUtoSoA() override;
0039
0040 void acquire(edm::Event const&, edm::EventSetup const&, edm::WaitingTaskWithArenaHolder) override;
0041 void produce(edm::Event&, const edm::EventSetup&) override;
0042
0043 private:
0044 cms::cuda::ContextState ctxState_;
0045 edm::EDGetTokenT<cms::cuda::Product<HGCRecHitGPUProduct>> recHitGPUToken_;
0046 edm::EDPutTokenT<HGCRecHitCPUProduct> recHitCPUSoAToken_;
0047
0048 std::unique_ptr<HGCRecHitCPUProduct> prodPtr_;
0049 };
0050
0051 EERecHitGPUtoSoA::EERecHitGPUtoSoA(const edm::ParameterSet& ps)
0052 : recHitGPUToken_{consumes<cms::cuda::Product<HGCRecHitGPUProduct>>(
0053 ps.getParameter<edm::InputTag>("EERecHitGPUTok"))},
0054 recHitCPUSoAToken_(produces<HGCRecHitCPUProduct>()) {}
0055
0056 EERecHitGPUtoSoA::~EERecHitGPUtoSoA() {}
0057
0058 void EERecHitGPUtoSoA::acquire(edm::Event const& event,
0059 edm::EventSetup const& setup,
0060 edm::WaitingTaskWithArenaHolder w) {
0061 cms::cuda::ScopedContextAcquire ctx{event.streamID(), std::move(w)};
0062 const auto& gpuRecHits = ctx.get(event, recHitGPUToken_);
0063
0064 prodPtr_ = std::make_unique<HGCRecHitCPUProduct>(gpuRecHits.nHits(), ctx.stream());
0065
0066 HGCRecHitCPUProduct& prod_ = *prodPtr_;
0067 KernelManagerHGCalRecHit km(prod_.get(), gpuRecHits.get());
0068 km.transfer_soa_to_host(ctx.stream());
0069 }
0070
0071 void EERecHitGPUtoSoA::produce(edm::Event& event, const edm::EventSetup& setup) { event.put(std::move(prodPtr_)); }
0072
0073 #include "FWCore/Framework/interface/MakerMacros.h"
0074 DEFINE_FWK_MODULE(EERecHitGPUtoSoA);
