Skip to content

Make TrackPar(Cov) compatible to GPU #5043

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

Merged
davidrohr merged 9 commits into from
Dec 12, 2020
Merged

Make TrackPar(Cov) compatible to GPU #5043

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
08baf58
GPU: Fix macro name
davidrohr Dec 9, 2020
ef43df8
GPU: Was missing the GPU annotations for the GPU versions of Cartesia…
davidrohr Dec 9, 2020
a75e6fb
Make rotateZ / sincos / sector2angle compatible to GPU
davidrohr Dec 9, 2020
469a851
GPU: Add a dummy stream logger implementation for GPU in order not to…
davidrohr Dec 9, 2020
0098fd8
GPU: Add gpu::gpustd::array to forward std::array on host / mimic as …
davidrohr Dec 9, 2020
7de7844
Hide all stuff that uses ROOT from GPU compilation
davidrohr Dec 9, 2020
a71928f
Move PID constants from static class member to a namespace, since sta…
davidrohr Dec 9, 2020
4e37ae8
GPU: Make TrackPar / TrackParCov compile for GPU
davidrohr Dec 9, 2020
ba96502
GPU: Workaround for reoccuring HIP problem parsing host-code during d…
davidrohr Dec 11, 2020
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
26 changes: 14 additions & 12 deletions Common/MathUtils/include/MathUtils/CartesianGPU.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,32 +14,34 @@
#ifndef ALICEO2_CARTESIANGPU_H
#define ALICEO2_CARTESIANGPU_H

#include "GPUCommonDef.h"

namespace o2::math_utils
{

namespace detail
{
template <typename T, int I>
struct GPUPoint2D {
GPUPoint2D() = default;
GPUPoint2D(T a, T b) : xx(a), yy(b) {}
GPUdDefault() GPUPoint2D() = default;
GPUd() GPUPoint2D(T a, T b) : xx(a), yy(b) {}
GPUd() float X() const { return xx; }
GPUd() float Y() const { return yy; }
GPUd() float R() const { return o2::gpu::CAMath::Sqrt(xx * xx + yy * yy); }
GPUd() void SetX(float v) { xx = v; }
GPUd() void SetY(float v) { yy = v; }
T xx;
T yy;
float X() const { return xx; }
float Y() const { return yy; }
float R() const { return o2::gpu::CAMath::Sqrt(xx * xx + yy * yy); }
void SetX(float v) { xx = v; }
void SetY(float v) { yy = v; }
};

template <typename T, int I>
struct GPUPoint3D : public GPUPoint2D<T, I> {
GPUPoint3D() = default;
GPUPoint3D(T a, T b, T c) : GPUPoint2D<T, I>(a, b), zz(c) {}
GPUdDefault() GPUPoint3D() = default;
GPUd() GPUPoint3D(T a, T b, T c) : GPUPoint2D<T, I>(a, b), zz(c) {}
GPUd() float Z() const { return zz; }
GPUd() float R() const { return o2::gpu::CAMath::Sqrt(GPUPoint2D<T, I>::xx * GPUPoint2D<T, I>::xx + GPUPoint2D<T, I>::yy * GPUPoint2D<T, I>::yy + zz * zz); }
GPUd() void SetZ(float v) { zz = v; }
T zz;
float Z() const { return zz; }
float R() const { return o2::gpu::CAMath::Sqrt(GPUPoint2D<T, I>::xx * GPUPoint2D<T, I>::xx + GPUPoint2D<T, I>::yy * GPUPoint2D<T, I>::yy + zz * zz); }
void SetZ(float v) { zz = v; }
};
} // namespace detail

Expand Down
9 changes: 5 additions & 4 deletions Common/MathUtils/include/MathUtils/Utils.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -108,23 +108,24 @@ GPUdi() void sincos(float ang, float& s, float& c)
{
detail::sincos<float>(ang, s, c);
}

#ifndef __OPENCL__
GPUdi() void sincosd(double ang, double& s, double& c)
{
detail::sincos<double>(ang, s, c);
}
#endif

#ifndef GPUCA_GPUCODE_DEVICE
inline void rotateZ(float xL, float yL, float& xG, float& yG, float snAlp, float csAlp)
GPUdi() void rotateZ(float xL, float yL, float& xG, float& yG, float snAlp, float csAlp)
{
return detail::rotateZ<float>(xL, yL, xG, yG, snAlp, csAlp);
}

inline void rotateZd(double xL, double yL, double& xG, double& yG, double snAlp, double csAlp)
GPUdi() void rotateZd(double xL, double yL, double& xG, double& yG, double snAlp, double csAlp)
{
return detail::rotateZ<double>(xL, yL, xG, yG, snAlp, csAlp);
}

#ifndef GPUCA_GPUCODE_DEVICE
inline void rotateZInv(float xG, float yG, float& xL, float& yL, float snAlp, float csAlp)
{
detail::rotateZInv<float>(xG, yG, xL, yL, snAlp, csAlp);
Expand Down
9 changes: 5 additions & 4 deletions Common/MathUtils/include/MathUtils/detail/CircleXY.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
#ifndef MATHUTILS_INCLUDE_MATHUTILS_DETAIL_CIRCLEXY_H_
#define MATHUTILS_INCLUDE_MATHUTILS_DETAIL_CIRCLEXY_H_

#include "GPUCommonDef.h"
#include "GPUCommonRtypes.h"

namespace o2
Expand All @@ -32,18 +33,18 @@ struct CircleXY {
T rC; // circle radius
T xC; // x-center
T yC; // y-center
CircleXY(T r = T(), T x = T(), T y = T());
T getCenterD2() const;
GPUd() CircleXY(T r = T(), T x = T(), T y = T());
GPUd() T getCenterD2() const;
ClassDefNV(CircleXY, 2);
};

template <typename T>
CircleXY<T>::CircleXY(T r, T x, T y) : rC(std::move(r)), xC(std::move(x)), yC(std::move(y))
GPUdi() CircleXY<T>::CircleXY(T r, T x, T y) : rC(r), xC(x), yC(y)
{
}

template <typename T>
inline T CircleXY<T>::getCenterD2() const
GPUdi() T CircleXY<T>::getCenterD2() const
{
return xC * xC + yC * yC;
}
Expand Down
Loading