From c86cb072428638b4f94fe38800496743447eab95 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Thu, 31 Aug 2023 18:47:31 +0200
Subject: [PATCH] Fixed missing parameters
---
include/aidge/operator/AddImpl.hpp | 3 ++-
src/operator/BatchNormImpl.cpp | 3 ++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/include/aidge/operator/AddImpl.hpp b/include/aidge/operator/AddImpl.hpp
index 21ca02e7..4df4f18a 100644
--- a/include/aidge/operator/AddImpl.hpp
+++ b/include/aidge/operator/AddImpl.hpp
@@ -79,9 +79,10 @@ class AddImpl_cpu : public OperatorImpl {
return 0;
}
- NbElts_t getRequiredMemory(const IOIndex_t /*outputIdx*/, const std::vector<DimSize_t>& inputsSize) const override final {
+ NbElts_t getRequiredMemory(const IOIndex_t outputIdx, const std::vector<DimSize_t>& inputsSize) const override final {
// Requires the whole tensors, regardless of available data on inputs
assert(outputIdx == 0 && "operator has only one output");
+ (void) outputIdx;
const auto& outputDims = std::static_pointer_cast<Tensor>(mOp.getOutput(0))->dims();
return std::accumulate(outputDims.begin(), outputDims.end(), NbElts_t(1), std::multiplies<NbElts_t>());
diff --git a/src/operator/BatchNormImpl.cpp b/src/operator/BatchNormImpl.cpp
index a95054dd..c8783b36 100644
--- a/src/operator/BatchNormImpl.cpp
+++ b/src/operator/BatchNormImpl.cpp
@@ -34,10 +34,11 @@ Aidge::NbElts_t Aidge::BatchNormImpl2D_cpu::getNbRequiredProtected(IOIndex_t /*i
return 0;
}
-Aidge::NbElts_t Aidge::BatchNormImpl2D_cpu::getRequiredMemory(const Aidge::IOIndex_t /*outputIdx*/,
+Aidge::NbElts_t Aidge::BatchNormImpl2D_cpu::getRequiredMemory(const Aidge::IOIndex_t outputIdx,
const std::vector<Aidge::DimSize_t> &/*inputsSize*/) const {
// Requires the whole tensors, regardless of available data on inputs
assert(outputIdx == 0 && "operator has only one output");
+ (void) outputIdx;
const auto &outputDims = std::static_pointer_cast<Tensor>(mOp.getOutput(0))->dims();
return std::accumulate(outputDims.begin(), outputDims.end(), NbElts_t(1), std::multiplies<NbElts_t>());
--
GitLab