diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py index e78812b21a03baa6963f1f0efaefdb4c73e4d0db..9112ae7ef026c637e7e5a6375f8e9d17b918d63b 100644 --- a/tests/brevitas/test_brevitas_avg_pool_export.py +++ b/tests/brevitas/test_brevitas_avg_pool_export.py @@ -45,7 +45,7 @@ def test_brevitas_avg_pool_export( scale = np.ones((1, channels, 1, 1)) output_scale = torch.from_numpy(scale).float() input_quant_tensor = pack_quant_tensor( - tensor=input_tensor, scale=output_scale, bit_width=ibw_tensor + tensor=input_tensor, scale=output_scale, bit_width=ibw_tensor, signed=signed ) bo.export_finn_onnx(b_avgpool, ishape, export_onnx_path, input_t=input_quant_tensor) model = ModelWrapper(export_onnx_path) @@ -65,7 +65,7 @@ def test_brevitas_avg_pool_export( inp = gen_finn_dt_tensor(dtype, ishape) input_tensor = torch.from_numpy(inp).float() input_quant_tensor = pack_quant_tensor( - tensor=input_tensor, scale=output_scale, bit_width=ibw_tensor + tensor=input_tensor, scale=output_scale, bit_width=ibw_tensor, signed=signed ) b_avgpool.eval() expected = b_avgpool.forward(input_quant_tensor).tensor.detach().numpy() @@ -84,7 +84,7 @@ def test_brevitas_avg_pool_export( input_tensor = torch.from_numpy(inp_tensor).float() input_scale = torch.from_numpy(scale).float() input_quant_tensor = pack_quant_tensor( - tensor=input_tensor, scale=input_scale, bit_width=ibw_tensor + tensor=input_tensor, scale=input_scale, bit_width=ibw_tensor, signed=signed ) # export again to set the scale values correctly bo.export_finn_onnx(b_avgpool, ishape, export_onnx_path, input_t=input_quant_tensor)