diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-06-23 12:12:56 +0200 |
|---|---|---|
| committer | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-07-07 09:13:42 +0200 |
| commit | e843d3311b8945baa32654af0dccb229b6861438 (patch) | |
| tree | 335db03df5745da1ed68b62e24424f7ae3a32294 /ethosu/vela/mark_tensors.py | |
| parent | 42e4189689a7ded7e2a804f6263a7c588fbb66cd (diff) | |
| download | ethos-u-vela-e843d3311b8945baa32654af0dccb229b6861438.tar.gz | |
MLBEDSW-2548: Fix for Double Buffer size estimate
This will give a worst case estimate of the Double Buffer size in the
Scheduler and it will no longer be able to choose strategies that end
up with a buffer that doesn't fit in SRAM.
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I763731f63c7672679f3b8cd6db65dad03b946ae5
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index 705f839b..5f3a13fa 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -368,7 +368,7 @@ def mark_tensor_format(nng, arch, verbose_tensor_format=False): if src_tens is not None: op = tens.find_npu_op() npu_block_type = op.attrs["npu_block_type"] - weight_compressor.compress_weights(arch, nng, tens, npu_block_type, 32, 32, op.get_dilation_h_w()) + weight_compressor.compress_weights(arch, nng, tens, npu_block_type, 16, 16, op.get_dilation_h_w()) # Alias compressed weights back into source tensor src_tens.copy_compressed_weight_info(tens) |