Fix bias check

Xilinx · May 14, 2024 · a626478 · a626478
1 parent c85e6ec
commit a626478
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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"