Skip to content

ITS: allow to reconstruct incomplete track topologies #15390

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
f3sch wants to merge 1 commit into
base: dev
Choose a base branch
from
Open

ITS: allow to reconstruct incomplete track topologies #15390

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 32 additions & 25 deletions Detectors/ITSMFT/ITS/tracking/GPU/ITStrackingGPU/TimeFrameGPU.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,11 @@ class TimeFrameGPU : public TimeFrame<NLayers>
using typename TimeFrame<NLayers>::ROFOverlapTableN;
using typename TimeFrame<NLayers>::ROFVertexLookupTableN;
using typename TimeFrame<NLayers>::ROFMaskTableN;
using typename TimeFrame<NLayers>::TrackingTopologyN;
using typename TimeFrame<NLayers>::TrackSeedN;
static constexpr int MaxTransitions = TrackingTopologyN::MaxTransitions;
static constexpr int MaxCells = TrackingTopologyN::MaxCells;
static constexpr int MaxStreams = MaxCells > NLayers ? MaxCells : NLayers;

public:
TimeFrameGPU() = default;
Expand All @@ -43,7 +47,9 @@ class TimeFrameGPU : public TimeFrame<NLayers>
void registerHostMemory(const int);
void unregisterHostMemory(const int);
void initialise(const TrackingParameters&, int maxLayers);
void initialise(const TrackingParameters&, int maxLayers, int iteration);
void loadIndexTableUtils();
void loadTrackingTopologies();
void loadTrackingFrameInfoDevice(const int);
void createTrackingFrameInfoDeviceArray();
void loadUnsortedClustersDevice(const int);
Expand Down Expand Up @@ -85,7 +91,7 @@ class TimeFrameGPU : public TimeFrame<NLayers>
void createNeighboursLUTDevice(const int, const unsigned int);
void createTrackITSExtDevice(const size_t);
void downloadTrackITSExtDevice();
void downloadCellsNeighboursDevice(std::vector<bounded_vector<std::pair<int, int>>>&, const int);
void downloadCellsNeighboursDevice(std::vector<bounded_vector<CellNeighbour>>&, const int);
void downloadNeighboursLUTDevice(bounded_vector<int>&, const int);
void downloadCellsDevice();
void downloadCellsLUTDevice();
Expand All @@ -109,6 +115,7 @@ class TimeFrameGPU : public TimeFrame<NLayers>
const auto getDeviceROFOverlapTableView() { return mDeviceROFOverlapTableView; }
const auto getDeviceROFVertexLookupTableView() { return mDeviceROFVertexLookupTableView; }
const auto getDeviceROFMaskTableView() { return mDeviceROFMaskTableView; }
const auto getDeviceTrackingTopologyView() const { return mDeviceTrackingTopologyView; }
int* getDeviceROFramesClusters(const int layer) { return mROFramesClustersDevice[layer]; }
auto& getTrackITSExt() { return mTrackITSExt; }
Vertex* getDeviceVertices() { return mPrimaryVerticesDevice; }
Expand All @@ -120,10 +127,9 @@ class TimeFrameGPU : public TimeFrame<NLayers>
TrackITSExt* getDeviceTrackITSExt() { return mTrackITSExtDevice; }
int* getDeviceNeighboursLUT(const int layer) { return mNeighboursLUTDevice[layer]; }
gsl::span<int*> getDeviceNeighboursLUTs() { return mNeighboursLUTDevice; }
gpuPair<int, int>* getDeviceNeighbourPairs(const int layer) { return mNeighbourPairsDevice[layer]; }
std::array<int*, NLayers - 2>& getDeviceNeighboursAll() { return mNeighboursDevice; }
int* getDeviceNeighbours(const int layer) { return mNeighboursDevice[layer]; }
int** getDeviceNeighboursArray() { return mNeighboursDevice.data(); }
CellNeighbour** getDeviceArrayNeighbours() { return mNeighboursDeviceArray; }
std::array<CellNeighbour*, MaxCells>& getDeviceNeighboursAll() { return mNeighboursDevice; }
CellNeighbour* getDeviceNeighbours(const int layer) { return mNeighboursDevice[layer]; }
TrackingFrameInfo* getDeviceTrackingFrameInfo(const int);
const TrackingFrameInfo** getDeviceArrayTrackingFrameInfo() const { return mTrackingFrameInfoDeviceArray; }
const Cluster** getDeviceArrayClusters() const { return mClustersDeviceArray; }
Expand All @@ -147,10 +153,10 @@ class TimeFrameGPU : public TimeFrame<NLayers>
void setDevicePropagator(const o2::base::PropagatorImpl<float>* p) final { this->mPropagatorDevice = p; }

