cuda-convolution-inference/cuda_conv_interface.h at main · George-ao/cuda-convolution-inference · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
#ifndef CUDA_CONV_INTERFACE_H
#define CUDA_CONV_INTERFACE_H

class GPUInterface
{
    public:
    void get_device_properties();
    void conv_forward_gpu_prolog(const float *host_output, const float *host_input, const float *host_mask, float **device_output_ptr, float **device_input_ptr, float **device_mask_ptr, const int Batch, const int Map_out, const int Channel, const int Height, const int Width, const int K);
    void conv_forward_gpu(float *device_output, const float *device_input, const float *device_mask, const int Batch, const int Map_out, const int Channel, const int Height, const int Width, const int K);
    void conv_forward_gpu_epilog(float *host_output, float *device_output, float *device_input, float *device_mask, const int Batch, const int Map_out, const int Channel, const int Height, const int Width, const int K);
};

#endif