gpuLib.hpp

Go to the documentation of this file.

 
#ifndef EXAMPLEGPULIB_H
#define EXAMPLEGPULIB_H
 
#include <cassert>
#include <string>
#include <vector>
 
#include "cublas_v2.h"
#include "cuda_runtime.h"
 
#include "rtctk/exampleDataTask/CumulativeAverage_cuda.cuh"
 
class GpuLib {
public:
    GpuLib(int input_length, int output_length, int gpu);
    ~GpuLib();
 
    void SetMatrix(float* mat, bool flip = true);
    std::vector<float> GetMatrix();
 
    void ResetAvgSlopes();
 
    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 current_sample;
    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_vector
    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;
};
 
#endif  // EXAMPLEGPULIB_H