commit 465582ccb27861182841e03cb8b40fadb9e8c7d8
author Tim Hall <tim.hall@arm.com> Tue May 26 09:33:14 2020 +0100
committer Tim Hall <tim.hall@arm.com> Thu Jun 18 17:53:52 2020 +0100
tree b46c20f41c89aaca45f8e0bfc15f9a8c59eec980
parent 600351a0669605e740820df34780c3927a0d5559

vela: Fix tensor purpose for some CPU only ops

 - Add support for marking the tensor purpose of CPU only ops such as LESS which mark their input based upon their output

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Ia7898089f0b18ccd4f183e2ef961a67f4d169e4c

ethosu/vela/architecture_features.py

diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py
index b59122e..f1cabdd 100644
--- a/ethosu/vela/architecture_features.py
+++ b/ethosu/vela/architecture_features.py
@@ -243,6 +243,7 @@
 
         self.tensor_storage_mem_area = {
             # permanent mem_area
+            TensorPurpose.Unknown: MemArea.Unknown,
             TensorPurpose.Weights: self.permanent_storage_mem_area,
             TensorPurpose.FeatureMap: self.feature_map_storage_mem_area,
         }

ethosu/vela/mark_tensors.py

diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py
index 5231e86..b557f90 100644
--- a/ethosu/vela/mark_tensors.py
+++ b/ethosu/vela/mark_tensors.py
@@ -314,6 +314,8 @@
             fmt = arch.default_feature_map_format
         elif tens.purpose == TensorPurpose.Weights:
             fmt = arch.default_weight_format
+        elif tens.purpose == TensorPurpose.Unknown:
+            fmt = TensorFormat.Unknown
         else:
             assert 0, "unknown tensor purpose %s" % (tens.purpose,)
         return fmt