commit 68df8a1f5469daac53b7a418d92204f7026e4228
author Tim Hall <tim.hall@arm.com> Wed Mar 16 16:51:16 2022 +0000
committer Tim Hall <tim.hall@arm.com> Thu Mar 31 10:26:56 2022 +0100
tree 0db9df020c89e25c20ed4a7a738dedd1b0ceb023
parent d85750702229af97c0b0bbda6e397a23254b6144

vela: Added debug info to external API

 - Added optional name attributes to operators and tensors

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

ethosu/vela/api.py

diff --git a/ethosu/vela/api.py b/ethosu/vela/api.py
index 3382ea9..399fd46 100644
--- a/ethosu/vela/api.py
+++ b/ethosu/vela/api.py
@@ -27,7 +27,7 @@
 
 
 API_VERSION_MAJOR = 1
-API_VERSION_MINOR = 2
+API_VERSION_MINOR = 3
 API_VERSION = f"{API_VERSION_MAJOR}.{API_VERSION_MINOR}"
 
 
@@ -253,6 +253,8 @@
         self.layout: NpuLayout = NpuLayout.NHWC
         # x/y/c strides used by the NPU when traversing the feature map, if None, vela will use default strides
         self.strides: Optional[NpuShape3D] = None
+        # Used for debug
+        self.name: Optional[str] = None
 
 
 class NpuKernel:
@@ -290,6 +292,8 @@
 
     def __init__(self, op_type: NpuOperationType):
         self.op_type = op_type
+        # Used for debug
+        self.name: Optional[str] = None
 
 
 class NpuDmaOperation(NpuOperation):