commit 1a9d20e1dd1726bc540a42cdd4164752ab78d07e
author Dwight Lidman <dwight.lidman@arm.com> Tue Aug 11 12:10:36 2020 +0200
committer Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> Wed Aug 12 12:45:49 2020 +0000
tree 770dc013b466f64abfbf94f3a3173561c9b6d701
parent 7ad408bd84e324f49f90c3a4ad01ee5f7ce9f1ba

MLBEDSW-2726: Vela crashes when marking tensor with TensorPurpose.Unknown

This commit adds a missing entry for TensorPurpose.Unknown,
mapping to MemType.Unknown in the tensor_storage_mem_type
dictionary in the ArchitectureFeatures class in
architecture_features.py

Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I6c3d942e8c6f1c71c6496bdd621ca8d46ea76147

ethosu/vela/architecture_features.py

diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py
index 822bc11..021597e 100644
--- a/ethosu/vela/architecture_features.py
+++ b/ethosu/vela/architecture_features.py
@@ -268,6 +268,7 @@
         }
 
         self.tensor_storage_mem_type = {
+            TensorPurpose.Unknown: MemType.Unknown,
             TensorPurpose.Weights: MemType.Permanent_NPU,
             TensorPurpose.FeatureMap: MemType.Scratch,
             TensorPurpose.LUT: MemType.Scratch,