diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-09-15 14:05:38 +0200 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-09-25 08:32:55 +0200 |
commit | 0b9c9a3873da3d368e184308f4f9a4c202e3fb67 (patch) | |
tree | f057bfeacd6c6323cd0e7f33cd7ec33f941b4f8d /ethosu/vela/tensor.py | |
parent | 8854dc9088586e1bb0bf2640b2289903dfa3c822 (diff) | |
download | ethos-u-vela-0b9c9a3873da3d368e184308f4f9a4c202e3fb67.tar.gz |
MLBEDSW-2337: Intermediate feature maps in fast storage
Attempts to use fast storage for feature maps used in between
cascaded passes.
This is only relevant for system configurations where feature maps
are by default not placed in SRAM, but there is SRAM for fast storage.
Change-Id: I207b7cf32cfcb5bea3e6b93c2da1161c4af5221d
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index eedbadad..c0786bfc 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -278,7 +278,7 @@ class TensorAddressMap: def set_address_for_tens(cls, tens_id, mem_type, address): # Check previous address if there is one previous_address = cls.address_map[tens_id].get(mem_type) - if previous_address is not None: + if address is not None and previous_address is not None: assert previous_address == address, "Two different addresses cannot be assigned to the same tensor." # Set tensor's address for memory type |