diff options
| author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-08-25 11:31:47 +0200 |
|---|---|---|
| committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-08-27 16:39:29 +0200 |
| commit | 880e73543120648f08886365a45e8b2ce32d5ff1 (patch) | |
| tree | 420627fb8e7e5000f23f82c52ba6569f4a696813 /ethosu/vela/vela.py | |
| parent | a41cd4de2af1e43b76a2a33d78eeb2d90a88b757 (diff) | |
| download | ethos-u-vela-880e73543120648f08886365a45e8b2ce32d5ff1.tar.gz | |
[MLBEDSW-2846] Do not use NHCWB16 for reduce_sum int32
Added checks for not using NHCWB16 for reduce_sum int32 which makes
int8/uint8 softmax work.
Also enabled softmax graph rewrite by default and fixed a saturation
problem.
Change-Id: Ic01bd9ece7e5c3edb2900b7915cc747efe9e5760
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Diffstat (limited to 'ethosu/vela/vela.py')
| -rw-r--r-- | ethosu/vela/vela.py | 9 |
1 files changed, 1 insertions, 8 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 19080926..91899c28 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -254,18 +254,12 @@ def main(args=None): help="Control if NHCWB16 or NHWC should be used in between cascaded passes (default: %(default)s)", ) parser.add_argument( - "--softmax-support", - type=ast.literal_eval, - default=False, - choices=[True, False], - help="Control if Softmax should be transformed into a set of npu operations (default: %(default)s)", - ) - parser.add_argument( "--weight-estimation-scaling", type=float, default=1.0, help=("Performs an additional scaling of weight compression scale estimate (default: %(default)s)"), ) + args = parser.parse_args(args=args) # Read configuration file @@ -295,7 +289,6 @@ def main(args=None): block_config_limit=args.block_config_limit, global_memory_clock_scale=args.global_memory_clock_scale, max_blockdep=args.max_block_dependency, - softmax_support=args.softmax_support, weight_estimation_scaling=args.weight_estimation_scaling, ) |