// Host-specific getters
gsl::span<int, NLayers - 1> getNTracklets() { return mNTracklets; }
gsl::span<int, NLayers - 2> getNCells() { return mNCells; }
gsl::span<int> getNTracklets() { return {mNTracklets.data(), static_cast<gsl::span<int>::size_type>(this->mTrackingTopologyView.nTransitions)}; }
gsl::span<int> getNCells() { return {mNCells.data(), static_cast<gsl::span<int>::size_type>(this->mTrackingTopologyView.nCells)}; }
auto& getArrayNCells() { return mNCells; }
gsl::span<int, NLayers - 3> getNNeighbours() { return mNNeighbours; }
gsl::span<int> getNNeighbours() { return {mNNeighbours.data(), static_cast<gsl::span<int>::size_type>(this->mTrackingTopologyView.nCells)}; }
auto& getArrayNNeighbours() { return mNNeighbours; }

// Host-available device getters
Expand All @@ -169,16 +175,18 @@ class TimeFrameGPU : public TimeFrame<NLayers>
void allocMem(void**, size_t, bool, int32_t = o2::gpu::GPUMemoryResource::MEMORY_GPU); // Abstract owned and unowned memory allocations on default stream

// Host-available device buffer sizes
std::array<int, NLayers - 1> mNTracklets;
std::array<int, NLayers - 2> mNCells;
std::array<int, NLayers - 3> mNNeighbours;
std::array<int, MaxTransitions> mNTracklets{};
std::array<int, MaxCells> mNCells{};
std::array<int, MaxCells> mNNeighbours{};

// Device pointers
IndexTableUtilsN* mIndexTableUtilsDevice;
// device navigation views
ROFOverlapTableN::View mDeviceROFOverlapTableView;
ROFVertexLookupTableN::View mDeviceROFVertexLookupTableView;
ROFMaskTableN::View mDeviceROFMaskTableView;
std::vector<typename TrackingTopologyN::View> mDeviceTrackerTopologyViews;
typename TrackingTopologyN::View mDeviceTrackingTopologyView;

// Hybrid pref
Vertex* mPrimaryVerticesDevice;
Expand All @@ -193,30 +201,29 @@ class TimeFrameGPU : public TimeFrame<NLayers>
const int** mClustersIndexTablesDeviceArray;
uint8_t** mUsedClustersDeviceArray;
const int** mROFramesClustersDeviceArray;
std::array<Tracklet*, NLayers - 1> mTrackletsDevice;
std::array<int*, NLayers - 1> mTrackletsLUTDevice;
std::array<int*, NLayers - 2> mCellsLUTDevice;
std::array<int*, NLayers - 3> mNeighboursLUTDevice;
std::array<Tracklet*, MaxTransitions> mTrackletsDevice{};
std::array<int*, MaxTransitions> mTrackletsLUTDevice{};
std::array<int*, MaxCells> mCellsLUTDevice{};
std::array<int*, MaxCells> mNeighboursLUTDevice{};

Tracklet** mTrackletsDeviceArray{nullptr};
int** mCellsLUTDeviceArray{nullptr};
int** mNeighboursCellDeviceArray{nullptr};
int** mNeighboursCellLUTDeviceArray{nullptr};
int** mTrackletsLUTDeviceArray{nullptr};
std::array<CellSeed*, NLayers - 2> mCellsDevice;
std::array<CellSeed*, MaxCells> mCellsDevice{};
CellSeed** mCellsDeviceArray;
std::array<int*, NLayers - 3> mNeighboursIndexTablesDevice;
std::array<int*, MaxCells> mNeighboursIndexTablesDevice{};
TrackSeedN* mTrackSeedsDevice{nullptr};
int* mTrackSeedsLUTDevice{nullptr};
unsigned int mNTracks{0};
std::array<o2::track::TrackParCovF*, NLayers - 2> mCellSeedsDevice;
std::array<o2::track::TrackParCovF*, MaxCells> mCellSeedsDevice{};
o2::track::TrackParCovF** mCellSeedsDeviceArray;
std::array<float*, NLayers - 2> mCellSeedsChi2Device;
std::array<float*, MaxCells> mCellSeedsChi2Device{};
float** mCellSeedsChi2DeviceArray;

