Skip to content
Snippets Groups Projects
Commit 963989a0 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fix test following change in LSTM outputs order

parent 655e54d2
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!37Support for recurrent networks
Pipeline #39475 failed
...@@ -304,10 +304,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { ...@@ -304,10 +304,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
auto microGraphScheduler = std::dynamic_pointer_cast<MetaOperator_Op>(op)->getMicroGraphScheduler(); auto microGraphScheduler = std::dynamic_pointer_cast<MetaOperator_Op>(op)->getMicroGraphScheduler();
microGraphScheduler->saveSchedulingDiagram("lstm_values_scheduling"); microGraphScheduler->saveSchedulingDiagram("lstm_values_scheduling");
op->getOutput(1)->print(); op->getOutput(0)->print();
myHiddenState->print(); myHiddenState->print();
REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
} }
SECTION("LSTM(forward_values_seq)") { SECTION("LSTM(forward_values_seq)") {
auto pop = Pop(); auto pop = Pop();
...@@ -360,10 +360,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { ...@@ -360,10 +360,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
myGraph->save("lstm_seq_mygraph", true, true); myGraph->save("lstm_seq_mygraph", true, true);
op->getOutput(1)->print(); op->getOutput(0)->print();
myHiddenState->print(); myHiddenState->print();
REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
} }
SECTION("LSTM(forward_values_seq_flatten)") { SECTION("LSTM(forward_values_seq_flatten)") {
auto pop = Pop(); auto pop = Pop();
...@@ -421,9 +421,9 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { ...@@ -421,9 +421,9 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") {
scheduler.forward(true, true); scheduler.forward(true, true);
scheduler.saveSchedulingDiagram("lstm_seq_flatten_schedule"); scheduler.saveSchedulingDiagram("lstm_seq_flatten_schedule");
op->getOutput(1)->print(); op->getOutput(0)->print();
myHiddenState->print(); myHiddenState->print();
REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState));
} }
} }
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment