From c88a96f7a18298767210c394517cfcc1ca9ece5c Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Wed, 10 Jun 2020 09:04:33 +0200 Subject: MLBEDSW-2455: Bug fix int16 elementwise with scalar Bug fix in the generation of the NPU_SET_IFM2_SCALAR parameter. Signed-off-by: Louis Verhaard Change-Id: Ie261a90dcfa61ed269d27a100eb48c58af8a325d --- ethosu/vela/register_command_stream_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 11c0c20d..38ab6c15 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -817,9 +817,9 @@ def generate_register_command_stream(nng, sg, arch, verbose=False): if tens.shape == []: # Empty shape, elementwise constant - ifm2_scalar = tens.quant_values.astype(np.uint8) + ifm2_scalar = tens.quant_values assert ifm2_scalar.size == 1 - emit.cmd0_with_param(cmd0.NPU_SET_IFM2_SCALAR, ifm2_scalar.item(0)) + emit.cmd0_with_param(cmd0.NPU_SET_IFM2_SCALAR, int(ifm2_scalar.item(0))) continue height_0, height_1, width_0, addresses = tens.addresses_for_rolling_buffer( -- cgit v1.2.1