Skip to content

Commit 19803f6

Browse files
f3schshahor02
authored andcommitted
ITS: GPU: add minimal version of clearing
Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>
1 parent d00ca87 commit 19803f6

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

Detectors/ITSMFT/ITS/tracking/GPU/cuda/TrackingKernels.cu

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@
3232
#include "ITStracking/Cluster.h"
3333
#include "ITStracking/Cell.h"
3434
#include "DataFormatsITS/TrackITS.h"
35-
3635
#include "ITStrackingGPU/TrackingKernels.h"
3736
#include "ITStrackingGPU/Utils.h"
37+
#include "utils/strtag.h"
3838

3939
// O2 track model
4040
#include "ReconstructionDataFormats/Track.h"
@@ -1106,6 +1106,8 @@ void processNeighboursHandler(const int startLayer,
11061106
const int nBlocks,
11071107
const int nThreads)
11081108
{
1109+
constexpr uint64_t Tag = qStr2Tag("ITS_PNH1");
1110+
alloc->pushTagOnStack(Tag);
11091111
auto allocInt = gpu::TypedAllocator<int>(alloc);
11101112
auto allocCellSeed = gpu::TypedAllocator<CellSeed<nLayers>>(alloc);
11111113
thrust::device_vector<int, gpu::TypedAllocator<int>> foundSeedsTable(nCells[startLayer] + 1, 0, allocInt);
@@ -1216,6 +1218,7 @@ void processNeighboursHandler(const int startLayer,
12161218
auto s{end - outSeeds.begin()};
12171219
seedsHost.reserve(seedsHost.size() + s);
12181220
thrust::copy(outSeeds.begin(), outSeeds.begin() + s, std::back_inserter(seedsHost));
1221+
alloc->popTagOffStack(Tag);
12191222
}
12201223

12211224
template <int nLayers>

0 commit comments

Comments
 (0)