From 9eb718ecbbe967a38199c2f2b0f1055056c80aad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Wed, 27 Mar 2024 11:00:20 +0100
Subject: [PATCH] chore : added sum dbg prints
---
unit_tests/recipes/Test_removeFlatten.cpp | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/unit_tests/recipes/Test_removeFlatten.cpp b/unit_tests/recipes/Test_removeFlatten.cpp
index 84157f0c2..572fd121a 100644
--- a/unit_tests/recipes/Test_removeFlatten.cpp
+++ b/unit_tests/recipes/Test_removeFlatten.cpp
@@ -34,8 +34,9 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
SECTION("flatten last layer : nothing removed because pattern searched is "
"Flatten=>FC") {
std::shared_ptr<Aidge::GraphView> g = Sequential({fc0, flatten});
-
+ fmt::print("debug {}", __LINE__);
removeFlatten(g);
+ fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
@@ -47,8 +48,10 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
}
SECTION("flatten first layer : flatten removed") {
auto g = Sequential({flatten, fc0});
+ fmt::print("debug {}", __LINE__);
removeFlatten(g);
+ fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
@@ -59,10 +62,11 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
CHECK(g->getRootNode() == fc0);
}
SECTION("flatten middle layer") {
-
auto g = Sequential({fc0, flatten, fc1});
+ fmt::print("debug {}", __LINE__);
removeFlatten(g);
+ fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
@@ -77,9 +81,11 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
// prod->addChild(flatten, 0);
// flatten->addChild(fc0, 0);
// auto g = std::make_shared<GraphView>({prod, flatten, fc0});
+ fmt::print("debug {}", __LINE__);
removeFlatten(g);
+ fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 0);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
--
GitLab