Skip to content
Snippets Groups Projects
Commit eb313ed3 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Fix python unit test with fuse mul add changes.

parent 70363b47
No related branches found
No related tags found
2 merge requests!152Update Aidge export to take a graph view has an argument instead of a...,!137Fix fuse mulAdd (and more ;))
Pipeline #48537 passed
...@@ -49,23 +49,19 @@ class test_recipes(unittest.TestCase): ...@@ -49,23 +49,19 @@ class test_recipes(unittest.TestCase):
add0 = aidge_core.Add(2, name="Add0") add0 = aidge_core.Add(2, name="Add0")
matmul1 = aidge_core.MatMul(name="MatMul1") matmul1 = aidge_core.MatMul(name="MatMul1")
add1 = aidge_core.Add(2, name="Add1") add1 = aidge_core.Add(2, name="Add1")
graph_view = aidge_core.sequential([matmul0, add0, matmul1, add1])
w0 = aidge_core.Producer([1, 1], name="W0") w0 = aidge_core.Producer([1, 1], name="W0")
w0.add_child(matmul0, 0, 1) w0.add_child(matmul0, 0, 0)
graph_view.add(w0)
b0 = aidge_core.Producer([1], name="B0") b0 = aidge_core.Producer([1], name="B0")
b0.add_child(add0, 0, 1) b0.add_child(add0, 0, 1)
graph_view.add(b0)
w1 = aidge_core.Producer([1, 1], name="W1") w1 = aidge_core.Producer([1, 1], name="W1")
w1.add_child(matmul1, 0, 1) w1.add_child(matmul1, 0, 0)
graph_view.add(w1)
b1 = aidge_core.Producer([1], name="B1") b1 = aidge_core.Producer([1], name="B1")
b1.add_child(add1, 0, 1) b1.add_child(add1, 0, 1)
graph_view = aidge_core.sequential([matmul0, add0, matmul1, add1])
graph_view.add(w0)
graph_view.add(b0)
graph_view.add(w1)
graph_view.add(b1) graph_view.add(b1)
old_nodes = graph_view.get_nodes() old_nodes = graph_view.get_nodes()
......
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