From 866ceb9a1d99cb9f0c2a1d409c0fe7f9cccdb3f0 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Thu, 28 Mar 2024 10:00:35 +0000
Subject: [PATCH] Update 'SequentialScheduler' test
---
unit_tests/scheduler/Test_Scheduler.cpp | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index 1fa0e577..525dbf43 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -20,7 +20,7 @@
#include "aidge/scheduler/Scheduler.hpp"
#include "aidge/backend/cpu.hpp"
-#include "aidge/recipies/GraphViewHelper.hpp"
+#include "aidge/recipes/GraphViewHelper.hpp"
using namespace Aidge;
@@ -372,7 +372,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
compile_gradient(gv);
SequentialScheduler scheduler(gv);
scheduler.forward();
- auto predictedOutput = gv->getOrderedOutputs()[0];
+ auto predictedOutput = gv->getOrderedOutputs()[0].first;
std::shared_ptr<Tensor> targetOutput =
std::make_shared<Tensor>(Array4D<float, 2, 1, 5, 5>{{{{{0.0f, 1.0f, 1.0f, 2.0f, 2.0f},
@@ -386,4 +386,5 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
{6.0f, 6.0f, 6.0f, 7.0f, 7.0f},
{7.0f, 7.0f, 7.0f, 7.0f, 7.0f}}}}});
- REQUIRE_NOTHROW(scheduler.backward(predictedOutput - targetOutput));
+ REQUIRE_NOTHROW(scheduler.backward({targetOutput}));
+}
\ No newline at end of file
--
GitLab