Skip to content

Commit 77390f9

Browse files
committed
Fix nCells printout
1 parent a1efad8 commit 77390f9

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

Detectors/ITSMFT/ITS/tracking/GPU/ITStrackingGPU/TimeFrameGPU.h

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,9 @@ class TimeFrameGPU : public TimeFrame
152152
gsl::span<Tracklet*> getDeviceTracklet() { return mTrackletsDevice; }
153153
gsl::span<CellSeed*> getDeviceCells() { return mCellsDevice; }
154154

155+
// Overridden getters
156+
int getNumberOfCells() const;
157+
155158
private:
156159
void allocMemAsync(void**, size_t, Stream*, bool); // Abstract owned and unowned memory allocations
157160
bool mHostRegistered = false;
@@ -253,6 +256,12 @@ inline std::vector<unsigned int> TimeFrameGPU<nLayers>::getClusterSizes()
253256
return sizes;
254257
}
255258

259+
template <int nLayers>
260+
inline int TimeFrameGPU<nLayers>::getNumberOfCells() const
261+
{
262+
return std::accumulate(mNCells.begin(), mNCells.end(), 0);
263+
}
264+
256265
} // namespace gpu
257266
} // namespace its
258267
} // namespace o2

0 commit comments

Comments
 (0)