From 3bc81a6e8ca9c7506a936ff74bfaa50d8d685a18 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 24 Aug 2020 18:15:24 +0200
Subject: [PATCH] [Test] spec signed in pack_quant_tensor
---
tests/brevitas/test_brevitas_avg_pool_export.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py
index e78812b21..9112ae7ef 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)
--
GitLab