From d3fc9948263ff1cb85aada25571153fb86ca4d94 Mon Sep 17 00:00:00 2001
From: Timothy Costigan <156176839+costigt-dev@users.noreply.github.com>
Date: Wed, 21 Feb 2024 13:29:02 +0000
Subject: [PATCH] Fix (export): fix interface for
 QCDQDecoupledWeightQuantProxyHandlerMixin (#870)

---
 src/brevitas/export/common/handler/qcdq.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/brevitas/export/common/handler/qcdq.py b/src/brevitas/export/common/handler/qcdq.py
index b8f3c1fc8..c4659ac87 100644
--- 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(