@@ -624,13 +624,9 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
624624 GPUTPCNNClusterizerHost nnApplications[GetProcessingSettings ().nTPCClustererLanes ];
625625
626626 if (GetProcessingSettings ().nn .applyNNclusterizer ) {
627- uint32_t maxClusters = 0 ;
628627 int32_t deviceId = -1 ;
629628 int32_t numLanes = GetProcessingSettings ().nTPCClustererLanes ;
630629 int32_t maxThreads = mRec ->MemoryScalers ()->nTPCdigits / 6000 ;
631- for (uint32_t lane = 0 ; lane < NSECTORS; lane++) {
632- maxClusters = std::max (maxClusters, processors ()->tpcClusterer [lane].mNMaxClusters );
633- }
634630 mRec ->runParallelOuterLoop (doGPU, numLanes, [&](uint32_t lane) {
635631 nnApplications[lane].init (nn_settings);
636632 if (nnApplications[lane].modelsUsed [0 ]) {
@@ -667,12 +663,12 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
667663 int32_t lane = sector % numLanes;
668664 clustererNN.deviceId = deviceId;
669665 clustererNN.mISector = sector;
670- clustererNN.nnClusterizerTotalClusters = maxClusters ;
666+ clustererNN.nnClusterizerTotalClusters = processors ()-> tpcClusterer [lane]. mNMaxClusters ;
671667 nnApplications[lane].initClusterizer (nn_settings, clustererNN);
672668 if (doGPU){
673669 clustererNNShadow.deviceId = deviceId;
674670 clustererNNShadow.mISector = sector;
675- clustererNNShadow.nnClusterizerTotalClusters = maxClusters ;
671+ clustererNNShadow.nnClusterizerTotalClusters = processors ()-> tpcClusterer [lane]. mNMaxClusters ;
676672 nnApplications[lane].initClusterizer (nn_settings, clustererNNShadow);
677673 }
678674 AllocateRegisteredMemory (clustererNN.mMemoryId );
@@ -1034,7 +1030,7 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
10341030 for (size_t i = 0 ; i < clusterer.mPmemory ->counters .nClusters ; ++i) {
10351031 acceptedClusters += clustererNNShadow.outputDataClass [i];
10361032 }
1037- LOG (info) << " [NN CF] Apply NN (fragment " << fragment.index << " , lane: " << lane << " , sector: " << iSector << " ): filling data " << time_fill << " s ; clusterizer: " << time_clusterizer << " s ; " << clusterer.mPmemory ->counters .nClusters << " clusters, " << acceptedClusters << " accepted. --> " << clusterer.mPmemory ->counters .nClusters / (time_fill + time_clusterizer) << " clusters/s" ;
1033+ LOG (info) << " [NN CF] Apply NN (fragment " << fragment.index << " , lane: " << lane << " , sector: " << iSector << " ): filling data " << time_fill << " s ; clusterizer: " << time_clusterizer << " s ; " << clusterer.mPmemory ->counters .nClusters << " clusters, " << acceptedClusters << " accepted. --> " << ( int32_t ) clusterer.mPmemory ->counters .nClusters / (time_fill + time_clusterizer) << " clusters/s" ;
10381034 }
10391035#else
10401036 GPUFatal (" Project not compiled with neural network clusterization. Aborting." );
0 commit comments