From 1fcaee4bfd3014c3cf86299064c49ed0a906a250 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 4 Mar 2025 21:58:32 +0000
Subject: [PATCH] Fix test for shapeFolding.
---
aidge_core/unit_tests/test_forward_dims_constant_shape.py | 2 +-
src/recipes/ShapeFolding.cpp | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/aidge_core/unit_tests/test_forward_dims_constant_shape.py b/aidge_core/unit_tests/test_forward_dims_constant_shape.py
index aea0260c8..fd3fee04b 100644
--- a/aidge_core/unit_tests/test_forward_dims_constant_shape.py
+++ b/aidge_core/unit_tests/test_forward_dims_constant_shape.py
@@ -89,7 +89,7 @@ class test_forward_dims_constant_shape(unittest.TestCase):
def tearDown(self):
pass
- def test_attributes_name(self):
+ def test_constant_shape_folding(self):
# Note: Except Div every operator are backend independent
self.graph.set_backend("cpu")
self.graph.set_datatype(aidge_core.dtype.float32)
diff --git a/src/recipes/ShapeFolding.cpp b/src/recipes/ShapeFolding.cpp
index ca6ef5087..72d13ab52 100644
--- a/src/recipes/ShapeFolding.cpp
+++ b/src/recipes/ShapeFolding.cpp
@@ -30,7 +30,7 @@ bool Aidge::constantShapeFolding(std::shared_ptr<GraphView> graph, const std::ve
if (not_shape_present)
return false;
do{
- forwarded = graph->forwardDims(dims, true);
+ forwarded = graph->forwardDims(dims, false);
modified = constantFolding(graph, true);
was_modified = true;
} while(modified);
--
GitLab