Skip to content

Commit b6a30ac

Browse files
committed
TPC clusterizer reads CTP digits --> Respect in workflow dependency
1 parent 4661dcd commit b6a30ac

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

MC/bin/o2dpg_sim_workflow.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1117,6 +1117,7 @@ def getDigiTaskName(det):
11171117
# reco
11181118
# -----------
11191119
tpcreconeeds=[FT0FV0EMCCTPDIGItask['name']]
1120+
tpcclusterneed=[TPCDigitask['name'], FT0FV0EMCCTPDIGItask['name']]
11201121
if not args.combine_tpc_clusterization:
11211122
# We treat TPC clusterization in multiple (sector) steps in order to
11221123
# stay within the memory limit or to parallelize over sector from outside (not yet supported within cluster algo)
@@ -1125,7 +1126,7 @@ def getDigiTaskName(det):
11251126
for s in range(0,35,sectorpertask):
11261127
taskname = 'tpcclusterpart' + str((int)(s/sectorpertask)) + '_' + str(tf)
11271128
tpcclustertasks.append(taskname)
1128-
tpcclussect = createTask(name=taskname, needs=[TPCDigitask['name']], tf=tf, cwd=timeframeworkdir, lab=["RECO"], cpu='2', mem='8000')
1129+
tpcclussect = createTask(name=taskname, needs=tpcclusterneed, tf=tf, cwd=timeframeworkdir, lab=["RECO"], cpu='2', mem='8000')
11291130
digitmergerstr = '${O2_ROOT}/bin/o2-tpc-chunkeddigit-merger --tpc-sectors ' + str(s)+'-'+str(s+sectorpertask-1) + ' --tpc-lanes ' + str(NWORKERS_TF) + ' | '
11301131
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]
11311132
tpcclussect['env'] = { "OMP_NUM_THREADS" : "4" }
@@ -1138,7 +1139,7 @@ def getDigiTaskName(det):
11381139
workflow['stages'].append(TPCCLUSMERGEtask)
11391140
tpcreconeeds.append(TPCCLUSMERGEtask['name'])
11401141
else:
1141-
tpcclus = createTask(name='tpccluster_' + str(tf), needs=[TPCDigitask['name']], tf=tf, cwd=timeframeworkdir, lab=["RECO"], cpu=NWORKERS_TF, mem='2000')
1142+
tpcclus = createTask(name='tpccluster_' + str(tf), needs=tpcclusterneed, tf=tf, cwd=timeframeworkdir, lab=["RECO"], cpu=NWORKERS_TF, mem='2000')
11421143
tpcclus['cmd'] = '${O2_ROOT}/bin/o2-tpc-chunkeddigit-merger --tpc-lanes ' + str(NWORKERS_TF)
11431144
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]
11441145
workflow['stages'].append(tpcclus)

0 commit comments

Comments
 (0)