Skip to content

Commit

Permalink
Fix (export): fix interface for QCDQDecoupledWeightQuantProxyHandlerM…
Browse files Browse the repository at this point in the history
…ixin (Xilinx#870)
  • Loading branch information
costigt-dev authored Feb 21, 2024
1 parent 6251f9e commit d3fc994
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/brevitas/export/common/handler/qcdq.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,8 @@ class QCDQCastDecoupledWeightQuantProxyHandlerMixin(QCDQCastWeightQuantProxyHand
def symbolic_execution(self, x: Tensor):
out, scale, zero_point, bit_width = super().symbolic_execution(x)
# Return post-rounding scale and zero-point in place of pre-rounding as a placeholder
return out, scale, zero_point, scale, zero_point, bit_width
# The order of arguments must match the order in the forward method of DecoupledRescalingIntQuant
return out, scale, zero_point, bit_width, scale, zero_point


class QCDQCastDecoupledWeightQuantWithInputProxyHandlerMixin(
Expand Down

0 comments on commit d3fc994

Please sign in to comment.