commit 5ff994417eae32175b5c72dc433edc4a9f12a8cc
author Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Jul 10 10:12:17 2020 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Jul 10 10:12:17 2020 +0200
tree 27807000b40462adce741a63f04cfa2ad4a78a7e
parent d89c09eb8d9f5a705c16ebbdac90876034971765

MLBEDSW-2634 Avoid encoding empty weights stream

Avoid encoding empty weights stream.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I120ede14f19705e169c5f03ed344036a58b5f84f

ethosu/vela/weight_compressor.py

diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index 9300040..8ebd751 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -76,6 +76,8 @@
 
 
 def encode(weight_stream):
+    if len(weight_stream) == 0:
+        return []
     assert np.amin(weight_stream) >= -255
     assert np.amax(weight_stream) <= 255