From 1d32b0bed09eb89be2dd8d9fec115a6a5c98cc9b Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Thu, 15 Feb 2024 14:40:09 +0000
Subject: [PATCH] [Upd] backward test in Test_Scheduler.cpp
---
unit_tests/scheduler/Test_Scheduler.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index dc13c6e5..7321c151 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -349,7 +349,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
}
TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward]") {
- std::shared_ptr<GraphView> gv = Sequential({ReLU(), ReLU()});
+ std::shared_ptr<GraphView> gv = Sequential({ReLU("relu0"), Sqrt("srqt0"), ReLU("relu1")});
std::shared_ptr<Tensor> inputTensor =
std::make_shared<Tensor>(Array4D<int, 2, 1, 5, 5>{{{{{0, 1, 2, 3, 4},
@@ -366,7 +366,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
// implem already set to default
auto myProd = Producer(inputTensor, "prod");
myProd -> addChild(gv);
- gv -> compile("cpu", DataType::Float32);
+ gv -> compile("cpu", DataType::Int32);
compile_gradient(gv);
SequentialScheduler scheduler(gv);
scheduler.backward();
--
GitLab