From 916f1ef647ea5a0758599fd223d8b16f298ffcee Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 28 Jun 2024 13:27:36 +0200
Subject: [PATCH] Implement resetConsummerProducer() for meta op
---
include/aidge/operator/MetaOperator.hpp | 1 +
src/operator/MetaOperator.cpp | 14 ++++++++++++++
2 files changed, 15 insertions(+)
diff --git a/include/aidge/operator/MetaOperator.hpp b/include/aidge/operator/MetaOperator.hpp
index 73a5c6a99..744564b4b 100644
--- a/include/aidge/operator/MetaOperator.hpp
+++ b/include/aidge/operator/MetaOperator.hpp
@@ -108,6 +108,7 @@ public:
Elts_t getNbProducedData(IOIndex_t outputIdx) const override;
void updateConsummerProducer() override;
+ void resetConsummerProducer() override;
void forward() override;
void backward() override {
AIDGE_THROW_OR_ABORT(std::runtime_error, "backward() not implemented yet for a MetaOperator");
diff --git a/src/operator/MetaOperator.cpp b/src/operator/MetaOperator.cpp
index 7362f67fc..e7c500337 100644
--- a/src/operator/MetaOperator.cpp
+++ b/src/operator/MetaOperator.cpp
@@ -134,6 +134,20 @@ Aidge::Elts_t Aidge::MetaOperator_Op::getNbProducedData(IOIndex_t outputIdx) con
}
}
+void Aidge::MetaOperator_Op::resetConsummerProducer() {
+ if (mImpl) {
+ mImpl->resetConsummerProducer();
+ }
+ else {
+ if (!mScheduler) {
+ // Lazy initialization
+ mScheduler = std::make_shared<SequentialScheduler>(mGraph, mUpperNode.lock());
+ }
+
+ mScheduler->resetScheduling();
+ }
+}
+
void Aidge::MetaOperator_Op::updateConsummerProducer() {
if (mImpl) {
mImpl->updateConsummerProducer();
--
GitLab