Skip to content

Commit

Permalink
return args and precommit
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe5 committed Oct 30, 2024
1 parent 5cc8b95 commit c5ac133
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/brevitas/export/onnx/qonnx/handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,8 @@ def prepare_for_export(self, module):

def symbolic_execution(self, x: Tensor):
x = BrevitasFloatQuantFn.apply(x, *self.symbolic_kwargs.values())
return x, *self.return_args.values()
return_args = (x, *self.return_args.values())
return return_args


class BrevitasWeightFloatQuantProxyHandler(BrevitasFloatQuantProxyHandler):
Expand Down

0 comments on commit c5ac133

Please sign in to comment.