CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
customizePixelOnlyForProfiling.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
2 
3 # Customise the Pixel-only reconstruction to run on GPU
4 #
5 # Run the unpacker, clustering, ntuplets, track fit and vertex reconstruction on GPU.
7 
8  process.consumer = cms.EDAnalyzer("GenericConsumer",
9  eventProducts = cms.untracked.vstring('pixelTracksCUDA', 'pixelVerticesCUDA')
10  )
11 
12  process.consume_step = cms.EndPath(process.consumer)
13 
14  process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
15 
16  return process
17 
18 
19 # Customise the Pixel-only reconstruction to run on GPU, and copy the data to the host
20 #
21 # Run the unpacker, clustering, ntuplets, track fit and vertex reconstruction on GPU,
22 # and copy all the products to the host in SoA format.
23 #
24 # The same customisation can be also used on the SoA CPU workflow, running up to the
25 # tracks and vertices on the CPU in SoA format, without conversion to legacy format.
27 
28  #? process.siPixelRecHitSoAFromLegacy.convertToLegacy = False
29 
30  process.consumer = cms.EDAnalyzer("GenericConsumer",
31  eventProducts = cms.untracked.vstring('pixelTracksSoA', 'pixelVerticesSoA')
32  )
33 
34  process.consume_step = cms.EndPath(process.consumer)
35 
36  process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
37 
38  return process
39 
40 
41 # Customise the Pixel-only reconstruction to run on GPU, copy the data to the host,
42 # and convert to legacy format
43 #
44 # Run the unpacker, clustering, ntuplets, track fit and vertex reconstruction on GPU;
45 # copy all the products to the host in SoA format; and convert them to legacy format.
46 #
47 # The same customisation can be also used on the CPU workflow, running up to the
48 # tracks and vertices on the CPU.
50 
51  process.consumer = cms.EDAnalyzer("GenericConsumer",
52  eventProducts = cms.untracked.vstring('pixelTracks', 'pixelVertices')
53  )
54 
55  process.consume_step = cms.EndPath(process.consumer)
56 
57  process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
58 
59  return process