MLBEDSW-2804: Removed invalid assert for shift range
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I857aeb7aeb34f4b8ea47e6ac954cead268335e32
diff --git a/ethosu/vela/scaling.py b/ethosu/vela/scaling.py
index fcf054e..74f2166 100644
--- a/ethosu/vela/scaling.py
+++ b/ethosu/vela/scaling.py
@@ -46,6 +46,10 @@
reduced_multiplier = int((multiplier + (1 << 15)) >> 16)
reduced_shift = shift - 16
+ if not (0 <= shift < (1 << 6)):
+ # Shift outside of valid range, set scale to 0
+ return 0, 16
+
return reduced_multiplier, reduced_shift
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index a275e41..175646b 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -460,9 +460,6 @@
else:
quantised_scales = [quantise_scale(scale) for scale in scales]
- for _, shift in quantised_scales:
- assert shift >= 16
-
# pack the biases and scales
if len(quantised_scales) == 1:
# If only 1 quantised scale is used, repeat that value for the length of the biases