Skip to content
Snippets Groups Projects
Commit c0c587e1 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed missing input in unit test

parent 55ee7322
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...,!143Multiple refactors
Pipeline #48729 passed
...@@ -41,6 +41,7 @@ class test_OperatorImpl(unittest.TestCase): ...@@ -41,6 +41,7 @@ class test_OperatorImpl(unittest.TestCase):
generic_matmul_op = matmul.get_operator() generic_matmul_op = matmul.get_operator()
generic_matmul_op.set_forward_dims(lambda x: x) generic_matmul_op.set_forward_dims(lambda x: x)
generic_matmul_op.set_impl(testImpl(generic_matmul_op)) generic_matmul_op.set_impl(testImpl(generic_matmul_op))
generic_matmul_op.set_input(0, aidge_core.Tensor(np.arange(18).reshape(1,2,3,3)))
generic_matmul_op.forward() generic_matmul_op.forward()
self.assertEqual(GLOBAL_CPT, 1) self.assertEqual(GLOBAL_CPT, 1)
......
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