From 664ca093f63921310207fac61e3142147fecf67d Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Thu, 9 Jan 2025 10:01:21 +0100
Subject: [PATCH] Fixed bug with format
---
src/scheduler/Scheduler.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/scheduler/Scheduler.cpp b/src/scheduler/Scheduler.cpp
index 242a2d0e6..d456c8af2 100644
--- a/src/scheduler/Scheduler.cpp
+++ b/src/scheduler/Scheduler.cpp
@@ -722,12 +722,12 @@ Aidge::MemoryManager Aidge::Scheduler::generateMemoryAutoConcat(bool incProducer
const auto parentOp = std::static_pointer_cast<OperatorTensor>(concatParent->getOperator());
const auto parentRequiredSize = parentOp->getRequiredMemory(outputIdx, {});
const auto parentOutputDims = (parentOp->getOutput(outputIdx)) ? parentOp->getOutput(outputIdx)->dims() : std::vector<DimSize_t>();
+ const auto parentOutputFormat = (parentOp->getOutput(outputIdx)) ? parentOp->getOutput(outputIdx)->dataFormat() : DataFormat::Default;
// By default, specifies a fully monolithic memory block
std::size_t parentSize = parentRequiredSize.data;
- if (parentOutputDims.size() > 3) {
- // If it is possible, assume a NCHW layout
+ if (parentOutputFormat == DataFormat::NHWC) {
parentSize = parentOutputDims.end()[-3];
}
--
GitLab