From c39e085afbe6d59faff3040d0b08d9dc27e3c8db Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 21 May 2020 14:45:45 +0100
Subject: [PATCH] [Test] Set quant type in avg pool export test
---
tests/brevitas/test_brevitas_avg_pool_export.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py
index 01da19b93..d4a5776ed 100644
--- a/tests/brevitas/test_brevitas_avg_pool_export.py
+++ b/tests/brevitas/test_brevitas_avg_pool_export.py
@@ -1,6 +1,7 @@
import onnx # noqa
import brevitas.onnx as bo
from brevitas.nn import QuantAvgPool2d
+from brevitas.core.quant import QuantType
import pytest
export_onnx_path = "test_avg_pool.onnx"
@@ -19,5 +20,6 @@ def test_brevitas_avg_pool_export(kernel_size, stride, signed, bit_width):
signed=signed,
min_overall_bit_width=bit_width,
max_overall_bit_width=bit_width,
+ quant_type=QuantType.INT,
)
bo.export_finn_onnx(b_avgpool, ishape, export_onnx_path)
--
GitLab