diff --git a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py index a1dff7a0ad798189e732a4da3554769bce0a503e..2ac9ad2867a389a4006df3c405277eb0422118f8 100644 --- a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py +++ b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py @@ -652,8 +652,7 @@ class MatrixVectorActivation(HLSCustomOp): def minimize_weight_bit_width(self, model): """Minimize the bit width based on the values of the weights""" - runtime_writable = self.get_nodeattr("runtime_writeable_weights") == 0 - if runtime_writable: + if not self.get_nodeattr("runtime_writeable_weights"): weights = model.get_initializer(self.onnx_node.input[1]) w_min = weights.min() w_max = weights.max() diff --git a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py index 5d97244e5b59b8bf17b42aa57b2a789772f6adb5..fd74a7b0c9e4ac74f025c21448635a0aec524c86 100644 --- a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py +++ b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py @@ -171,8 +171,7 @@ class VectorVectorActivation(HLSCustomOp): def minimize_weight_bit_width(self, model): """Minimize the bit width based on the values of the weights""" - runtime_writable = self.get_nodeattr("runtime_writeable_weights") == 0 - if runtime_writable: + if not self.get_nodeattr("runtime_writeable_weights"): weights = model.get_initializer(self.onnx_node.input[1]) w_min = weights.min() w_max = weights.max()