diff --git a/include/aidge/backend/cpu/operator/FCImpl_kernels.hpp b/include/aidge/backend/cpu/operator/FCImpl_kernels.hpp index 0d2d9e450142a91173381937d31e1ccecf2c779c..01f62aff0707ab38d7c0b76fdebc7cab9bc5a642 100644 --- a/include/aidge/backend/cpu/operator/FCImpl_kernels.hpp +++ b/include/aidge/backend/cpu/operator/FCImpl_kernels.hpp @@ -167,7 +167,7 @@ void FCImpl_cpu_backward_kernel(const DimSize_t batchSize, for (std::size_t o = 0; o < outputFeatureSize; ++o) { sum += weight[o*inputFeatureSize + c] * input[b*outputFeatureSize + o]; } - output[b*inputFeatureSize + c]+= sum; + output[b*inputFeatureSize + c] = sum; } } } diff --git a/src/operator/HeavisideImpl.cpp b/src/operator/HeavisideImpl.cpp index 217799f04484116408e024f10973655b5161c7a3..8ddfb51030afc17899c9b095911b049d4358689c 100644 --- a/src/operator/HeavisideImpl.cpp +++ b/src/operator/HeavisideImpl.cpp @@ -26,6 +26,8 @@ template <> void Aidge::HeavisideImplCpu::forward() { const auto impl = Registrar<HeavisideImplCpu>::create(getBestMatch(getRequiredSpec())); + Log::info("(HeavisideImpl.cpp) Before forward"); + Log::info("(HeavisideImpl.cpp) Input 0 \n{}", input0->toString()); impl.forward(input0->size(), getCPUPtr(mOp.getRawInput(0)), getCPUPtr(mOp.getRawOutput(0)), op_.value()); }