Skip to content

Commit

Permalink
Fix (export/qonnx): Fixed symbolic kwargs order. (#988)
Browse files Browse the repository at this point in the history
  • Loading branch information
nickfraser committed Jul 22, 2024
1 parent 2004568 commit fc7e7b1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/brevitas/export/onnx/qonnx/handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,9 +72,9 @@ def prepare_for_export(self, module: WeightQuantProxyFromInjector):
'scale': first_qweight.scale,
'zero_point': first_qweight.zero_point,
'bit_width': first_qweight.bit_width,
'signed': first_qweight.signed,
# narrow_range is not a property of the QuantTensor, take it from the proxy instead
'narrow_range': module.is_narrow_range,
'signed': first_qweight.signed,
# override rounding mode since quantization has been pre-applied
'rounding_mode': 'ROUND'}

Expand Down

0 comments on commit fc7e7b1

Please sign in to comment.