TrackITSExt* mTrackITSExtDevice;
std::array<gpuPair<int, int>*, NLayers - 2> mNeighbourPairsDevice;
std::array<int*, NLayers - 2> mNeighboursDevice;
std::array<CellNeighbour*, MaxCells> mNeighboursDevice{};
CellNeighbour** mNeighboursDeviceArray{nullptr};
std::array<TrackingFrameInfo*, NLayers> mTrackingFrameInfoDevice;
const TrackingFrameInfo** mTrackingFrameInfoDeviceArray;

Expand Down Expand Up @@ -245,19 +252,19 @@ inline std::vector<unsigned int> TimeFrameGPU<NLayers>::getClusterSizes()
template <int NLayers>
inline size_t TimeFrameGPU<NLayers>::getNumberOfTracklets() const
{
return std::accumulate(mNTracklets.begin(), mNTracklets.end(), 0);
return std::accumulate(mNTracklets.begin(), mNTracklets.begin() + this->mTrackingTopologyView.nTransitions, 0);
}

template <int NLayers>
inline size_t TimeFrameGPU<NLayers>::getNumberOfCells() const
{
return std::accumulate(mNCells.begin(), mNCells.end(), 0);
return std::accumulate(mNCells.begin(), mNCells.begin() + this->mTrackingTopologyView.nCells, 0);
}

template <int NLayers>
inline size_t TimeFrameGPU<NLayers>::getNumberOfNeighbours() const
{
return std::accumulate(mNNeighbours.begin(), mNNeighbours.end(), 0);
return std::accumulate(mNNeighbours.begin(), mNNeighbours.begin() + this->mTrackingTopologyView.nCells, 0);
}

} // namespace o2::its::gpu
Expand Down
71 changes: 41 additions & 30 deletions Detectors/ITSMFT/ITS/tracking/GPU/ITStrackingGPU/TrackingKernels.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,14 +17,14 @@

#include "ITStracking/BoundedAllocator.h"
#include "ITStracking/ROFLookupTables.h"
#include "ITStracking/Definitions.h"
#include "ITStracking/TrackingTopology.h"
#include "ITStrackingGPU/Utils.h"
#include "DetectorsBase/Propagator.h"
#include "GPUCommonDef.h"

