diff --git a/include/aidge/backend/cpu/operator/FoldImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/FoldImpl_forward_kernels.hpp index 81b96d4b3b1e9c59ad424967eae54b75224f83bd..e2db9475cb452df6b43df02bbeff67741c1f71fa 100644 --- a/include/aidge/backend/cpu/operator/FoldImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/FoldImpl_forward_kernels.hpp @@ -52,8 +52,8 @@ void FoldImpl2D_cpu_forward_kernel(const Fold_Op<2>::Attrs &attrs, const std::ve for (DimSize_t n = 0; n < dims[0]; ++n) { for (DimSize_t outC = 0; outC < outChannels; ++outC) { - const auto inOffsetH = outC % kernelDims[1]; - const auto inOffsetW = (outC / kernelDims[1]) % kernelDims[0]; + const auto inOffsetW = outC % kernelDims[1]; + const auto inOffsetH = (outC / kernelDims[1]) % kernelDims[0]; const auto inC = outC / kernelDims[0] / kernelDims[1]; for (DimSize_t outH = 0; outH < outHeight; ++outH) { diff --git a/unit_tests/operator/Test_FoldImpl.cpp b/unit_tests/operator/Test_FoldImpl.cpp index 010b23d395acea79a0b2503e3a43f10a11661f77..6832f5a42d796d9261495794e0758ce1b6df0346 100644 --- a/unit_tests/operator/Test_FoldImpl.cpp +++ b/unit_tests/operator/Test_FoldImpl.cpp @@ -173,6 +173,6 @@ TEST_CASE("[cpu/operator] Fold(forward)", "[Fold][CPU]") { SequentialScheduler scheduler(g); scheduler.forward(); - opFold->getOutput(0)->print(); + //opFold->getOutput(0)->print(); REQUIRE(*(opFold->getOutput(0)) == *myOutput); } \ No newline at end of file