Skip to content

Commit

Permalink
Add customize functions to disable conversions to legacy formats, and…
Browse files Browse the repository at this point in the history
… to disable unnecessary GPU->CPU transfers
  • Loading branch information
makortel committed Aug 14, 2018
1 parent de9cc8b commit 60e931f
Showing 1 changed file with 29 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,38 @@ def customizePixelTracksForProfiling(process):
),
verbosity = cms.untracked.uint32(0),
)

process.MessageLogger.cerr.FwkReport.reportEvery = 100

process.outPath = cms.EndPath(process.out)

process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.outPath)

return process

def customizePixelTracksForProfilingDisableConversion(process):
process = customizePixelTracksForProfiling(process)

# Turn off cluster shape filter so that CA doesn't depend on clusters
process.pixelTracksHitQuadruplets.SeedComparitorPSet = cms.PSet(ComponentName = cms.string("none"))

# Replace pixel track producer with a dummy one for now
from RecoPixelVertexing.PixelTrackFitting.pixelTrackProducerFromCUDA_cfi import pixelTrackProducerFromCUDA as _pixelTrackProducerFromCUDA
process.pixelTracks = _pixelTrackProducerFromCUDA.clone()

# Disable conversions to legacy
process.siPixelClustersPreSplitting.gpuEnableConversion = False
process.siPixelRecHitsPreSplitting.gpuEnableConversion = False
process.pixelTracksHitQuadruplets.gpuEnableConversion = False

return process

def customizePixelTracksForProfilingDisableTransfer(process):
process = customizePixelTracksForProfilingDisableConversion(process)

# Disable "unnecessary" transfers to CPU
process.siPixelClustersPreSplitting.gpuEnableTransfer = False
process.siPixelRecHitsPreSplitting.gpuEnableTransfer = False
process.pixelTracksHitQuadruplets.gpuEnableTransfer = False

return process

0 comments on commit 60e931f

Please sign in to comment.