From 686f624cbf30e375bacf9d0e9a58a9ec47f46f01 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 4 Dec 2024 18:48:14 +0100
Subject: [PATCH] Fix scheduling with Memorize
---
src/scheduler/Scheduler.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/scheduler/Scheduler.cpp b/src/scheduler/Scheduler.cpp
index 0e8cba07e..396e90c09 100644
--- a/src/scheduler/Scheduler.cpp
+++ b/src/scheduler/Scheduler.cpp
@@ -989,7 +989,7 @@ Aidge::Scheduler::getPriorProducersConsumers(const std::shared_ptr<Node>& node)
prior.requiredProducers.insert(parent.first);
prior.priorConsumers.insert(node);
}
- else if (parent.first->type() == Memorize_Op::Type) {
+ else if (parent.first->type() == Memorize_Op::Type && parent.second == 1) {
// Break cycles
return PriorProducersConsumers(); // not scheduled
}
--
GitLab