Skip to content

Commit

Permalink
Migrate cluster track associator (#409)
Browse files Browse the repository at this point in the history
Migrate ClusterTPAssociationHeterogeneous using the depreacted HeterogeneousEDProducer to
ClusterTPAssociationProducerCUDA, and implement a simple analyzer to consume its procuct.

To test it, add a dummy analyzer to an MC workflow:

    process.load("SimTracker.TrackerHitAssociation.clusterTPCUDAdump_cfi")
    process.validation_step = cms.EndPath(process.globalValidationPixelTrackingOnly + process.clusterTPCUDAdump)
    process.tpClusterProducerCUDAPreSplitting.dumpCSV = True
  • Loading branch information
VinInn authored and fwyzard committed Oct 20, 2020
1 parent 168ae02 commit f33e1ab
Showing 1 changed file with 13 additions and 6 deletions.
19 changes: 13 additions & 6 deletions Validation/RecoTrack/python/TrackValidation_cff.py
Original file line number Diff line number Diff line change
Expand Up @@ -485,13 +485,11 @@ def _getMVASelectors(postfix):
# Built tracks, in the standard sequence mainly for monitoring the track selection MVA
tpClusterProducerPreSplitting = tpClusterProducer.clone(pixelClusterSrc = "siPixelClustersPreSplitting")
quickTrackAssociatorByHitsPreSplitting = quickTrackAssociatorByHits.clone(cluster2TPSrc = "tpClusterProducerPreSplitting")
tpClusterProducerHeterogeneousPreSplitting = tpClusterProducerHeterogeneous.clone(

tpClusterProducerCUDAPreSplitting = tpClusterProducerCUDA.clone(
pixelClusterSrc = "siPixelClustersPreSplitting"
)
from Configuration.ProcessModifiers.gpu_cff import gpu
gpu.toReplaceWith(tpClusterProducerPreSplitting, tpClusterProducerConverter.clone(
src = "tpClusterProducerHeterogeneousPreSplitting"
))

_trackValidatorSeedingBuilding = trackValidator.clone( # common for built tracks and seeds (in trackingOnly)
associators = ["quickTrackAssociatorByHits"],
UseAssociators = True,
Expand Down Expand Up @@ -595,14 +593,23 @@ def _uniqueFirstLayers(layerList):
)
tracksValidationTruth = cms.Task(
tpClusterProducer,
tpClusterProducerHeterogeneousPreSplitting,
tpClusterProducerPreSplitting,
quickTrackAssociatorByHits,
quickTrackAssociatorByHitsPreSplitting,
trackingParticleRecoTrackAsssociation,
VertexAssociatorByPositionAndTracks,
trackingParticleNumberOfLayersProducer
)

#gpu tp ???
from Configuration.ProcessModifiers.gpu_cff import gpu
tpClusterProducerPreSplittingCUDA = cms.Task(
tpClusterProducerCUDAPreSplitting
)
_tracksValidationTruth_gpu = tracksValidationTruth.copy()
_tracksValidationTruth_gpu.add(tpClusterProducerPreSplittingCUDA)
gpu.toReplaceWith(tracksValidationTruth,_tracksValidationTruth_gpu)

fastSim.toModify(tracksValidationTruth, lambda x: x.remove(tpClusterProducer))

tracksPreValidation = cms.Task(
Expand Down

0 comments on commit f33e1ab

Please sign in to comment.