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

Add stride to index calculation

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-07 21:32:39 +01:00
parent fc2c1616b4
commit 7e75943a6b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/conv2d.cu
View File

@@ -108,7 +108,7 @@ void Layers::Conv2d::host_conv(const float* input, float* output) {
// For now stride = 1 // For now stride = 1
int kernelIndex = k * (kernelSize * inputChannels * numFilters) + l * (inputChannels * numFilters) + c * (numFilters) + f; int kernelIndex = k * (kernelSize * inputChannels * numFilters) + l * (inputChannels * numFilters) + c * (numFilters) + f;
int inputIndex = i * (inputSize * inputChannels) + j * (inputChannels) + c; int inputIndex = (i * stride + k) * (inputSize * inputChannels) + (j + stride + l) * (inputChannels) + c;
sum += kernels[kernelIndex] * input[inputIndex]; sum += kernels[kernelIndex] * input[inputIndex];
} }