diff options
author | Tim Hall <tim.hall@arm.com> | 2020-11-09 16:46:37 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-11-11 11:38:41 +0000 |
commit | e6ccd87a2f40877cacdd9721a5116a6853dfe573 (patch) | |
tree | 8e22dacc02e82df59cb460b68d39e5fd338abf4d /ethosu/vela/pass_packing.py | |
parent | e168b969dc75fc3057413a80fdf0e164ab936910 (diff) | |
download | ethos-u-vela-e6ccd87a2f40877cacdd9721a5116a6853dfe573.tar.gz |
MLBEDSW-3019: Add profiling debug database
- Added mechanism to track input to output graph transforms for
debugging the resultant command stream.
- Provides base implementation for MLBEDSW-2661
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I2dfe8a409fbde7ad0282bfab5acb11ba1c8b82d8
Diffstat (limited to 'ethosu/vela/pass_packing.py')
-rw-r--r-- | ethosu/vela/pass_packing.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 5673c2df..59376a85 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -18,6 +18,7 @@ import collections import enum +from .debug_database import DebugDatabase from .nn_graph import Pass from .nn_graph import PassPlacement from .operation import create_avgpool_nop @@ -430,7 +431,6 @@ def pack_into_passes(nng, arch, verbose_packing=False): # Configure a 1x1 AvgPool and attach the op onto it op = op_list[0] inp = op.inputs[0] - avgpool_op = create_avgpool_nop(op.name + "_avgpool") avgpool_op.add_input_tensor(inp) avgpool_out = inp.clone("_avgpooled") @@ -440,6 +440,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): op.inputs[0] = avgpool_out op_list.insert(0, avgpool_op) + DebugDatabase.add_optimised(op, avgpool_op) return avgpool_op return None |