From f7c567b32ad675ac0eac2de283e61f2cc6fb76c8 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 22 Jan 2025 18:43:48 +0000 Subject: [PATCH] fix a2q --- src/brevitas/quant/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/brevitas/quant/base.py b/src/brevitas/quant/base.py index 790ceed32..e27cd1094 100644 --- a/src/brevitas/quant/base.py +++ b/src/brevitas/quant/base.py @@ -363,6 +363,7 @@ class AccumulatorAwarePerChannelPreNorm(PerChannelPreNorm): pre_scaling_impl = AccumulatorAwareParameterPreScaling accumulator_bit_width = (this << 1).accumulator_bit_width accumulator_bit_width_impl = (this << 1).accumulator_bit_width_impl + restrict_pre_scaling_impl = (this << 1).restrict_pre_scaling_impl class AccumulatorAwareZeroCenterPerChannelPreNorm(AccumulatorAwarePerChannelPreNorm):