From 1a9d20e1dd1726bc540a42cdd4164752ab78d07e Mon Sep 17 00:00:00 2001 From: Dwight Lidman Date: Tue, 11 Aug 2020 12:10:36 +0200 Subject: 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 Change-Id: I6c3d942e8c6f1c71c6496bdd621ca8d46ea76147 --- ethosu/vela/architecture_features.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py index 822bc115..021597e6 100644 --- a/ethosu/vela/architecture_features.py +++ b/ethosu/vela/architecture_features.py @@ -268,6 +268,7 @@ Note the difference between ArchitectureFeatures and CompilerOptions } self.tensor_storage_mem_type = { + TensorPurpose.Unknown: MemType.Unknown, TensorPurpose.Weights: MemType.Permanent_NPU, TensorPurpose.FeatureMap: MemType.Scratch, TensorPurpose.LUT: MemType.Scratch, -- cgit v1.2.1