From 2ed7c7703862e67eeb003e3c763a5b35655d8124 Mon Sep 17 00:00:00 2001
From: Giuseppe Franco <giuseppefranco4@gmail.com>
Date: Thu, 11 Apr 2024 13:22:13 +0100
Subject: [PATCH] Fix bias check

---
 .../imagenet_classification/ptq/ptq_common.py               | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
index f5bae76d8..37e16399a 100644
--- a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
+++ b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -185,15 +185,15 @@ def quantize_model(
     # We check all of the provided values are positive integers
     check_positive_int(
         weight_bit_width,
-        act_bit_width,  # bias_bit_width,
+        act_bit_width,
+        bias_bit_width,
         layerwise_first_last_bit_width,
         layerwise_first_last_mantissa_bit_width,
         layerwise_first_last_exponent_bit_width,
         weight_mantissa_bit_width,
         weight_exponent_bit_width,
         act_mantissa_bit_width,
-        act_exponent_bit_width,
-    )
+        act_exponent_bit_width)
 
     if act_scale_computation_type == 'dynamic':
         assert bias_bit_width is None, "Bias quantization is not supported with dynamic activation quantization"