Skip to content

Commit 0f56b14

Browse files
authored
Merge pull request #42061 from srimanob/132_EnableProdLikePatatrackWFProfiling
Enable .59521 and .59621 wfs for profiling
2 parents 09e7de6 + 2706dee commit 0f56b14

File tree

1 file changed

+53
-6
lines changed

1 file changed

+53
-6
lines changed

Configuration/PyReleaseValidation/python/upgradeWorkflowComponents.py

+53-6
Original file line numberDiff line numberDiff line change
@@ -772,7 +772,7 @@ def condition(self, fragment, stepList, key, hasHarvest):
772772
# - 2023 conditions, Z->mumu
773773
# - 2026D88 conditions, TTbar
774774
class PatatrackWorkflow(UpgradeWorkflow):
775-
def __init__(self, digi = {}, reco = {}, harvest = {}, **kwargs):
775+
def __init__(self, digi = {}, reco = {}, mini = {}, harvest = {}, **kwargs):
776776
# adapt the parameters for the UpgradeWorkflow init method
777777
super(PatatrackWorkflow, self).__init__(
778778
steps = [
@@ -788,6 +788,7 @@ def __init__(self, digi = {}, reco = {}, harvest = {}, **kwargs):
788788
'RecoNanoFakeHLT',
789789
'HARVESTNano',
790790
'HARVESTNanoFakeHLT',
791+
'MiniAOD',
791792
'Nano',
792793
'ALCA',
793794
'ALCAPhase2'
@@ -805,17 +806,20 @@ def __init__(self, digi = {}, reco = {}, harvest = {}, **kwargs):
805806
'RecoNanoFakeHLT',
806807
'HARVESTNano',
807808
'HARVESTNanoFakeHLT',
809+
'MiniAOD',
808810
'Nano',
809811
'ALCA',
810812
'ALCAPhase2'
811813
],
812814
**kwargs)
813815
self.__digi = digi
814816
self.__reco = reco
815-
self.__reco.update({
816-
'--datatier': 'GEN-SIM-RECO,DQMIO',
817-
'--eventcontent': 'RECOSIM,DQM'
818-
})
817+
if 'DQM' in self.__reco:
818+
self.__reco.update({
819+
'--datatier': 'GEN-SIM-RECO,DQMIO',
820+
'--eventcontent': 'RECOSIM,DQM'
821+
})
822+
self.__mini = mini
819823
self.__harvest = harvest
820824

821825
def condition(self, fragment, stepList, key, hasHarvest):
@@ -847,6 +851,11 @@ def setup_(self, step, stepName, stepDict, k, properties):
847851
stepDict[stepName][k] = None
848852
else:
849853
stepDict[stepName][k] = merge([self.__reco, stepDict[step][k]])
854+
elif 'MiniAOD' in step:
855+
if self.__mini is None:
856+
stepDict[stepName][k] = None
857+
else:
858+
stepDict[stepName][k] = merge([self.__mini, stepDict[step][k]])
850859
elif 'HARVEST' in step:
851860
if self.__harvest is None:
852861
stepDict[stepName][k] = None
@@ -1390,6 +1399,25 @@ def setup_(self, step, stepName, stepDict, k, properties):
13901399
suffix = 'Patatrack_FullRecoTripletsCPU',
13911400
offset = 0.595,
13921401
)
1402+
# - ProdLike
1403+
upgradeWFs['PatatrackFullRecoTripletsCPUProdLike'] = PatatrackWorkflow(
1404+
digi = {
1405+
# the HLT menu is already set up for using GPUs if available and if the "gpu" modifier is enabled
1406+
'--datatier':'GEN-SIM-RAW',
1407+
'--eventcontent':'RAWSIM',
1408+
},
1409+
reco = {
1410+
# skip the @pixelTrackingOnlyValidation which cannot run together with the full reconstruction
1411+
'-s': 'RAW2DIGI:RawToDigi+RawToDigi_pixelOnly,L1Reco,RECO:reconstruction+reconstruction_pixelTrackingOnly,RECOSIM',
1412+
'--procModifiers': 'pixelNtupletFit',
1413+
'--customise' : 'RecoTracker/Configuration/customizePixelTracksForTriplets.customizePixelTracksForTriplets',
1414+
'--datatier':'AODSIM',
1415+
'--eventcontent':'AODSIM',
1416+
},
1417+
harvest = None,
1418+
suffix = 'Patatrack_FullRecoTripletsCPUProdLike',
1419+
offset = 0.59521,
1420+
)
13931421

13941422
# Workflow running the Pixel triplets, ECAL and HCAL reconstruction on GPU (optional), together with the full offline reconstruction on CPU
13951423
# - HLT on GPU (optional)
@@ -1412,6 +1440,26 @@ def setup_(self, step, stepName, stepDict, k, properties):
14121440
suffix = 'Patatrack_FullRecoTripletsGPU',
14131441
offset = 0.596,
14141442
)
1443+
# - ProdLike
1444+
upgradeWFs['PatatrackFullRecoTripletsGPUProdLike'] = PatatrackWorkflow(
1445+
digi = {
1446+
# the HLT menu is already set up for using GPUs if available and if the "gpu" modifier is enabled
1447+
'--procModifiers': 'gpu',
1448+
'--datatier':'GEN-SIM-RAW',
1449+
'--eventcontent':'RAWSIM',
1450+
},
1451+
reco = {
1452+
# skip the @pixelTrackingOnlyValidation which cannot run together with the full reconstruction
1453+
'-s': 'RAW2DIGI:RawToDigi+RawToDigi_pixelOnly,L1Reco,RECO:reconstruction+reconstruction_pixelTrackingOnly,RECOSIM',
1454+
'--procModifiers': 'pixelNtupletFit,gpu',
1455+
'--customise': 'RecoTracker/Configuration/customizePixelTracksForTriplets.customizePixelTracksForTriplets',
1456+
'--datatier':'AODSIM',
1457+
'--eventcontent':'AODSIM',
1458+
},
1459+
harvest = None,
1460+
suffix = 'Patatrack_FullRecoTripletsGPUProdLike',
1461+
offset = 0.59621,
1462+
)
14151463

14161464
# Workflow running the Pixel triplets, ECAL and HCAL reconstruction on CPU and GPU, together with the full offline reconstruction on CPU
14171465
# - HLT on GPU (required)
@@ -1437,7 +1485,6 @@ def setup_(self, step, stepName, stepDict, k, properties):
14371485
offset = 0.597,
14381486
)
14391487

1440-
14411488
# end of Patatrack workflows
14421489

14431490
class UpgradeWorkflow_ProdLike(UpgradeWorkflow):

0 commit comments

Comments
 (0)