Initial cuda conv kernel implementation

This commit is contained in:
2024-03-08 23:35:54 +01:00
parent 4b6fcbc191
commit e51aabc2f2
2 changed files with 69 additions and 0 deletions

View File

@@ -0,0 +1,16 @@
#ifndef CONVOLUTION_H
#define CONVOLUTION_H
__global__ void convolution_kernel(
const float* d_input,
const float* d_kernel,
float* d_output,
int inputSize,
int nChannels,
int kernelSize,
int stride,
int nFilters,
int outputSize
);
#endif // CONVOLUTION_H