File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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;
You can’t perform that action at this time.
0 commit comments