From 7e75943a6b54402b2101dade1daac661f688c1c3 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Thu, 7 Mar 2024 21:32:39 +0100
Subject: [PATCH] Add stride to index calculation

---
 src/layers/conv2d.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/layers/conv2d.cu b/src/layers/conv2d.cu
index 841ffa6..c31bc4a 100644
--- a/src/layers/conv2d.cu
+++ b/src/layers/conv2d.cu
@@ -108,7 +108,7 @@ void Layers::Conv2d::host_conv(const float* input, float* output) {
                             // For now stride = 1
                             
                             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];
                         }