HcalMahiPulseOffsetsGPU.h

Go to the documentation of this file.

#ifndef RecoLocalCalo_HcalRecAlgos_interface_HcalMahiPulseOffsetsGPU_h
#define RecoLocalCalo_HcalRecAlgos_interface_HcalMahiPulseOffsetsGPU_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#ifndef __CUDACC__
#include "HeterogeneousCore/CUDAUtilities/interface/HostAllocator.h"
#include "HeterogeneousCore/CUDACore/interface/ESProduct.h"
#endif

class HcalMahiPulseOffsetsGPU {
public:
  struct Product {
    ~Product();
    int* values;
  };

#ifndef __CUDACC__
  // rearrange reco params
  HcalMahiPulseOffsetsGPU(edm::ParameterSet const&);

  // will trigger deallocation of Product thru ~Product
  ~HcalMahiPulseOffsetsGPU() = default;

  std::vector<int, cms::cuda::HostAllocator<int>> const& getValues() const { return values_; }

  // get device pointers
  Product const& getProduct(cudaStream_t) const;

private:
  std::vector<int, cms::cuda::HostAllocator<int>> values_;

  cms::cuda::ESProduct<Product> product_;
#endif
};

#endif