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
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,
}
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