diff options
author | Manupa Karunaratne <manupa.karunaratne@arm.com> | 2020-07-20 12:05:32 +0100 |
---|---|---|
committer | Manupa Karunaratne <manupa.karunaratne@arm.com> | 2020-07-27 13:28:03 +0100 |
commit | d83d2e11d3dff5031fec513ca2aa22c19c9ea4d8 (patch) | |
tree | 0603685c456b736fe9ed9faf44df65fa89a8af03 /ethosu/vela/vela.py | |
parent | ebe26c7af49ad277df347dac7c4c05959d83f387 (diff) | |
download | ethos-u-vela-d83d2e11d3dff5031fec513ca2aa22c19c9ea4d8.tar.gz |
[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>
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 20bc525b..1766750e 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -170,7 +170,7 @@ def main(args=None): "--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( |