Fix (export): fix interface for QCDQDecoupledWeightQuantProxyHandlerM…

…ixin (#870)
Xilinx · Feb 21, 2024 · d3fc994 · d3fc994
1 parent 6251f9e
commit d3fc994
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/brevitas/export/common/handler/qcdq.py b/src/brevitas/export/common/handler/qcdq.py
@@ -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(