Skip to content
Snippets Groups Projects
Commit efef877d authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

chore : removed debug prints

parent 9eb718ec
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
...@@ -34,9 +34,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { ...@@ -34,9 +34,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
SECTION("flatten last layer : nothing removed because pattern searched is " SECTION("flatten last layer : nothing removed because pattern searched is "
"Flatten=>FC") { "Flatten=>FC") {
std::shared_ptr<Aidge::GraphView> g = Sequential({fc0, flatten}); std::shared_ptr<Aidge::GraphView> g = Sequential({fc0, flatten});
fmt::print("debug {}", __LINE__);
removeFlatten(g); removeFlatten(g);
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1);
...@@ -48,10 +47,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { ...@@ -48,10 +47,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
} }
SECTION("flatten first layer : flatten removed") { SECTION("flatten first layer : flatten removed") {
auto g = Sequential({flatten, fc0}); auto g = Sequential({flatten, fc0});
fmt::print("debug {}", __LINE__);
removeFlatten(g); removeFlatten(g);
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1);
...@@ -62,11 +59,10 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { ...@@ -62,11 +59,10 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
CHECK(g->getRootNode() == fc0); CHECK(g->getRootNode() == fc0);
} }
SECTION("flatten middle layer") { SECTION("flatten middle layer") {
auto g = Sequential({fc0, flatten, fc1}); auto g = Sequential({fc0, flatten, fc1});
fmt::print("debug {}", __LINE__);
removeFlatten(g); removeFlatten(g);
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1);
...@@ -81,11 +77,9 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { ...@@ -81,11 +77,9 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
// prod->addChild(flatten, 0); // prod->addChild(flatten, 0);
// flatten->addChild(fc0, 0); // flatten->addChild(fc0, 0);
// auto g = std::make_shared<GraphView>({prod, flatten, fc0}); // auto g = std::make_shared<GraphView>({prod, flatten, fc0});
fmt::print("debug {}", __LINE__);
removeFlatten(g); removeFlatten(g);
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 0); CHECK(g->getOrderedInputs().size() == 0);
CHECK(g->getOrderedOutputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0); CHECK(g->getOrderedInputs()[0].first == fc0);
......
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