Fix

Xilinx · May 17, 2024 · 66615ef · 66615ef
1 parent ede0a9d
commit 66615ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py
@@ -314,8 +314,8 @@ def __init__(
 
     def single_layer_update(self):
         # raise error in case no quant-input is here
-        if self.quant_input is None:
-            raise ValueError('Expected self.quant_input to calculate L1-norm upper bound, but recevied None. ' + \
+        if self.quant_metadata is None:
+            raise ValueError('Expected self.quant_metadata to calculate L1-norm upper bound, but recevied None. ' + \
                 'Make sure that either the input to the model is a IntQuantTensor or the layer has an input quant enabled. ' \
                 'Also, check if `use_quant_activations=True` in `gpfq_mode` when `accumulator_bit_width` is specified. ' + \
                 'Alternatively, provide a custom `a2q_layer_filter_fnc` to `gpfq_mode` to filter layers without a quant_tensor input.')