Skip to content

Commit 2660534

Browse files
committed
test
1 parent f57e25e commit 2660534

6 files changed

Lines changed: 15 additions & 17 deletions

File tree

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ template <int nLayers>
469469
void TimeFrameGPU<nLayers>::createTrackITSExtDevice(bounded_vector<CellSeed>& seeds)
470470
{
471471
START_GPU_STREAM_TIMER(mGpuStreams[0]->get(), "reserving tracks");
472-
mTrackITSExt = bounded_vector<TrackITSExt>(seeds.size(), this->getMemoryPool().get());
472+
mTrackITSExt = bounded_vector<TrackITSExt>(seeds.size(), {}, this->getMemoryPool().get());
473473
LOGP(debug, "gpu-allocation: reserving {} tracks, for {} MB.", seeds.size(), seeds.size() * sizeof(o2::its::TrackITSExt) / MB);
474474
allocMemAsync(reinterpret_cast<void**>(&mTrackITSExtDevice), seeds.size() * sizeof(o2::its::TrackITSExt), mGpuStreams[0], this->getExtAllocator());
475475
GPUChkErrS(cudaMemsetAsync(mTrackITSExtDevice, 0, seeds.size() * sizeof(o2::its::TrackITSExt), mGpuStreams[0]->get()));

Detectors/ITSMFT/ITS/tracking/include/ITStracking/TimeFrame.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,7 @@ struct TimeFrame {
271271
int mNrof = 0;
272272
int mNExtendedTracks{0};
273273
int mNExtendedUsedClusters{0};
274-
bounded_vector<int> mROFramesPV = {0};
274+
bounded_vector<int> mROFramesPV;
275275
bounded_vector<Vertex> mPrimaryVertices;
276276

277277
// State if memory will be externally managed.

Detectors/ITSMFT/ITS/tracking/include/ITStracking/VertexerTraits.h

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,6 @@
3232
#include "GPUCommonDef.h"
3333
#include "GPUCommonMath.h"
3434

35-
#include <oneapi/tbb.h>
36-
#include <oneapi/tbb/partitioner.h>
3735
#include <oneapi/tbb/task_arena.h>
3836

3937
namespace o2
@@ -129,7 +127,6 @@ class VertexerTraits
129127
private:
130128
std::shared_ptr<BoundedMemoryResource> mMemoryPool;
131129
tbb::task_arena mTaskArena;
132-
tbb::affinity_partitioner mPartitioner;
133130
};
134131

135132
inline void VertexerTraits::initialise(const TrackingParameters& trackingParams, const int iteration)

Detectors/ITSMFT/ITS/tracking/src/TimeFrame.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,7 @@ void TimeFrame<nLayers>::prepareClusters(const TrackingParameters& trkParam, con
267267
h.bin = bin;
268268
h.ind = clsPerBin[bin]++;
269269
}
270-
bounded_vector<int> lutPerBin(clsPerBin.size(), mMemoryPool.get());
270+
bounded_vector<int> lutPerBin(clsPerBin.size(), 0, mMemoryPool.get());
271271
lutPerBin[0] = 0;
272272
for (unsigned int iB{1}; iB < lutPerBin.size(); ++iB) {
273273
lutPerBin[iB] = lutPerBin[iB - 1] + clsPerBin[iB - 1];

Detectors/ITSMFT/ITS/tracking/src/Tracker.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ void Tracker::clustersToTracks(LogFunc logger, LogFunc error)
124124
}
125125
if (mTraits->supportsFindShortPrimaries() && mTrkParams[0].FindShortTracks) {
126126
auto nTracksB = mTimeFrame->getNumberOfTracks();
127-
total += evaluateTask(&Tracker::findShortPrimaries, "Short primaries finding", iteration, logger);
127+
total += evaluateTask(&Tracker::findShortPrimaries, "Short primaries finding", 0, logger);
128128
auto nTracksA = mTimeFrame->getNumberOfTracks();
129129
logger(std::format(" `-> found {} additional tracks", nTracksA - nTracksB));
130130
}

Detectors/ITSMFT/ITS/tracking/src/VertexerTraits.cxx

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@
1414
#include <string>
1515
#include <chrono>
1616

17+
#include <oneapi/tbb/blocked_range.h>
18+
#include <oneapi/tbb/parallel_for.h>
19+
1720
#include "ITStracking/VertexerTraits.h"
1821
#include "ITStracking/BoundedAllocator.h"
1922
#include "ITStracking/ClusterLines.h"
@@ -180,8 +183,8 @@ void VertexerTraits::computeTracklets(const int iteration)
180183
mTaskArena.execute([&] {
181184
tbb::parallel_for(
182185
tbb::blocked_range<short>(0, (short)mTimeFrame->getNrof()),
183-
[&](const tbb::blocked_range<short>& rofs) {
184-
for (short pivotRofId = rofs.begin(); pivotRofId < rofs.end(); ++pivotRofId) {
186+
[&](const tbb::blocked_range<short>& Rofs) {
187+
for (short pivotRofId = Rofs.begin(); pivotRofId < Rofs.end(); ++pivotRofId) {
185188
bool skipROF = iteration && (int)mTimeFrame->getPrimaryVertices(pivotRofId).size() > mVrtParams[iteration].vertPerRofThreshold;
186189
short startROF{std::max((short)0, static_cast<short>(pivotRofId - mVrtParams[iteration].deltaRof))};
187190
short endROF{std::min(static_cast<short>(mTimeFrame->getNrof()), static_cast<short>(pivotRofId + mVrtParams[iteration].deltaRof + 1))};
@@ -216,8 +219,7 @@ void VertexerTraits::computeTracklets(const int iteration)
216219
mTimeFrame->getNTrackletsROF(pivotRofId, 0) = std::accumulate(mTimeFrame->getNTrackletsCluster(pivotRofId, 0).begin(), mTimeFrame->getNTrackletsCluster(pivotRofId, 0).end(), 0);
217220
mTimeFrame->getNTrackletsROF(pivotRofId, 1) = std::accumulate(mTimeFrame->getNTrackletsCluster(pivotRofId, 1).begin(), mTimeFrame->getNTrackletsCluster(pivotRofId, 1).end(), 0);
218221
}
219-
},
220-
mPartitioner);
222+
});
221223
});
222224

223225
mTimeFrame->computeTrackletsPerROFScans();
@@ -227,8 +229,8 @@ void VertexerTraits::computeTracklets(const int iteration)
227229
mTaskArena.execute([&] {
228230
tbb::parallel_for(
229231
tbb::blocked_range<short>(0, (short)mTimeFrame->getNrof()),
230-
[&](const tbb::blocked_range<short>& rofs) {
231-
for (short pivotRofId = rofs.begin(); pivotRofId < rofs.end(); ++pivotRofId) {
232+
[&](const tbb::blocked_range<short>& Rofs) {
233+
for (short pivotRofId = Rofs.begin(); pivotRofId < Rofs.end(); ++pivotRofId) {
232234
bool skipROF = iteration && (int)mTimeFrame->getPrimaryVertices(pivotRofId).size() > mVrtParams[iteration].vertPerRofThreshold;
233235
short startROF{std::max((short)0, static_cast<short>(pivotRofId - mVrtParams[iteration].deltaRof))};
234236
short endROF{std::min(static_cast<short>(mTimeFrame->getNrof()), static_cast<short>(pivotRofId + mVrtParams[iteration].deltaRof + 1))};
@@ -263,8 +265,7 @@ void VertexerTraits::computeTracklets(const int iteration)
263265
mVrtParams[iteration].maxTrackletsPerCluster);
264266
}
265267
}
266-
},
267-
mPartitioner);
268+
});
268269
});
269270

270271
/// Create tracklets labels for L0-L1, information is as flat as in tracklets vector (no rofId)
@@ -337,8 +338,8 @@ void VertexerTraits::computeTrackletMatching(const int iteration)
337338
mTaskArena.execute([&] {
338339
tbb::parallel_for(
339340
tbb::blocked_range<short>(0, (short)mTimeFrame->getNrof()),
340-
[&](const tbb::blocked_range<short>& rofs) {
341-
for (short pivotRofId = rofs.begin(); pivotRofId < rofs.end(); ++pivotRofId) {
341+
[&](const tbb::blocked_range<short>& Rofs) {
342+
for (short pivotRofId = Rofs.begin(); pivotRofId < Rofs.end(); ++pivotRofId) {
342343
if (iteration && (int)mTimeFrame->getPrimaryVertices(pivotRofId).size() > mVrtParams[iteration].vertPerRofThreshold) {
343344
continue;
344345
}

0 commit comments

Comments
 (0)