diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-10-01 08:22:10 +0200 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-10-02 12:34:38 +0200 |
| commit | 3010d9b5c90628e07c7d0f0c33e7355b8bc3e19d (patch) | |
| tree | f0383ad0a88785e97ff4cf764f9a27b1be27f1dd /ethosu/vela/weight_compressor.py | |
| parent | 8912c5311d4ccf2ad1bbb57fdc71c1f25037f56e (diff) | |
| download | ethos-u-vela-3010d9b5c90628e07c7d0f0c33e7355b8bc3e19d.tar.gz | |
MLBEDSW-3060 Adjust check if weights fit in sram
When deciding if weights fit sram:
A compression of the weights has been added when a
weight compression test limit makes it impossible to
fit weights in a double buffer in sram.
The worst compression ratio from compression, is used
to decide if weights can be fit in sram.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I9458769866b3f9fc15659185aae09658ed10fb38
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
| -rw-r--r-- | ethosu/vela/weight_compressor.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index c5a3f3fd..8426705a 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -280,7 +280,6 @@ def core_deinterleave(hwio, core, ncores): # Compress the weights def compress_weights(arch, nng, tens, npu_block_type, ofm_block_depth, ofm_depth_step, dilation): assert tens.purpose == TensorPurpose.Weights - assert tens.format == TensorFormat.WeightsCompressed # Check the weight cache if nng.weight_cache is None: |