Fuse bn

• Fix Add + Mul recipies
• Add fuseBN recipies
• Fix parameters issue #8 (closed)
• Update tensor to get/set values
• Add approxEq Tensor utils method to check if two tensor are approximatly equals
• Update the way scheduler is updated
• Add unit test for:
  • fuse Add + Mul
  • remove padding
  • Add unit test for fuseBN recipies (TODO in aidge_backend_cpu because of dependancie to the backend for the test)

aidge_core/unit_tests/test_recipies.py

@@ -21,7 +21,7 @@ class test_parameters(unittest.TestCase):
     def tearDown(self):
         pass
 
-    def test_conv(self):
+    def test_remove_flatten(self):
         graph_view = aidge_core.sequential([
             aidge_core.GenericOperator("Flatten", 1, 1, 1, name="Flatten0"),
             aidge_core.FC(50, name='0')
@@ -33,6 +33,45 @@ class test_parameters(unittest.TestCase):
 
         self.assertTrue(all([i in old_nodes for i in graph_view.get_nodes()]))
 
+    def test_fuse_matmul_add(self):
+        matmul0 = aidge_core.GenericOperator("MatMul", 1, 2, 1, name="MatMul0")
+        add0 = aidge_core.Add(name="Add0")
+        matmul1 = aidge_core.GenericOperator("MatMul", 1, 2, 1, name="MatMul1")
+        add1 = aidge_core.Add(name="Add1")
+
+        graph_view = aidge_core.sequential([matmul0, add0, matmul1, add1])
+
+        w0 = aidge_core.Producer([1, 1], name="W0")
+        w0.add_child(matmul0, 0, 1)
+        graph_view.add(w0)
+
+        b0 = aidge_core.Producer([1], name="B0")
+        b0.add_child(add0, 0, 1)
+        graph_view.add(b0)
+
+        w1 = aidge_core.Producer([1, 1], name="W1")
+        w1.add_child(matmul1, 0, 1)
+        graph_view.add(w1)
+
+        b1 = aidge_core.Producer([1], name="B1")
+        b1.add_child(add1, 0, 1)
+        graph_view.add(b1)
+
+        graph_view.save("matmul")
+        old_nodes = graph_view.get_nodes()
+        aidge_core.fuse_mul_add(graph_view)
+
+        self.assertTrue(len(graph_view.get_nodes()) == len(old_nodes) - 2)
+        self.assertTrue("MatMul0" not in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("Add0" not in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("MatMul1" not in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("Add1" not in [i.name for i in graph_view.get_nodes()])
+
+        self.assertTrue("W0" in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("B0" in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("W1" in [i.name for i in graph_view.get_nodes()])
+        self.assertTrue("B1" in [i.name for i in graph_view.get_nodes()])
+        # TODO : Vérifier que FC bien crée
 
 if __name__ == '__main__':
     unittest.main()