GPU: Consider per-row number of clusters for determining MaxRowStartHits

davidrohr · davidrohr · commit c18c482abc26 · 2022-07-29T23:54:55.000+02:00
diff --git a/GPU/GPUTracking/SliceTracker/GPUTPCTracker.cxx b/GPU/GPUTracking/SliceTracker/GPUTPCTracker.cxx
@@ -149,7 +149,17 @@ void GPUTPCTracker::SetMaxData(const GPUTrackingInOutPointers& io)
   } else {
     mNMaxStartHits = mRec->MemoryScalers()->NTPCStartHits(mData.NumberOfHits());
   }
-  mNMaxRowStartHits = mRec->MemoryScalers()->NTPCRowStartHits(mData.NumberOfHits());
+  if (io.clustersNative) {
+    unsigned int maxRowHits = 0;
+    for (unsigned int i = 0; i < GPUCA_ROW_COUNT; i++) {
+      if (io.clustersNative->nClusters[mISlice][i] > maxRowHits) {
+        maxRowHits = io.clustersNative->nClusters[mISlice][i];
+      }
+    }
+    mNMaxRowStartHits = mRec->MemoryScalers()->NTPCRowStartHits(maxRowHits * GPUCA_ROW_COUNT);
+  } else {
+    mNMaxRowStartHits = mRec->MemoryScalers()->NTPCRowStartHits(mData.NumberOfHits());
+  }
   mNMaxTracklets = mRec->MemoryScalers()->NTPCTracklets(mData.NumberOfHits());
   mNMaxRowHits = mRec->MemoryScalers()->NTPCTrackletHits(mData.NumberOfHits());
   mNMaxTracks = mRec->MemoryScalers()->NTPCSectorTracks(mData.NumberOfHits());
