GPU TPC: added dynamic buffer allocation during track-model decoding

cima22 · cima22 · commit e25a8fa61d1d · 2025-10-16T21:56:43.000+02:00
diff --git a/GPU/GPUTracking/DataCompression/GPUTPCDecompression.cxx b/GPU/GPUTracking/DataCompression/GPUTPCDecompression.cxx
@@ -116,5 +116,7 @@ void GPUTPCDecompression::RegisterMemoryAllocation()
 
 void GPUTPCDecompression::SetMaxData(const GPUTrackingInOutPointers& io)
 {
-  mMaxNativeClustersPerBuffer = mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;
+  uint32_t maxUnattachedClsPerSectorRow = std::max_element(io.tpcCompressedClusters->nSliceRowClusters, io.tpcCompressedClusters->nSliceRows);
+  float clsRatio = io.tpcCompressedClusters->nUnattachedClusters > 0 ? float(io.tpcCompressedClusters->nAttachedClusters) / float(io.tpcCompressedClusters->nUnattachedClusters) : 1.0f;
+  mMaxNativeClustersPerBuffer = std::min(maxUnattachedClsPerSectorRow * clsRatio, mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;)
 }

Original file line number	Diff line number	Diff line change
`@@ -116,5 +116,7 @@ void GPUTPCDecompression::RegisterMemoryAllocation()`
`116`	`116`
`117`	`117`	`void GPUTPCDecompression::SetMaxData(const GPUTrackingInOutPointers& io)`
`118`	`118`	`{`
`119`		`- mMaxNativeClustersPerBuffer = mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;`
	`119`	`+ uint32_t maxUnattachedClsPerSectorRow = std::max_element(io.tpcCompressedClusters->nSliceRowClusters, io.tpcCompressedClusters->nSliceRows);`
	`120`	`+ float clsRatio = io.tpcCompressedClusters->nUnattachedClusters > 0 ? float(io.tpcCompressedClusters->nAttachedClusters) / float(io.tpcCompressedClusters->nUnattachedClusters) : 1.0f;`
	`121`	`+ mMaxNativeClustersPerBuffer = std::min(maxUnattachedClsPerSectorRow * clsRatio, mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;)`
`120`	`122`	`}`