diff options
| author | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-12-02 15:29:22 +0000 |
|---|---|---|
| committer | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-12-14 11:38:03 +0000 |
| commit | 7a6f8438aaf750380a9fff799ca81ff5c7e2ae43 (patch) | |
| tree | 163c6868ec869ae52a5cfd848207dcf0ef7b212e /ethosu/vela/weight_compressor.py | |
| parent | 9b43f846b144d39bfb0cf16853bf6901c74b6672 (diff) | |
| download | ethos-u-vela-7a6f8438aaf750380a9fff799ca81ff5c7e2ae43.tar.gz | |
MLBEDSW-2066 Improve Exception messages
Minor refactoring to use fstrings.
Improve Error classes to correctly inherit the base class.
Use existing exception classes instead of plain exceptions where it
makes sense.
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: I0941c04e91010da1db77299517a8e2d896371e77
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
| -rw-r--r-- | ethosu/vela/weight_compressor.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index fce17d19..188b16ad 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -452,18 +452,14 @@ def calc_scales_and_pack_biases(tens, arch, ofm_depth_step, rescale_for_faf=Fals for weight_scale in weight_scales ] else: - raise UnsupportedFeatureError( - "Compression of {} is not implemented; tensor: {}".format(ifm_dtype, tens.name) - ) + raise UnsupportedFeatureError(f"Compression of {ifm_dtype} is not implemented; Tensor: '{tens.name}'") else: if ifm_dtype == DataType.uint8: scales = [np.double(ifm_scale * weight_scale * 0x3000) for weight_scale in weight_scales] elif ifm_dtype == DataType.int8 or ifm_dtype == DataType.int16: scales = [(np.double(ifm_scale * 0x3000) * np.double(weight_scale)) for weight_scale in weight_scales] else: - raise UnsupportedFeatureError( - "Compression of {} is not implemented; tensor: {}".format(ifm_dtype, tens.name) - ) + raise UnsupportedFeatureError(f"Compression of {ifm_dtype} is not implemented; Tensor: '{tens.name}'") # quantise all of the weight scales into (scale_factor, shift) if ifm_dtype == DataType.int16: |