diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2021-01-11 12:52:48 +0100 |
---|---|---|
committer | Diqing Zhong <diqing.zhong@arm.com> | 2021-01-19 11:29:05 +0100 |
commit | db5124c2b5e10b34c61b3e016bb597ba1c1574df (patch) | |
tree | 82bbd348c6a0ff9d3d4e6a44067c032922385d0b /ethosu/vela/npu_serialisation.py | |
parent | 9a0cff1cd1334f4d3e7dfb542ad0be4f0e71a9de (diff) | |
download | ethos-u-vela-db5124c2b5e10b34c61b3e016bb597ba1c1574df.tar.gz |
MLBEDSW-3144: Add weights compression ratio
- Also removed the original bit_per_element
Change-Id: I51bfbd28e14f316aae2d542bb610a3ed57b8b53b
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/npu_serialisation.py')
-rw-r--r-- | ethosu/vela/npu_serialisation.py | 10 |
1 files changed, 0 insertions, 10 deletions
diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index a11907b2..fc6b96b7 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -72,16 +72,6 @@ def serialise_npu_subgraph_into_tensors(nng, sg, arch, scratch_tens, scratch_fas command_stream_size_bytes = len(payload_bytes) - # Adjust the bits per element calculation to exclude metadata generated by Vela - nng.total_size[flash_area] = nng.total_size.get(flash_area, 0) - flash_size - command_stream_size_bytes - nng.total_elements[flash_area] = nng.total_elements.get(flash_area, 0) - flash_size - command_stream_size_bytes - nng.total_size[scratch_area] = nng.total_size.get(scratch_area, 0) - scratch_size - nng.total_elements[scratch_area] = nng.total_elements.get(scratch_area, 0) - scratch_size - - if scratch_area != scratch_fast_area: - nng.total_size[scratch_fast_area] = nng.total_size.get(scratch_fast_area, 0) - nng.total_elements[scratch_fast_area] = nng.total_elements.get(scratch_fast_area, 0) - if flash_tens == scratch_tens is None: # First Npu subgraph, create scratch and flash tensors sg.scratch_tensor = make_memory_tensor( |