Merge pull request #1163 from Xilinx/fix/mvau_hw_op_exec

Extension of execution function of MVAU HW op
Xilinx · Sep 18, 2024 · d575f4c · d575f4c
2 parents 37837df + e22201f
commit d575f4c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py
@@ -130,6 +130,8 @@ def get_nodeattr_types(self):
     def execute_node(self, context, graph):
         node = self.onnx_node
         in_act = context[node.input[0]]
+        # ensure that shape is compatible
+        in_act = in_act.reshape(self.get_normal_input_shape())
         mvau_w_init = [x for x in graph.initializer if x.name == node.input[1]][0]
         mvau_w = np_helper.to_array(mvau_w_init)
         # Matrix multiplication