SiPixelRecHits/python/SiPixelRecHits_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002 from HeterogeneousCore.AlpakaCore.functions import *
0003 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0004
0005
0006 # legacy pixel rechit producer
0007 from RecoLocalTracker.SiPixelRecHits.siPixelRecHitConverter_cfi import siPixelRecHitConverter as _siPixelRecHitConverter
0008 siPixelRecHits = _siPixelRecHitConverter.clone()
0009
0010 # HIon Modifiers
0011 from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
0012 # Phase 2 Tracker Modifier
0013 from Configuration.Eras.Modifier_phase2_tracker_cff import phase2_tracker
0014
0015 # The legacy pixel rechit producer
0016 siPixelRecHitsPreSplitting = siPixelRecHits.clone(
0017         src = 'siPixelClustersPreSplitting'
0018 )
0019
0020 siPixelRecHitsPreSplittingTask = cms.Task(
0021     siPixelRecHitsPreSplitting
0022 )
0023
0024 ######################################################################
0025
0026 ### Alpaka Pixel Hits Reco
0027 from RecoLocalTracker.SiPixelRecHits.siPixelRecHitAlpakaPhase1_cfi import siPixelRecHitAlpakaPhase1 as _siPixelRecHitAlpakaPhase1
0028 from RecoLocalTracker.SiPixelRecHits.siPixelRecHitAlpakaPhase2_cfi import siPixelRecHitAlpakaPhase2 as _siPixelRecHitAlpakaPhase2
0029 from RecoLocalTracker.SiPixelRecHits.siPixelRecHitAlpakaHIonPhase1_cfi import siPixelRecHitAlpakaHIonPhase1 as _siPixelRecHitAlpakaHIonPhase1
0030
0031
0032 # Hit SoA producer on the device
0033 siPixelRecHitsPreSplittingAlpaka = _siPixelRecHitAlpakaPhase1.clone(
0034     src = "siPixelClustersPreSplittingAlpaka"
0035 )
0036 phase2_tracker.toReplaceWith(siPixelRecHitsPreSplittingAlpaka,_siPixelRecHitAlpakaPhase2.clone(
0037     src = "siPixelClustersPreSplittingAlpaka"
0038 ))
0039 (pp_on_AA & ~phase2_tracker).toReplaceWith(siPixelRecHitsPreSplittingAlpaka,_siPixelRecHitAlpakaHIonPhase1.clone(
0040     src = "siPixelClustersPreSplittingAlpaka"
0041 ))
0042
0043 # Hit SoA producer on the cpu, for validation
0044 siPixelRecHitsPreSplittingAlpakaSerial = makeSerialClone(siPixelRecHitsPreSplittingAlpaka,
0045     src = "siPixelClustersPreSplittingAlpakaSerial"
0046 )
0047
0048 from RecoLocalTracker.SiPixelRecHits.siPixelRecHitFromSoAAlpaka_cfi import siPixelRecHitFromSoAAlpaka as _siPixelRecHitFromSoAAlpaka
0049
0050 alpaka.toReplaceWith(siPixelRecHitsPreSplitting, _siPixelRecHitFromSoAAlpaka.clone(
0051             pixelRecHitSrc = cms.InputTag('siPixelRecHitsPreSplittingAlpaka'),
0052             src = cms.InputTag('siPixelClustersPreSplitting'))
0053 )
0054
0055 (alpaka & pp_on_AA & ~phase2_tracker).toModify(siPixelRecHitsPreSplitting,
0056             maxHitsInModules = cms.uint32(2048)
0057 )
0058
0059
0060 alpaka.toReplaceWith(siPixelRecHitsPreSplittingTask, cms.Task(
0061                         # Reconstruct the pixel hits with alpaka on the device
0062                         siPixelRecHitsPreSplittingAlpaka,
0063                         # Reconstruct the pixel hits with alpaka on the cpu (if requested by the validation)
0064                         siPixelRecHitsPreSplittingAlpakaSerial,
0065                         # Convert hit soa on host to legacy formats
0066                         siPixelRecHitsPreSplitting))