From 0d41e77f6953ef418e488f774c7cd4826072e8d0 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 6 Feb 2024 10:47:19 +0100
Subject: [PATCH] Fixed issue with init Producer
---
unit_tests/scheduler/Test_Scheduler.cpp | 9 +--------
1 file changed, 1 insertion(+), 8 deletions(-)
diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index e1815725..1ae235f9 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -218,8 +218,6 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
auto add2 = Add(2, "add2");
auto bias = Producer(biasTensor, "bias");
auto init = Producer(initTensor, "init");
- init->getOperator()->setBackend("cpu");
- init->getOperator()->setDataType(Aidge::DataType::Int32);
std::shared_ptr<GraphView> g = Sequential({add1, mem, add2});
init->addChild(mem, 0, 1);
@@ -230,12 +228,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
g->add(mem);
g->add(add1);
g->add(add2);
- //g->add(init); // not working because of forwardDims()
- // TODO: FIXME:
- // forwardDims() starts with inputNodes(). If the initializer
- // of Memorize is inside the graph, forwardDims() will get stuck
- // to the node taking the recursive connection because Memorize
- // output dims must first be computed from the initializer.
+ g->add(init);
g->add(bias);
g->setBackend("cpu");
--
GitLab