mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Initial cuda conv kernel implementation
This commit is contained in:
16
include/kernels/convolution.cuh
Normal file
16
include/kernels/convolution.cuh
Normal 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
|
||||
Reference in New Issue
Block a user