namespace o2::its
{
class CellSeed;
struct CellNeighbour;
template <int>
class TrackSeed;
class TrackingFrameInfo;
Expand All @@ -38,7 +38,9 @@ class ExternalAllocator;
template <int NLayers>
void countTrackletsInROFsHandler(const IndexTableUtils<NLayers>* utils,
const typename ROFMaskTable<NLayers>::View& rofMask,
const int layer,
const int transitionId,
const int fromLayer,
const int toLayer,
const typename ROFOverlapTable<NLayers>::View& rofOverlaps,
const typename ROFVertexLookupTable<NLayers>::View& vertexLUT,
const int vertexId,
Expand All @@ -53,20 +55,23 @@ void countTrackletsInROFsHandler(const IndexTableUtils<NLayers>* utils,
gsl::span<int*> trackletsLUTsHost,
const bool selectUPCVertices,
const float NSigmaCut,
bounded_vector<float>& phiCuts,
const typename TrackingTopology<NLayers>::View topology,
bounded_vector<float>& transitionPhiCuts,
const float resolutionPV,
std::array<float, NLayers>& minR,
std::array<float, NLayers>& maxR,
bounded_vector<float>& resolutions,
std::vector<float>& radii,
bounded_vector<float>& mulScatAng,
bounded_vector<float>& transitionMSAngles,
o2::its::ExternalAllocator* alloc,
gpu::Streams& streams);

template <int NLayers>
void computeTrackletsInROFsHandler(const IndexTableUtils<NLayers>* utils,
const typename ROFMaskTable<NLayers>::View& rofMask,
const int layer,
const int transitionId,
const int fromLayer,
const int toLayer,
const typename ROFOverlapTable<NLayers>::View& rofOverlaps,
const typename ROFVertexLookupTable<NLayers>::View& vertexLUT,
const int vertexId,
Expand All @@ -84,13 +89,14 @@ void computeTrackletsInROFsHandler(const IndexTableUtils<NLayers>* utils,
gsl::span<int*> trackletsLUTsHost,
const bool selectUPCVertices,
const float NSigmaCut,
bounded_vector<float>& phiCuts,
const typename TrackingTopology<NLayers>::View topology,
bounded_vector<float>& transitionPhiCuts,
const float resolutionPV,
std::array<float, NLayers>& minR,
std::array<float, NLayers>& maxR,
bounded_vector<float>& resolutions,
std::vector<float>& radii,
bounded_vector<float>& mulScatAng,
bounded_vector<float>& transitionMSAngles,
o2::its::ExternalAllocator* alloc,
gpu::Streams& streams);

Expand All @@ -101,7 +107,8 @@ void countCellsHandler(const Cluster** sortedClusters,
Tracklet** tracklets,
int** trackletsLUT,
const int nTracklets,
const int layer,
const int cellTopologyId,
const typename TrackingTopology<NLayers>::View topology,
CellSeed* cells,
int** cellsLUTsDeviceArray,
int* cellsLUTsHost,
Expand All @@ -120,7 +127,8 @@ void computeCellsHandler(const Cluster** sortedClusters,
Tracklet** tracklets,
int** trackletsLUT,
const int nTracklets,
const int layer,
const int cellTopologyId,
const typename TrackingTopology<NLayers>::View topology,
CellSeed* cells,
int** cellsLUTsDeviceArray,
int* cellsLUTsHost,
Expand All @@ -133,33 +141,31 @@ void computeCellsHandler(const Cluster** sortedClusters,

template <int NLayers>
void countCellNeighboursHandler(CellSeed** cellsLayersDevice,
int* neighboursLUTs,
int* neighboursCursor,
int** cellsLUTs,
gpuPair<int, int>* cellNeighbours,
int* neighboursIndexTable,
const Tracklet** tracklets,
const int sourceCellTopologyId,
const int targetCellTopologyId,
const float maxChi2ClusterAttachment,
const float bz,
const int layerIndex,
const unsigned int nCells,
const unsigned int nCellsNext,
const int maxCellNeighbours,
o2::its::ExternalAllocator* alloc,
gpu::Stream& stream);

void scanCellNeighboursHandler(int* neighboursCursor,
int* neighboursLUT,
const unsigned int nCells,
o2::its::ExternalAllocator* alloc,
gpu::Stream& stream);

template <int NLayers>
void computeCellNeighboursHandler(CellSeed** cellsLayersDevice,
int* neighboursLUTs,
int* neighboursCursor,
int** cellsLUTs,
gpuPair<int, int>* cellNeighbours,
int* neighboursIndexTable,
const Tracklet** tracklets,
CellNeighbour* cellNeighbours,
const int sourceCellTopologyId,
const int targetCellTopologyId,
const float maxChi2ClusterAttachment,
const float bz,
const int layerIndex,
const unsigned int nCells,
const unsigned int nCellsNext,
const int maxCellNeighbours,
gpu::Stream& stream);

int filterCellNeighboursHandler(gpuPair<int, int>*,
Expand All @@ -169,19 +175,24 @@ int filterCellNeighboursHandler(gpuPair<int, int>*,
o2::its::ExternalAllocator* = nullptr);

template <int NLayers>
void processNeighboursHandler(const int startLayer,
const int startLevel,
void processNeighboursHandler(const int startLevel,
const int defaultCellTopologyId,
CellSeed** allCellSeeds,
CellSeed* currentCellSeeds,
std::array<int, NLayers - 2>& nCells,
const int* currentCellTopologyIds,
const int* currentCellIds,
const int* nCells,
const unsigned char** usedClusters,
std::array<int*, NLayers - 2>& neighbours,
gsl::span<int*> neighboursDeviceLUTs,
CellNeighbour** neighbours,
int** neighboursDeviceLUTs,
const TrackingFrameInfo** foundTrackingFrameInfo,
bounded_vector<TrackSeed<NLayers>>& seedsHost,
const float bz,
const float MaxChi2ClusterAttachment,
const float maxChi2NDF,
const int maxHoles,
const int minTrackLength,
const LayerMask holeLayerMask,
const std::vector<float>& layerxX0Host,
const o2::base::Propagator* propagator,
const o2::base::PropagatorF::MatCorrType matCorrType,
Expand Down
Loading