|
1 | 1 | #ifndef RecoTracker_LSTCore_src_alpaka_NeuralNetwork_h |
2 | 2 | #define RecoTracker_LSTCore_src_alpaka_NeuralNetwork_h |
3 | 3 |
|
4 | | -#include "DataFormats/Math/interface/deltaPhi.h" |
5 | 4 | #include "FWCore/Utilities/interface/CMSUnrollLoop.h" |
| 5 | +#include "HeterogeneousCore/AlpakaMath/interface/deltaPhi.h" |
6 | 6 |
|
7 | 7 | #include "RecoTracker/LSTCore/interface/alpaka/Common.h" |
8 | 8 | #include "RecoTracker/LSTCore/interface/MiniDoubletsSoA.h" |
@@ -85,25 +85,25 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE::lst::t5dnn { |
85 | 85 | z1 / kZ_max, // inner T3: First hit z normalized |
86 | 86 | r1 / kR_max, // inner T3: First hit r normalized |
87 | 87 |
|
88 | | - eta2 - eta1, // inner T3: Difference in eta between hit 2 and 1 |
89 | | - reco::deltaPhi(phi2, phi1) / kPhi_norm, // inner T3: Difference in phi between hit 2 and 1 |
90 | | - (z2 - z1) / kZ_max, // inner T3: Difference in z between hit 2 and 1 normalized |
91 | | - (r2 - r1) / kR_max, // inner T3: Difference in r between hit 2 and 1 normalized |
92 | | - |
93 | | - eta3 - eta2, // inner T3: Difference in eta between hit 3 and 2 |
94 | | - reco::deltaPhi(phi3, phi2) / kPhi_norm, // inner T3: Difference in phi between hit 3 and 2 |
95 | | - (z3 - z2) / kZ_max, // inner T3: Difference in z between hit 3 and 2 normalized |
96 | | - (r3 - r2) / kR_max, // inner T3: Difference in r between hit 3 and 2 normalized |
97 | | - |
98 | | - eta4 - eta3, // outer T3: Difference in eta between hit 4 and 3 |
99 | | - reco::deltaPhi(phi4, phi3) / kPhi_norm, // inner T3: Difference in phi between hit 4 and 3 |
100 | | - (z4 - z3) / kZ_max, // outer T3: Difference in z between hit 4 and 3 normalized |
101 | | - (r4 - r3) / kR_max, // outer T3: Difference in r between hit 4 and 3 normalized |
102 | | - |
103 | | - eta5 - eta4, // outer T3: Difference in eta between hit 5 and 4 |
104 | | - reco::deltaPhi(phi5, phi4) / kPhi_norm, // inner T3: Difference in phi between hit 5 and 4 |
105 | | - (z5 - z4) / kZ_max, // outer T3: Difference in z between hit 5 and 4 normalized |
106 | | - (r5 - r4) / kR_max, // outer T3: Difference in r between hit 5 and 4 normalized |
| 88 | + eta2 - eta1, // inner T3: Difference in eta between hit 2 and 1 |
| 89 | + cms::alpakatools::deltaPhi(acc, phi2, phi1) / kPhi_norm, // inner T3: Difference in phi between hit 2 and 1 |
| 90 | + (z2 - z1) / kZ_max, // inner T3: Difference in z between hit 2 and 1 normalized |
| 91 | + (r2 - r1) / kR_max, // inner T3: Difference in r between hit 2 and 1 normalized |
| 92 | + |
| 93 | + eta3 - eta2, // inner T3: Difference in eta between hit 3 and 2 |
| 94 | + cms::alpakatools::deltaPhi(acc, phi3, phi2) / kPhi_norm, // inner T3: Difference in phi between hit 3 and 2 |
| 95 | + (z3 - z2) / kZ_max, // inner T3: Difference in z between hit 3 and 2 normalized |
| 96 | + (r3 - r2) / kR_max, // inner T3: Difference in r between hit 3 and 2 normalized |
| 97 | + |
| 98 | + eta4 - eta3, // outer T3: Difference in eta between hit 4 and 3 |
| 99 | + cms::alpakatools::deltaPhi(acc, phi4, phi3) / kPhi_norm, // inner T3: Difference in phi between hit 4 and 3 |
| 100 | + (z4 - z3) / kZ_max, // outer T3: Difference in z between hit 4 and 3 normalized |
| 101 | + (r4 - r3) / kR_max, // outer T3: Difference in r between hit 4 and 3 normalized |
| 102 | + |
| 103 | + eta5 - eta4, // outer T3: Difference in eta between hit 5 and 4 |
| 104 | + cms::alpakatools::deltaPhi(acc, phi5, phi4) / kPhi_norm, // inner T3: Difference in phi between hit 5 and 4 |
| 105 | + (z5 - z4) / kZ_max, // outer T3: Difference in z between hit 5 and 4 normalized |
| 106 | + (r5 - r4) / kR_max, // outer T3: Difference in r between hit 5 and 4 normalized |
107 | 107 |
|
108 | 108 | alpaka::math::log10(acc, innerRadius), // T5 inner radius (t5_innerRadius) |
109 | 109 | alpaka::math::log10(acc, bridgeRadius), // T5 bridge radius (t5_bridgeRadius) |
|
0 commit comments