diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2021-01-20 17:23:54 +0100 |
|---|---|---|
| committer | Louis Verhaard <louis.verhaard@arm.com> | 2021-02-12 17:40:21 +0100 |
| commit | d70025250fc49997801ea3a6ce83f2fa29a09d78 (patch) | |
| tree | 07462a32ed30ba9893bb7825e44b9606a400e709 /ethosu/vela/vela.py | |
| parent | ad45f792e699fe6abdc381f62690801aa50bd412 (diff) | |
| download | ethos-u-vela-d70025250fc49997801ea3a6ce83f2fa29a09d78.tar.gz | |
MLBEDSW-3808: Ported search allocator to python2.1.0.rc1
- Straight port of the C++ implementation to python.
- Renamed the allocator from "Search" to "HillClimb"
Change-Id: I50797d541f326d0264daf79bf7866aef32350a60
Signed-off-by: Louis Verhaard <louis.verhaard@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 c4510b18..72710025 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -317,7 +317,7 @@ def main(args=None): ) parser.add_argument( "--tensor-allocator", - default=TensorAllocator.Search, + default=TensorAllocator.HillClimb, type=lambda s: TensorAllocator[s], choices=list(TensorAllocator), help="Tensor Allocator algorithm (default: %(default)s)", |