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/high_level_command_to_npu_op.py

diff --git a/ethosu/vela/high_level_command_to_npu_op.py b/ethosu/vela/high_level_command_to_npu_op.py
index 8c5525b..e6bfc1c 100644
--- a/ethosu/vela/high_level_command_to_npu_op.py
+++ b/ethosu/vela/high_level_command_to_npu_op.py
@@ -296,6 +296,7 @@
     )
     strides = tens.get_strides(shape4D=op_shape4D)
     fm.strides = NpuShape3D(height=int(strides[2]), width=int(strides[3]), depth=int(strides[1]))
+    fm.name = tens.name
     return fm
 
 
@@ -539,6 +540,7 @@
     npu_op: NpuOperation
     if isinstance(cmd, DMA):
         npu_op = create_dma_op(cmd, arch)
+        npu_op.name = cmd.out_tensor.name
     elif isinstance(cmd, NpuStripe):
         npu_block_type = cmd.ps.primary_op.type.npu_block_type
         if npu_block_type in (NpuBlockType.ConvolutionMxN, NpuBlockType.VectorProduct):
@@ -551,6 +553,7 @@
             npu_op = create_npu_elementwise_op(cmd, arch)
         else:
             assert 0, f"Unknown command type {npu_block_type}"
+        npu_op.name = cmd.ps.primary_op.name
     return npu_op