diff options
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 5563b969..faae2cf3 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -861,14 +861,14 @@ def generate_register_command_stream(nng, sg, arch, verbose=False): if ifm_dtype.size_in_bits() == 8: if ifm_dtype.type & BaseType.Signed: - prec = ifm_precision.W8_S8 + prec = ifm_precision.S8 else: - prec = ifm_precision.W8_U8 + prec = ifm_precision.U8 elif ifm_dtype.size_in_bits() == 16: if ifm_dtype.type & BaseType.Signed: - prec = ifm_precision.W8_S16 + prec = ifm_precision.S16 else: - prec = ifm_precision.W8_U16 + prec = ifm_precision.U16 ifm_prec = prec.value ifm2_prec = ifm_prec |