mirror of
https://github.com/augustin64/projet-tipe
synced 2025-03-15 07:12:32 +01:00
27 lines
1004 B
C
27 lines
1004 B
C
#include "struct.h"
|
|
|
|
|
|
/*
|
|
* Effectue la convolution naïvement sur le processeur
|
|
*/
|
|
void make_convolution_cpu(Kernel_cnn* kernel, float*** input, float*** output, int output_width, int stride, int padding);
|
|
|
|
#ifdef __CUDACC__
|
|
/*
|
|
* Kernel de la convolution sur carte graphique
|
|
*/
|
|
__global__ void make_convolution_kernel(int k_size, int columns, int rows, float* bias, size_t pitch_bias, float**** weights, size_t pitch_weights, float*** input, size_t pitch_input, float*** output, size_t pitch_output, int output_width, int stride, int padding);
|
|
|
|
/*
|
|
* Effectue la convolution naïvement sur la carte graphique
|
|
*/
|
|
void make_convolution_device(Kernel_cnn* kernel, float*** input, float*** output, int output_width, int stride, int padding);
|
|
#endif
|
|
|
|
#ifdef __CUDACC__
|
|
extern "C"
|
|
#endif
|
|
/*
|
|
* Détermine si la convolution peut-être faite sur la carte graphique au moment de la compilation
|
|
*/
|
|
void make_convolution(Kernel_cnn* kernel, float*** input, float*** output, int output_width, int stride, int padding); |