aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/high_level_command_stream_generator.py
diff options
context:
space:
mode:
authorCharles Xu <charles.xu@arm.com>2020-06-23 12:42:28 +0200
committerCharles Xu <charles.xu@arm.com>2020-07-02 09:38:12 +0200
commit04ce34c07cf7fcd0a8ddb29df6f59846862b24f2 (patch)
treef6e9e5aa5a6e01beb7058c9849f771fa52a75bff /ethosu/vela/high_level_command_stream_generator.py
parent5f47c0511d3fc6b54aba331b58c85c3970b61718 (diff)
downloadethos-u-vela-04ce34c07cf7fcd0a8ddb29df6f59846862b24f2.tar.gz
MLBEDSW-2340: Make the tensor address default None
Signed-off-by: Charles Xu <charles.xu@arm.com> Change-Id: I53d9d56acee57cff208dccb4822c1f1a461c416d
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r--ethosu/vela/high_level_command_stream_generator.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py
index d02fd85d..232a56c1 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -216,7 +216,7 @@ def generate_high_level_command_stream_for_pass(strat, passes, block_configs, id
if len(passes) == 1:
# no cascading, can just issue one big stripe
# but only if we've done allocation and OFM does not overlap IFM
- if ifm_tensor.address != -1 and ofm_tensor.address != -1:
+ if ifm_tensor.address is not None and ofm_tensor.address is not None:
if (
ifm_tensor.address + ifm_tensor.storage_size() <= ofm_tensor.address
or ofm_tensor.address + ofm_tensor.storage_size() <= ifm_tensor.address