Skip to content

Commit bce04bc

Browse files
committed
Changing to fixed size for number of clusters
1 parent 9893b43 commit bce04bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -893,7 +893,7 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
893893
clustererNN.nnClusterizerElementSize = ((2 * nn_settings.nnClusterizerSizeInputRow + 1) * (2 * nn_settings.nnClusterizerSizeInputPad + 1) * (2 * nn_settings.nnClusterizerSizeInputTime + 1)) + (nn_settings.nnClusterizerAddIndexData ? 3 : 0);
894894
clustererNN.nnClusterizerBatchedMode = nn_settings.nnClusterizerBatchedMode;
895895
clustererNN.nnClusterizerBoundaryFillValue = nn_settings.nnClusterizerBoundaryFillValue;
896-
clustererNN.nnClusterizerTotalClusters = clusterer.mPmemory->counters.nClusters;
896+
clustererNN.nnClusterizerTotalClusters = clusterer.mNMaxClusterPerRow;
897897
if (nn_settings.nnClusterizerVerbosity < 0) {
898898
clustererNN.nnClusterizerVerbosity = nn_settings.nnInferenceVerbosity;
899899
} else {

0 commit comments

Comments
 (0)