diff --git a/include/aidge/operator/AddImpl.hpp b/include/aidge/operator/AddImpl.hpp index 21ca02e75fedc2a40da4b033e81b658b69690048..4df4f18a04c5aa473a0f8bc988aedeb5cba3826d 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 a95054dd787a4e9de757746f59c23674ec78250c..c8783b36d42231d7792a80cd134de3f491ec8374 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>());