From 7c75fde34e0301225762f663d55763254f9cb43a Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 28 Apr 2025 15:35:18 +0200
Subject: [PATCH] Fixed bug
---
aidge_export_cpp/kernels/convolution.hpp | 2 +-
aidge_export_cpp/unit_tests/test_export.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/aidge_export_cpp/kernels/convolution.hpp b/aidge_export_cpp/kernels/convolution.hpp
index 7df55ab..1669ee8 100644
--- a/aidge_export_cpp/kernels/convolution.hpp
+++ b/aidge_export_cpp/kernels/convolution.hpp
@@ -237,7 +237,7 @@ void convolution_depthwise_forward(
const int iOffset = NB_CHANNELS * iPos;
const int wOffset = (output*KERNEL_HEIGHT + sy)
- * KERNEL_WIDTH * NB_CHANNELS;
+ * KERNEL_WIDTH;
if (DILATION_X == 1 && ((PADDING_X == 0
&& OUTPUTS_WIDTH == OUTPUTS_WIDTH_NOPAD)
diff --git a/aidge_export_cpp/unit_tests/test_export.py b/aidge_export_cpp/unit_tests/test_export.py
index 387c595..65db07e 100644
--- a/aidge_export_cpp/unit_tests/test_export.py
+++ b/aidge_export_cpp/unit_tests/test_export.py
@@ -411,7 +411,7 @@ class test_operator_export(unittest.TestCase):
self.unit_test_export(model, "Conv2D", [[1, 3, 12, 12]], False, False)
def test_export_convDepthWise2D(self):
- print("Conv2D")
+ print("ConvDepthWise2D")
model = aidge_core.sequential([
aidge_core.ConvDepthWise2D(nb_channels=3, kernel_dims=(3, 3), name="conv")
])
--
GitLab