exampleGpuLib.hpp

Go to the documentation of this file.

#ifndef EXAMPLEGPULIB_H
#define EXAMPLEGPULIB_H
 
#include <string>
#include <vector>
#include <cassert>
 
#include "cuda_runtime.h"
#include "cublas_v2.h"
 
#include "rtctk/exampleDataTask/CumulativeAverage_cuda.cuh"
 
namespace rtctk::exampleDataTask {
 
class exampleGpuLib {
 
   public:
      exampleGpuLib(int input_length, int output_length, int gpu);
      ~exampleGpuLib();
 
      void SetMatrix(float * mat, bool flip = true);
      std::vector<float> GetMatrix();
 
 
      void SetAvgSlopes(std::vector<float> avg_array);
      std::vector<float> GetAvgSlopes();
 
      std::vector<float> GetResults(bool download = false);
 
      void NewSample(const float * sample, int callback_count);
      void Compute();
 
      void initReaderThread();
 
   protected:
      // sets the required GPU
      void SetGPU();
 
      void PrintCudaError(cudaError_t error);
      std::string CublasGetStatusString(cublasStatus_t error);
      void PrintCublasStatus(cublasStatus_t status);
 
   private:
      int m_gpu;
      int currentSample;
      float m_alpha;
      float m_beta;
 
      cublasHandle_t handle;
 
      // input_vector
      int m_slopes;
      int m_modes;
 
 
      // slopes vector
      float *m_slopes_vector;
      float *m_slopes_vector_d;
 
      // avg_array_vectore
      float * m_avg_slopes;
      float * m_avg_slopes_d;
 
      // matrix
      float * m_slopes_to_modes_matrix;
      float * m_slopes_to_modes_matrix_d;
 
      // output vector
 
      float * m_modes_vector;
      float * m_modes_vector_d;
 
};
 
} // namespace 
 
#endif