commit 52afbbbe7bdc1e501069166df881fef1357f74d7
author William Isaksson <william.isaksson@arm.com> Thu Jun 15 15:31:44 2023 +0000
committer tim.hall <tim.hall@arm.com> Mon Jul 10 16:37:04 2023 +0000
tree 5fcfd8373512466fa63e4f852850b11ac4055508
parent 762d3acb93c597b5299864f9b3ee551107978ee2

MLBEDSW-7752: setting query shapes to Shape4D(0)

Changes query initialization shapes to Shape4D(0,0,0,0) = [0,0,0,0]
instead of Shape4D(0) = [0,1,1,1]. The [0,1,1,1] tensors would affect
performance estimates and are not real.


Change-Id: Ic83b6f6a70c0c904b500f62756e1e125c99856c6
Signed-off-by: William Isaksson <william.isaksson@arm.com>

ethosu/vela/npu_performance.py

diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py
index 5b54538..2325a9c 100644
--- a/ethosu/vela/npu_performance.py
+++ b/ethosu/vela/npu_performance.py
@@ -104,7 +104,7 @@
 class PerformanceQuery:
     def __init__(self, npu_block_type=0):
         self.npu_block_type = npu_block_type
-        self.ifm_shape = Shape4D(0)
+        self.ifm_shape = Shape4D(0, 0, 0, 0)
         self.ifm_format = TensorFormat.NHWC
         self.ifm_memory_area = MemArea.Unknown
         self.ifm2_memory_area = MemArea.Unknown
@@ -112,11 +112,11 @@
         self.ifm2_bits = 0
         self.ifm2_shape = None
         self.ifm2_format = TensorFormat.NHWC
-        self.ofm_shape = Shape4D(0)
+        self.ofm_shape = Shape4D(0, 0, 0, 0)
         self.ofm_format = TensorFormat.NHWC
         self.ofm_memory_area = MemArea.Unknown
         self.ofm_bits = 0
-        self.const_shape = Shape4D(0)
+        self.const_shape = Shape4D(0, 0, 0, 0)
         self.const_memory_area = MemArea.Unknown
         self.kernel = Kernel(1, 1)
         self.config = ArchitectureBlockConfig()