Skip to content

Commit

Permalink
Make type aliases more consistent
Browse files Browse the repository at this point in the history
Use consistently "OnHost" and "OnDevice" instead of "OnCPU" and "OnGPU"
  • Loading branch information
fwyzard committed Jan 19, 2023
1 parent b726c04 commit 94ba72e
Show file tree
Hide file tree
Showing 4 changed files with 29 additions and 37 deletions.
12 changes: 6 additions & 6 deletions RecoLocalTracker/SiPixelRecHits/plugins/SiPixelRecHitFromCUDA.cc
Expand Up @@ -33,7 +33,7 @@ class SiPixelRecHitFromCUDAT : public edm::stream::EDProducer<edm::ExternalWork>
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

using HMSstorage = HostProduct<uint32_t[]>;
using HitsOnGPU = TrackingRecHitSoADevice<TrackerTraits>;
using HitsOnDevice = TrackingRecHitSoADevice<TrackerTraits>;

private:
void acquire(edm::Event const& iEvent,
Expand All @@ -42,9 +42,9 @@ class SiPixelRecHitFromCUDAT : public edm::stream::EDProducer<edm::ExternalWork>
void produce(edm::Event& iEvent, edm::EventSetup const& iSetup) override;

const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geomToken_;
const edm::EDGetTokenT<cms::cuda::Product<HitsOnGPU>> hitsToken_; // CUDA hits
const edm::EDGetTokenT<SiPixelClusterCollectionNew> clusterToken_; // legacy clusters
const edm::EDPutTokenT<SiPixelRecHitCollection> rechitsPutToken_; // legacy rechits
const edm::EDGetTokenT<cms::cuda::Product<HitsOnDevice>> hitsToken_; // CUDA hits
const edm::EDGetTokenT<SiPixelClusterCollectionNew> clusterToken_; // legacy clusters
const edm::EDPutTokenT<SiPixelRecHitCollection> rechitsPutToken_; // legacy rechits
const edm::EDPutTokenT<HMSstorage> hostPutToken_;

uint32_t nHits_;
Expand All @@ -55,7 +55,7 @@ class SiPixelRecHitFromCUDAT : public edm::stream::EDProducer<edm::ExternalWork>
template <typename TrackerTraits>
SiPixelRecHitFromCUDAT<TrackerTraits>::SiPixelRecHitFromCUDAT(const edm::ParameterSet& iConfig)
: geomToken_(esConsumes()),
hitsToken_(consumes<cms::cuda::Product<HitsOnGPU>>(iConfig.getParameter<edm::InputTag>("pixelRecHitSrc"))),
hitsToken_(consumes<cms::cuda::Product<HitsOnDevice>>(iConfig.getParameter<edm::InputTag>("pixelRecHitSrc"))),
clusterToken_(consumes<SiPixelClusterCollectionNew>(iConfig.getParameter<edm::InputTag>("src"))),
rechitsPutToken_(produces<SiPixelRecHitCollection>()),
hostPutToken_(produces<HMSstorage>()) {}
Expand All @@ -73,7 +73,7 @@ template <typename TrackerTraits>
void SiPixelRecHitFromCUDAT<TrackerTraits>::acquire(edm::Event const& iEvent,
edm::EventSetup const& iSetup,
edm::WaitingTaskWithArenaHolder waitingTaskHolder) {
cms::cuda::Product<HitsOnGPU> const& inputDataWrapped = iEvent.get(hitsToken_);
cms::cuda::Product<HitsOnDevice> const& inputDataWrapped = iEvent.get(hitsToken_);

cms::cuda::ScopedContextAcquire ctx{inputDataWrapped, std::move(waitingTaskHolder)};

Expand Down
18 changes: 8 additions & 10 deletions RecoPixelVertexing/PixelTriplets/plugins/CAHitNtupletCUDA.cc
@@ -1,35 +1,33 @@
#include <cuda_runtime.h>

#include "CUDADataFormats/Common/interface/Product.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/RunningAverage.h"
#include "HeterogeneousCore/CUDACore/interface/ScopedContext.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "RecoTracker/TkMSParametrization/interface/PixelRecoUtilities.h"

#include "CAHitNtupletGeneratorOnGPU.h"

#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"

template <typename TrackerTraits>
class CAHitNtupletCUDAT : public edm::global::EDProducer<> {
using HitsConstView = TrackingRecHitSoAConstView<TrackerTraits>;
using HitsOnGPU = TrackingRecHitSoADevice<TrackerTraits>; //TODO move to OnDevice
using HitsOnCPU = TrackingRecHitSoAHost<TrackerTraits>; //TODO move to OnHost
using HitsOnDevice = TrackingRecHitSoADevice<TrackerTraits>;
using HitsOnHost = TrackingRecHitSoAHost<TrackerTraits>;

using TrackSoAHost = TrackSoAHeterogeneousHost<TrackerTraits>;
using TrackSoADevice = TrackSoAHeterogeneousDevice<TrackerTraits>;
Expand All @@ -51,9 +49,9 @@ class CAHitNtupletCUDAT : public edm::global::EDProducer<> {
bool onGPU_;

edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> tokenField_;
edm::EDGetTokenT<cms::cuda::Product<HitsOnGPU>> tokenHitGPU_;
edm::EDGetTokenT<cms::cuda::Product<HitsOnDevice>> tokenHitGPU_;
edm::EDPutTokenT<cms::cuda::Product<TrackSoADevice>> tokenTrackGPU_;
edm::EDGetTokenT<HitsOnCPU> tokenHitCPU_;
edm::EDGetTokenT<HitsOnHost> tokenHitCPU_;
edm::EDPutTokenT<TrackSoAHost> tokenTrackCPU_;

GPUAlgo gpuAlgo_;
Expand Down
Expand Up @@ -10,6 +10,10 @@
#include <functional>
#include <vector>

#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoADevice.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoAHost.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Event.h"
Expand All @@ -21,12 +25,6 @@
#include "HeterogeneousCore/CUDAServices/interface/CUDAService.h"
#include "TrackingTools/DetLayers/interface/BarrelDetLayer.h"

#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"

#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoAHost.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoADevice.h"

#include "CAHitNtupletGeneratorOnGPU.h"

namespace {
Expand Down Expand Up @@ -281,7 +279,7 @@ void CAHitNtupletGeneratorOnGPU<TrackerTraits>::endJob() {

template <typename TrackerTraits>
TrackSoAHeterogeneousDevice<TrackerTraits> CAHitNtupletGeneratorOnGPU<TrackerTraits>::makeTuplesAsync(
HitsOnGPU const& hits_d, float bfield, cudaStream_t stream) const {
HitsOnDevice const& hits_d, float bfield, cudaStream_t stream) const {
using HelixFitOnGPU = HelixFitOnGPU<TrackerTraits>;
using TrackSoA = TrackSoAHeterogeneousDevice<TrackerTraits>;
using GPUKernels = CAHitNtupletGeneratorKernelsGPU<TrackerTraits>;
Expand Down Expand Up @@ -314,7 +312,7 @@ TrackSoAHeterogeneousDevice<TrackerTraits> CAHitNtupletGeneratorOnGPU<TrackerTra
}

template <typename TrackerTraits>
TrackSoAHeterogeneousHost<TrackerTraits> CAHitNtupletGeneratorOnGPU<TrackerTraits>::makeTuples(HitsOnCPU const& hits_h,
TrackSoAHeterogeneousHost<TrackerTraits> CAHitNtupletGeneratorOnGPU<TrackerTraits>::makeTuples(HitsOnHost const& hits_h,
float bfield) const {
using HelixFitOnGPU = HelixFitOnGPU<TrackerTraits>;
using TrackSoA = TrackSoAHeterogeneousHost<TrackerTraits>;
Expand Down
Expand Up @@ -2,24 +2,20 @@
#define RecoPixelVertexing_PixelTriplets_plugins_CAHitNtupletGeneratorOnGPU_h

#include <cuda_runtime.h>
// #include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHit2DHeterogeneous.h"
// #include "CUDADataFormats/Track/interface/PixelTrackHeterogeneous.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"

#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitsUtilities.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoAHost.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousDevice.h"
#include "CUDADataFormats/Track/interface/TrackSoAHeterogeneousHost.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoADevice.h"

#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoAHost.h"
#include "CUDADataFormats/TrackingRecHit/interface/TrackingRecHitsUtilities.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "HeterogeneousCore/CUDAUtilities/interface/SimpleVector.h"

#include "CAHitNtupletGeneratorKernels.h"
#include "HelixFitOnGPU.h"

#include "GPUCACell.h"
#include "HelixFitOnGPU.h"

namespace edm {
class Event;
Expand All @@ -32,8 +28,8 @@ class CAHitNtupletGeneratorOnGPU {
public:
using HitsView = TrackingRecHitSoAView<TrackerTraits>;
using HitsConstView = TrackingRecHitSoAConstView<TrackerTraits>;
using HitsOnGPU = TrackingRecHitSoADevice<TrackerTraits>; //TODO move to OnDevice
using HitsOnCPU = TrackingRecHitSoAHost<TrackerTraits>; //TODO move to OnHost
using HitsOnDevice = TrackingRecHitSoADevice<TrackerTraits>;
using HitsOnHost = TrackingRecHitSoAHost<TrackerTraits>;
using hindex_type = typename TrackingRecHitSoA<TrackerTraits>::hindex_type;

using HitToTuple = caStructures::HitToTupleT<TrackerTraits>;
Expand Down Expand Up @@ -66,9 +62,9 @@ class CAHitNtupletGeneratorOnGPU {
void beginJob();
void endJob();

TrackSoADevice makeTuplesAsync(HitsOnGPU const& hits_d, float bfield, cudaStream_t stream) const;
TrackSoADevice makeTuplesAsync(HitsOnDevice const& hits_d, float bfield, cudaStream_t stream) const;

TrackSoAHost makeTuples(HitsOnCPU const& hits_d, float bfield) const;
TrackSoAHost makeTuples(HitsOnHost const& hits_d, float bfield) const;

private:
void buildDoublets(const HitsConstView& hh, cudaStream_t stream) const;
Expand Down

0 comments on commit 94ba72e

Please sign in to comment.