Skip to content

Commit 06e26a8

Browse files
committed
Adding GPUd() for on-device functions
1 parent 70adf1e commit 06e26a8

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizer.cxx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,17 +122,17 @@ void GPUTPCNNClusterizer::inferenceNetworkReg2(processorType& clusterer, int8_t
122122
}
123123

124124
// THe following arithmetic is done because the network is trained with a split between IROC and OROC boundary
125-
int GPUTPCNNClusterizer::padOffset(int row_ref, int row_current, const GPUTPCGeometry& geo)
125+
GPUd() int GPUTPCNNClusterizer::padOffset(int row_ref, int row_current, const GPUTPCGeometry& geo)
126126
{
127127
return (int)((geo.NPads(row_current) - geo.NPads(row_ref)) / 2);
128128
}
129129

130-
int GPUTPCNNClusterizer::rowOffset(int row, int global_shift)
130+
GPUd() int GPUTPCNNClusterizer::rowOffset(int row, int global_shift)
131131
{
132132
return (row > 62 ? global_shift : 0);
133133
}
134134

135-
bool GPUTPCNNClusterizer::isBoundary(int row, int pad, int global_shift, const GPUTPCGeometry& geo)
135+
GPUd() bool GPUTPCNNClusterizer::isBoundary(int row, int pad, int global_shift, const GPUTPCGeometry& geo)
136136
{
137137
if (pad < 0 || row < 0) { // Faster short-circuit
138138
return true;

GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizer.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,9 @@ class GPUTPCNNClusterizer : public GPUKernelTemplate
7777
static void inferenceNetworkReg2(processorType&, int8_t = 0);
7878

7979
private:
80-
static int padOffset(int, int, const GPUTPCGeometry&);
81-
static int rowOffset(int, int);
82-
static bool isBoundary(int, int, int, const GPUTPCGeometry&);
80+
static GPUd() int padOffset(int, int, const GPUTPCGeometry&);
81+
static GPUd() int rowOffset(int, int);
82+
static GPUd() bool isBoundary(int, int, int, const GPUTPCGeometry&);
8383
};
8484

8585
} // namespace o2::gpu

0 commit comments

Comments
 (0)