From 963989a0238a8a49f2c3927cebb58194fe19ce7f Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 20 Feb 2024 15:47:53 +0100
Subject: [PATCH] Fix test following change in LSTM outputs order
---
unit_tests/operator/Test_MetaOperator.cpp | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/unit_tests/operator/Test_MetaOperator.cpp b/unit_tests/operator/Test_MetaOperator.cpp
index 5fc68acf..c0e9be1c 100644
--- a/unit_tests/operator/Test_MetaOperator.cpp
+++ b/unit_tests/operator/Test_MetaOperator.cpp
@@ -304,10 +304,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
auto microGraphScheduler = std::dynamic_pointer_cast<MetaOperator_Op>(op)->getMicroGraphScheduler();
microGraphScheduler->saveSchedulingDiagram("lstm_values_scheduling");
- op->getOutput(1)->print();
+ op->getOutput(0)->print();
myHiddenState->print();
- REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState));
+ REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
}
SECTION("LSTM(forward_values_seq)") {
auto pop = Pop();
@@ -360,10 +360,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
myGraph->save("lstm_seq_mygraph", true, true);
- op->getOutput(1)->print();
+ op->getOutput(0)->print();
myHiddenState->print();
- REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState));
+ REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
}
SECTION("LSTM(forward_values_seq_flatten)") {
auto pop = Pop();
@@ -421,9 +421,9 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
scheduler.forward(true, true);
scheduler.saveSchedulingDiagram("lstm_seq_flatten_schedule");
- op->getOutput(1)->print();
+ op->getOutput(0)->print();
myHiddenState->print();
- REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState));
+ REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
}
}
\ No newline at end of file
--
GitLab