@@ -1125,6 +1125,7 @@ def getDigiTaskName(det):
11251125 # reco
11261126 # -----------
11271127 tpcreconeeds = [FT0FV0EMCCTPDIGItask ['name' ]]
1128+ tpcclusterneed = [TPCDigitask ['name' ], FT0FV0EMCCTPDIGItask ['name' ]]
11281129 if not args .combine_tpc_clusterization :
11291130 # We treat TPC clusterization in multiple (sector) steps in order to
11301131 # stay within the memory limit or to parallelize over sector from outside (not yet supported within cluster algo)
@@ -1133,7 +1134,7 @@ def getDigiTaskName(det):
11331134 for s in range (0 ,35 ,sectorpertask ):
11341135 taskname = 'tpcclusterpart' + str ((int )(s / sectorpertask )) + '_' + str (tf )
11351136 tpcclustertasks .append (taskname )
1136- tpcclussect = createTask (name = taskname , needs = [ TPCDigitask [ 'name' ]] , tf = tf , cwd = timeframeworkdir , lab = ["RECO" ], cpu = '2' , mem = '8000' )
1137+ tpcclussect = createTask (name = taskname , needs = tpcclusterneed , tf = tf , cwd = timeframeworkdir , lab = ["RECO" ], cpu = '2' , mem = '8000' )
11371138 digitmergerstr = '${O2_ROOT}/bin/o2-tpc-chunkeddigit-merger --tpc-sectors ' + str (s )+ '-' + str (s + sectorpertask - 1 ) + ' --tpc-lanes ' + str (NWORKERS_TF ) + ' | '
11381139 tpcclussect ['cmd' ] = (digitmergerstr ,'' )[args .no_tpc_digitchunking ] + ' ${O2_ROOT}/bin/o2-tpc-reco-workflow ' + getDPL_global_options (bigshm = True ) + ' --input-type ' + ('digitizer' ,'digits' )[args .no_tpc_digitchunking ] + ' --output-type clusters,send-clusters-per-sector --tpc-native-cluster-writer \" --outfile tpc-native-clusters-part' + str ((int )(s / sectorpertask )) + '.root\" --tpc-sectors ' + str (s )+ '-' + str (s + sectorpertask - 1 ) + ' ' + putConfigValues (["GPU_global" ], {"GPU_proc.ompThreads" : 4 }) + ('' ,' --disable-mc' )[args .no_mc_labels ]
11391140 tpcclussect ['env' ] = { "OMP_NUM_THREADS" : "4" }
@@ -1146,7 +1147,7 @@ def getDigiTaskName(det):
11461147 workflow ['stages' ].append (TPCCLUSMERGEtask )
11471148 tpcreconeeds .append (TPCCLUSMERGEtask ['name' ])
11481149 else :
1149- tpcclus = createTask (name = 'tpccluster_' + str (tf ), needs = [ TPCDigitask [ 'name' ]] , tf = tf , cwd = timeframeworkdir , lab = ["RECO" ], cpu = NWORKERS_TF , mem = '2000' )
1150+ tpcclus = createTask (name = 'tpccluster_' + str (tf ), needs = tpcclusterneed , tf = tf , cwd = timeframeworkdir , lab = ["RECO" ], cpu = NWORKERS_TF , mem = '2000' )
11501151 tpcclus ['cmd' ] = '${O2_ROOT}/bin/o2-tpc-chunkeddigit-merger --tpc-lanes ' + str (NWORKERS_TF )
11511152 tpcclus ['cmd' ] += ' | ${O2_ROOT}/bin/o2-tpc-reco-workflow ' + getDPL_global_options () + ' --input-type digitizer --output-type clusters,send-clusters-per-sector ' + putConfigValues (["GPU_global" ,"TPCGasParam" ,"TPCCorrMap" ],{"GPU_proc.ompThreads" : 1 }) + ('' ,' --disable-mc' )[args .no_mc_labels ]
11521153 workflow ['stages' ].append (tpcclus )
0 commit comments