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

cima22 · cima22 · commit e20648ccd1da · 2025-10-16T23:00:45.000+02:00
diff --git a/GPU/GPUTracking/DataCompression/GPUTPCDecompression.cxx b/GPU/GPUTracking/DataCompression/GPUTPCDecompression.cxx
@@ -18,6 +18,7 @@
 #include "GPUO2DataTypes.h"
 #include "GPUMemorySizeScalers.h"
 #include "GPULogging.h"
+#include <algorithm>
 
 using namespace o2::gpu;
 
@@ -116,5 +117,8 @@ void GPUTPCDecompression::RegisterMemoryAllocation()
 
 void GPUTPCDecompression::SetMaxData(const GPUTrackingInOutPointers& io)
 {
-  mMaxNativeClustersPerBuffer = mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;
+  mMaxNativeClustersPerBuffer = *std::max_element(mInputGPU.nSliceRowClusters, mInputGPU.nSliceRowClusters + mInputGPU.nSliceRows);
+  float clsRatio = mInputGPU.nUnattachedClusters > 0 ? float(mInputGPU.nAttachedClusters) / float(mInputGPU.nUnattachedClusters) : 1.0f;
+  mMaxNativeClustersPerBuffer *= clsRatio;
+  mMaxNativeClustersPerBuffer = std::min(mMaxNativeClustersPerBuffer, mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow);
 }

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@`
`18`	`18`	`#include "GPUO2DataTypes.h"`
`19`	`19`	`#include "GPUMemorySizeScalers.h"`
`20`	`20`	`#include "GPULogging.h"`
	`21`	`+#include <algorithm>`
`21`	`22`
`22`	`23`	`using namespace o2::gpu;`
`23`	`24`
`@@ -116,5 +117,8 @@ void GPUTPCDecompression::RegisterMemoryAllocation()`
`116`	`117`
`117`	`118`	`void GPUTPCDecompression::SetMaxData(const GPUTrackingInOutPointers& io)`
`118`	`119`	`{`
`119`		`- mMaxNativeClustersPerBuffer = mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow;`
	`120`	`+ mMaxNativeClustersPerBuffer = *std::max_element(mInputGPU.nSliceRowClusters, mInputGPU.nSliceRowClusters + mInputGPU.nSliceRows);`
	`121`	`+ float clsRatio = mInputGPU.nUnattachedClusters > 0 ? float(mInputGPU.nAttachedClusters) / float(mInputGPU.nUnattachedClusters) : 1.0f;`
	`122`	`+ mMaxNativeClustersPerBuffer *= clsRatio;`
	`123`	`+ mMaxNativeClustersPerBuffer = std::min(mMaxNativeClustersPerBuffer, mRec->GetProcessingSettings().tpcMaxAttachedClustersPerSectorRow);`
`120`	`124`	`}`