Skip to content

Commit efc1022

Browse files
committed
ITS: GPU: use fastATan2
Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>
1 parent a82d4bd commit efc1022

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

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

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,14 +29,13 @@
2929
#include "ITStracking/IndexTableUtils.h"
3030
#include "ITStracking/MathUtils.h"
3131
#include "ITStracking/ExternalAllocator.h"
32-
#include "DataFormatsITS/TrackITS.h"
33-
#include "ReconstructionDataFormats/Vertex.h"
34-
3532
#include "ITStrackingGPU/TrackerTraitsGPU.h"
3633
#include "ITStrackingGPU/TrackingKernels.h"
3734
#include "ITStrackingGPU/Utils.h"
3835

39-
// O2 track model
36+
#include "MathUtils/Utils.h"
37+
#include "DataFormatsITS/TrackITS.h"
38+
#include "ReconstructionDataFormats/Vertex.h"
4039
#include "ReconstructionDataFormats/Track.h"
4140
#include "DetectorsBase/Propagator.h"
4241
using namespace o2::track;
@@ -629,7 +628,7 @@ GPUg() void computeLayerTrackletsMultiROFKernel(
629628
if constexpr (initRun) {
630629
trackletsLUT[layerIndex][currentSortedIndex]++; // we need l0 as well for usual exclusive sums.
631630
} else {
632-
const float phi{o2::gpu::CAMath::ATan2(currentCluster.yCoordinate - nextCluster.yCoordinate, currentCluster.xCoordinate - nextCluster.xCoordinate)};
631+
const float phi{o2::math_utils::fastATan2(currentCluster.yCoordinate - nextCluster.yCoordinate, currentCluster.xCoordinate - nextCluster.xCoordinate)};
633632
const float tanL{(currentCluster.zCoordinate - nextCluster.zCoordinate) / (currentCluster.radius - nextCluster.radius)};
634633
const int nextSortedIndex{ROFClusters[layerIndex + 1][targetROF] + nextClusterIndex};
635634
new (tracklets[layerIndex] + trackletsLUT[layerIndex][currentSortedIndex] + storedTracklets) Tracklet{currentSortedIndex, nextSortedIndex, tanL, phi, pivotROF, targetROF};

0 commit comments

Comments
 (0)