commit c2449827ec55f49b6087e3e385fb3c4f6776dc6a
author James Peet <james.peet@arm.com> Mon Jul 19 17:09:16 2021 +0100
committer James Peet <james.peet@arm.com> Mon Jul 26 12:05:57 2021 +0100
tree d8de5a99c17b6e418028429eadc9da3cfb286345
parent 7519d50c993d60faa1ea09e56abfbf17cef23b49

MLBEDSW-4892: Fix weight compression of asymmetrically dilated kernels.

Signed-off-by: James Peet <james.peet@arm.com>
Change-Id: I5bf39aa4f1fb48bcb0423edc4cd1d01f59aac1db

ethosu/vela/weight_compressor.py

diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index 6536143..10a1a6d 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -149,8 +149,8 @@
 
     ifm_ublock = ArchitectureFeatures.accelerator_configs[accelerator].ifm_ublock
     ofm_ublock = ArchitectureFeatures.accelerator_configs[accelerator].ofm_ublock
-    decomp_h = ArchitectureFeatures.SubKernelMax.height // dilation_xy[0]
-    decomp_w = ArchitectureFeatures.SubKernelMax.width // dilation_xy[1]
+    decomp_h = ArchitectureFeatures.SubKernelMax.height // dilation_xy[1]
+    decomp_w = ArchitectureFeatures.SubKernelMax.width // dilation_xy[0]
 
     return mlw_codec.reorder_encode(
         ifm_ublock.depth,