Skip to content

Commit b437e38

Browse files
committed
Minor changes
1 parent fff6dc3 commit b437e38

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)