mirror of
https://github.com/augustin64/projet-tipe
synced 2025-03-13 06:15:21 +01:00
23 lines
861 B
C
23 lines
861 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_dim);
|
|
|
|
#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_dim);
|
|
|
|
/*
|
|
* Effectue la convolution naïvement sur la carte graphique
|
|
*/
|
|
void make_convolution_device(Kernel_cnn* kernel, float*** input, float*** output, int output_dim);
|
|
#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_dim); |