From 5a9d94c5263b5f36631e68ddc4746c35f7015d17 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Fri, 20 Oct 2023 13:56:35 +0000
Subject: [PATCH] [Upd] forwa Concat after removal of 'NbInputs' attribute
---
.../aidge/backend/cpu/operator/ConcatImpl_forward_kernels.hpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/backend/cpu/operator/ConcatImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/ConcatImpl_forward_kernels.hpp
index e67419d3..b76f384b 100644
--- a/include/aidge/backend/cpu/operator/ConcatImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/ConcatImpl_forward_kernels.hpp
@@ -42,11 +42,11 @@ void ConcatImpl_cpu_forward_kernel(const Concat_Op::Attrs& attrs,
DimSize_t outputAxisValue = std::accumulate(concatAxisValues.begin(), concatAxisValues.end(), 0);
DimSize_t prodDimLower = 1;
- for (DimIdx_t i = 0; i < std::get<1>(attrs); ++i) {
+ for (DimIdx_t i = 0; i < std::get<0>(attrs); ++i) {
prodDimLower *= dimsFirstInput[i];
}
DimSize_t prodDimHigher = 1;
- for (DimIdx_t i = std::get<1>(attrs) + 1; static_cast<std::size_t>(i) < dimsFirstInput.size();
+ for (DimIdx_t i = std::get<0>(attrs) + 1; static_cast<std::size_t>(i) < dimsFirstInput.size();
++i) {
prodDimHigher *= dimsFirstInput[i];
}
--
GitLab