Skip to content

Commit f8562ed

Browse files
committed
TPC clusterizer reads CTP digits --> Respect in workflow dependency
1 parent 78e40d8 commit f8562ed

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
@@ -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

Comments
 (0)