@@ -420,6 +420,63 @@ def condition_(self, fragment, stepList, key, hasHarvest):
420
420
'--procModifiers' : 'gpu'
421
421
}
422
422
423
+ class UpgradeWorkflowPatatrack_HCALOnlyCPU (UpgradeWorkflowPatatrack ):
424
+ def setup_ (self , step , stepName , stepDict , k , properties ):
425
+ if 'Reco' in step :
426
+ stepDict [stepName ][k ] = merge ([self .step3 , stepDict [step ][k ]])
427
+ elif 'HARVEST' in step :
428
+ stepDict [stepName ][k ] = merge ([{'-s' : 'HARVESTING:@hcalOnlyValidation+@hcalOnly+@hcal2Only' }, stepDict [step ][k ]])
429
+
430
+ def condition_ (self , fragment , stepList , key , hasHarvest ):
431
+ return '2018' in key or '2021' in key
432
+
433
+ upgradeWFs ['PatatrackHCALOnlyCPU' ] = UpgradeWorkflowPatatrack_HCALOnlyCPU (
434
+ steps = [
435
+ 'RecoFull' ,
436
+ 'HARVESTFull' ,
437
+ 'RecoFullGlobal' ,
438
+ 'HARVESTFullGlobal' ,
439
+ ],
440
+ PU = [],
441
+ suffix = 'Patatrack_HCALOnlyCPU' ,
442
+ offset = 0.521 ,
443
+ )
444
+
445
+ upgradeWFs ['PatatrackHCALOnlyCPU' ].step3 = {
446
+ '-s' : 'RAW2DIGI:RawToDigi_hcalOnly,RECO:reconstruction_hcalOnly,VALIDATION:@hcalOnlyValidation,DQM:@hcalOnly+@hcal2Only' ,
447
+ '--datatier' : 'GEN-SIM-RECO,DQMIO' ,
448
+ '--eventcontent' : 'RECOSIM,DQM' ,
449
+ }
450
+
451
+ class UpgradeWorkflowPatatrack_HCALOnlyGPU (UpgradeWorkflowPatatrack ):
452
+ def setup_ (self , step , stepName , stepDict , k , properties ):
453
+ if 'Reco' in step :
454
+ stepDict [stepName ][k ] = merge ([self .step3 , stepDict [step ][k ]])
455
+ elif 'HARVEST' in step :
456
+ stepDict [stepName ][k ] = merge ([{'-s' : 'HARVESTING:@hcalOnlyValidation+@hcalOnly+@hcal2Only' }, stepDict [step ][k ]])
457
+
458
+ def condition_ (self , fragment , stepList , key , hasHarvest ):
459
+ return '2018' in key or '2021' in key
460
+
461
+ upgradeWFs ['PatatrackHCALOnlyGPU' ] = UpgradeWorkflowPatatrack_HCALOnlyGPU (
462
+ steps = [
463
+ 'RecoFull' ,
464
+ 'HARVESTFull' ,
465
+ 'RecoFullGlobal' ,
466
+ 'HARVESTFullGlobal' ,
467
+ ],
468
+ PU = [],
469
+ suffix = 'Patatrack_HCALOnlyGPU' ,
470
+ offset = 0.522 ,
471
+ )
472
+
473
+ upgradeWFs ['PatatrackHCALOnlyGPU' ].step3 = {
474
+ '-s' : 'RAW2DIGI:RawToDigi_hcalOnly,RECO:reconstruction_hcalOnly,VALIDATION:@hcalOnlyValidation,DQM:@hcalOnly+@hcal2Only' ,
475
+ '--datatier' : 'GEN-SIM-RECO,DQMIO' ,
476
+ '--eventcontent' : 'RECOSIM,DQM' ,
477
+ '--procModifiers' : 'gpu'
478
+ }
479
+
423
480
# end of Patatrack workflows
424
481
425
482
class UpgradeWorkflow_ProdLike (UpgradeWorkflow ):
0 commit comments