Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-08 10:44:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bias in conv layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-20 19:09:00 +02:00
parent d08567a563
commit 5e663b9029
3 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/kernels/convolution.cu
View File

@@ -7,6 +7,7 @@ using namespace CUDANet;
__global__ void Kernels::convolution(
const float* __restrict__ d_input,
const float* __restrict__ d_kernel,
const float* __restrict__ d_bias,
float* __restrict__ d_output,
const int inputSize,
const int nChannels,
@@ -51,5 +52,5 @@ __global__ void Kernels::convolution(
}
}
d_output[f * outputSize * outputSize + i * outputSize + j] = sum;
d_output[f * outputSize * outputSize + i * outputSize + j] = sum + d_bias[f];
}