diff options
| author | Dwight Lidman <dwight.lidman@arm.com> | 2020-08-11 12:10:36 +0200 |
|---|---|---|
| committer | Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> | 2020-08-12 12:45:49 +0000 |
| commit | 1a9d20e1dd1726bc540a42cdd4164752ab78d07e (patch) | |
| tree | 770dc013b466f64abfbf94f3a3173561c9b6d701 | |
| parent | 7ad408bd84e324f49f90c3a4ad01ee5f7ce9f1ba (diff) | |
| download | ethos-u-vela-1a9d20e1dd1726bc540a42cdd4164752ab78d07e.tar.gz | |
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
| -rw-r--r-- | ethosu/vela/architecture_features.py | 1 |
1 files changed, 1 insertions, 0 deletions
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, |