gpuVertexFinder.h

Go to the documentation of this file.

#ifndef RecoVertex_PixelVertexFinding_plugins_gpuVertexFinder_h
#define RecoVertex_PixelVertexFinding_plugins_gpuVertexFinder_h

#include <cstddef>
#include <cstdint>

#include "CUDADataFormats/Track/interface/PixelTrackUtilities.h"
#include "CUDADataFormats/Vertex/interface/ZVertexSoAHeterogeneousHost.h"
#include "CUDADataFormats/Vertex/interface/ZVertexSoAHeterogeneousDevice.h"
#include "CUDADataFormats/Vertex/interface/ZVertexUtilities.h"
#include "PixelVertexWorkSpaceUtilities.h"
#include "PixelVertexWorkSpaceSoAHost.h"
#include "PixelVertexWorkSpaceSoADevice.h"

namespace gpuVertexFinder {

  using VtxSoAView = zVertex::ZVertexSoAView;
  using WsSoAView = gpuVertexFinder::workSpace::PixelVertexWorkSpaceSoAView;

  __global__ void init(VtxSoAView pdata, WsSoAView pws) {
    zVertex::utilities::init(pdata);
    gpuVertexFinder::workSpace::utilities::init(pws);
  }

  template <typename TrackerTraits>
  class Producer {
    using TkSoAConstView = TrackSoAConstView<TrackerTraits>;

  public:
    Producer(bool oneKernel,
             bool useDensity,
             bool useDBSCAN,
             bool useIterative,
             bool doSplitting,
             int iminT,      // min number of neighbours to be "core"
             float ieps,     // max absolute distance to cluster
             float ierrmax,  // max error to be "seed"
             float ichi2max  // max normalized distance to cluster
             )
        : oneKernel_(oneKernel && !(useDBSCAN || useIterative)),
          useDensity_(useDensity),
          useDBSCAN_(useDBSCAN),
          useIterative_(useIterative),
          doSplitting_(doSplitting),
          minT(iminT),
          eps(ieps),
          errmax(ierrmax),
          chi2max(ichi2max) {}

    ~Producer() = default;

    ZVertexSoADevice makeAsync(cudaStream_t stream, const TkSoAConstView &tracks_view, float ptMin, float ptMax) const;
    ZVertexSoAHost make(const TkSoAConstView &tracks_view, float ptMin, float ptMax) const;

  private:
    const bool oneKernel_;
    const bool useDensity_;
    const bool useDBSCAN_;
    const bool useIterative_;
    const bool doSplitting_;

    int minT;       // min number of neighbours to be "core"
    float eps;      // max absolute distance to cluster
    float errmax;   // max error to be "seed"
    float chi2max;  // max normalized distance to cluster
  };

}  // namespace gpuVertexFinder

#endif  // RecoVertex_PixelVertexFinding_plugins_gpuVertexFinder_h