commit d83d2e11d3dff5031fec513ca2aa22c19c9ea4d8
author Manupa Karunaratne <manupa.karunaratne@arm.com> Mon Jul 20 12:05:32 2020 +0100
committer Manupa Karunaratne <manupa.karunaratne@arm.com> Mon Jul 27 13:28:03 2020 +0100
tree 0603685c456b736fe9ed9faf44df65fa89a8af03
parent ebe26c7af49ad277df347dac7c4c05959d83f387

[EXTAPI] refactor weight compression to be used by an external consumer

*lint
*added unit tests
*added typecheck
*added docstring for the api
Change-Id: Ibd4bc40d4381ac40ad2ea3d500b26c4ec565ab07
Signed-off-by: Manupa Karunaratne <manupa.karunaratne@arm.com>

ethosu/vela/vela.py

diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 20bc525..1766750 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -170,7 +170,7 @@
         "--accelerator-config",
         type=str,
         default="ethos-u55-256",
-        choices=list(architecture_features.ArchitectureFeatures.accelerator_configs.keys()),
+        choices=list(architecture_features.Accelerator.member_list()),
         help="Accelerator configuration to use (default: %(default)s)",
     )
     parser.add_argument(