Skip to content

Commit a545f08

Browse files
committed
Adding back the runParallelOuterLoop
1 parent 1d02b30 commit a545f08

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -664,7 +664,7 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
664664
nnTimers[11] = &getTimer<GPUTPCNNClusterizer, 11>("GPUTPCNNClusterizer_ONNXRegression2_2_", 11);
665665
}
666666

667-
for (int32_t lane = 0; lane < numLanes; lane++) {
667+
mRec->runParallelOuterLoop(doGPU, numLanes, [&](uint32_t lane) {
668668
nnApplications[lane].init(nn_settings, GetProcessingSettings().deterministicGPUReconstruction);
669669
if (nnApplications[lane].mModelsUsed[0]) {
670670
SetONNXGPUStream(*(nnApplications[lane].mModelClass).getSessionOptions(), lane, &deviceId);
@@ -708,7 +708,7 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
708708
if (nn_settings.nnClusterizerVerbosity > 0) {
709709
LOG(info) << "(ORT) Allocated ONNX stream for lane " << lane << " and device " << deviceId;
710710
}
711-
}
711+
});
712712
for (int32_t sector = 0; sector < NSECTORS; sector++) {
713713
GPUTPCNNClusterizer& clustererNN = processors()->tpcNNClusterer[sector];
714714
GPUTPCNNClusterizer& clustererNNShadow = doGPU ? processorsShadow()->tpcNNClusterer[sector] : clustererNN;

0 commit comments

Comments
 (0)