@@ -524,16 +524,16 @@ def condition_(self, fragment, stepList, key, hasHarvest):
524524 '--procModifiers' : 'trackingMkFitCommon,trackingMkFitInitialStep'
525525}
526526
527- # LST on CPU, initialStep+highPtTripletStep-only tracking-only
528- class UpgradeWorkflow_lstOnCPUIters01TrackingOnly (UpgradeWorkflowTracking ):
527+ # LST and mkFit on CPU, initialStep+highPtTripletStep-only tracking-only
528+ class UpgradeWorkflow_lstmkFitOnCPUIters01TrackingOnly (UpgradeWorkflowTracking ):
529529 def setup__ (self , step , stepName , stepDict , k , properties ):
530530 if 'Reco' in step : stepDict [stepName ][k ] = merge ([self .step3 , stepDict [step ][k ]])
531531 elif 'HARVEST' in step : stepDict [stepName ][k ] = merge ([{'-s' : 'HARVESTING:@trackingOnlyValidation+@trackingOnlyDQM' }, stepDict [step ][k ]])
532532 elif 'ALCA' in step : stepDict [stepName ][k ] = None
533533 def condition (self , fragment , stepList , key , hasHarvest ):
534534 result = (fragment == "TTbar_14TeV" ) and hasHarvest and ('Run4' in key )
535535 return result
536- upgradeWFs ['lstOnCPUIters01TrackingOnly ' ] = UpgradeWorkflow_lstOnCPUIters01TrackingOnly (
536+ upgradeWFs ['lstmkFitOnCPUIters01TrackingOnly ' ] = UpgradeWorkflow_lstmkFitOnCPUIters01TrackingOnly (
537537 steps = [
538538 'RecoGlobal' ,
539539 'HARVESTGlobal' ,
@@ -545,24 +545,24 @@ def condition(self, fragment, stepList, key, hasHarvest):
545545 'RecoGlobal' ,
546546 'HARVESTGlobal' ,
547547 ],
548- suffix = '_lstOnCPUIters01TrackingOnly ' ,
548+ suffix = '_lstmkFitOnCPUIters01TrackingOnly ' ,
549549 offset = 0.711 ,
550550)
551- upgradeWFs ['lstOnCPUIters01TrackingOnly ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
552- '--procModifiers' : 'trackingIters01, trackingLST' ,
551+ upgradeWFs ['lstmkFitOnCPUIters01TrackingOnly ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
552+ '--procModifiers' : 'trackingLST,seedingLST,trackingMkFitHighPtTripletStep,trackingIters01 ' ,
553553 '--accelerators' : 'cpu'
554554}
555555
556- # LST on GPU (if available), initialStep+highPtTripletStep-only tracking-only
557- class UpgradeWorkflow_lstOnGPUIters01TrackingOnly (UpgradeWorkflowTracking ):
556+ # LST and mkFit on GPU (if available), initialStep+highPtTripletStep-only tracking-only
557+ class UpgradeWorkflow_lstmkFitOnGPUIters01TrackingOnly (UpgradeWorkflowTracking ):
558558 def setup__ (self , step , stepName , stepDict , k , properties ):
559559 if 'Reco' in step : stepDict [stepName ][k ] = merge ([self .step3 , stepDict [step ][k ]])
560- elif 'HARVEST' in step : stepDict [stepName ][k ] = merge ([{'-s' : 'HARVESTING:@trackingOnlyValidation+@trackingOnlyDQM' , '--procModifiers' : 'trackingIters01, trackingLST' }, stepDict [step ][k ]])
560+ elif 'HARVEST' in step : stepDict [stepName ][k ] = merge ([{'-s' : 'HARVESTING:@trackingOnlyValidation+@trackingOnlyDQM' , '--procModifiers' : 'trackingLST,seedingLST,trackingMkFitHighPtTripletStep,trackingIters01 ' }, stepDict [step ][k ]])
561561 elif 'ALCA' in step : stepDict [stepName ][k ] = None
562562 def condition (self , fragment , stepList , key , hasHarvest ):
563563 result = (fragment == "TTbar_14TeV" ) and hasHarvest and ('Run4' in key )
564564 return result
565- upgradeWFs ['lstOnGPUIters01TrackingOnly ' ] = UpgradeWorkflow_lstOnGPUIters01TrackingOnly (
565+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnly ' ] = UpgradeWorkflow_lstmkFitOnGPUIters01TrackingOnly (
566566 steps = [
567567 'RecoGlobal' ,
568568 'HARVESTGlobal' ,
@@ -574,21 +574,21 @@ def condition(self, fragment, stepList, key, hasHarvest):
574574 'RecoGlobal' ,
575575 'HARVESTGlobal' ,
576576 ],
577- suffix = '_lstOnGPUIters01TrackingOnly ' ,
577+ suffix = '_lstmkFitOnGPUIters01TrackingOnly ' ,
578578 offset = 0.712 ,
579579)
580- upgradeWFs ['lstOnGPUIters01TrackingOnly ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
581- '--procModifiers' : 'trackingIters01, trackingLST' ,
580+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnly ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
581+ '--procModifiers' : 'trackingLST,seedingLST,trackingMkFitHighPtTripletStep,trackingIters01'
582582}
583583
584584# LST on GPU (if available), initialStep+highPtTripletStep-only tracking-only, CPU vs. GPU comparison
585- class UpgradeWorkflow_lstOnGPUIters01TrackingOnlyAlpakaValidationLST ( UpgradeWorkflow_lstOnGPUIters01TrackingOnly ):
585+ class UpgradeWorkflow_lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST ( UpgradeWorkflow_lstmkFitOnGPUIters01TrackingOnly ):
586586 pass
587- upgradeWFs ['lstOnGPUIters01TrackingOnlyAlpakaValidationLST ' ] = deepcopy (upgradeWFs ['lstOnGPUIters01TrackingOnly ' ])
588- upgradeWFs ['lstOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].suffix = '_lstOnGPUIters01TrackingOnlyAlpakaValidationLST '
589- upgradeWFs ['lstOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].offset = 0.713
590- upgradeWFs ['lstOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
591- '--procModifiers' : 'alpakaValidationLST,trackingIters01, trackingLST' ,
587+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST ' ] = deepcopy (upgradeWFs ['lstmkFitOnGPUIters01TrackingOnly ' ])
588+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].suffix = '_lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST '
589+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].offset = 0.713
590+ upgradeWFs ['lstmkFitOnGPUIters01TrackingOnlyAlpakaValidationLST ' ].step3 = upgradeWFs ['trackingOnly' ].step3 | {
591+ '--procModifiers' : 'alpakaValidationLST,trackingLST,seedingLST,trackingMkFitHighPtTripletStep,trackingIters01'
592592}
593593
594594#DeepCore seeding for JetCore iteration workflow
0 commit comments