TestCUDAProducerGPUKernel.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDACore_TestCUDAProducerGPUKernel_h
#define HeterogeneousCore_CUDACore_TestCUDAProducerGPUKernel_h

#include <string>

#include <cuda_runtime.h>

#include "HeterogeneousCore/CUDAUtilities/interface/device_unique_ptr.h"

class TestCUDAProducerGPUKernel {
public:
  static constexpr int NUM_VALUES = 4000;

  TestCUDAProducerGPUKernel() = default;
  ~TestCUDAProducerGPUKernel() = default;

  // returns (owning) pointer to device memory
  cms::cuda::device::unique_ptr<float[]> runAlgo(const std::string& label, cudaStream_t stream) const {
    return runAlgo(label, nullptr, stream);
  }
  cms::cuda::device::unique_ptr<float[]> runAlgo(const std::string& label,
                                                 const float* d_input,
                                                 cudaStream_t stream) const;

  void runSimpleAlgo(float* d_data, cudaStream_t stream) const;
};

#endif