From 26848250f2c346f0b1cd794142b01fa0c5a78908 Mon Sep 17 00:00:00 2001
From: Tobi-Alonso <tobi.alonso@gmail.com>
Date: Tue, 7 Jul 2020 14:29:54 +0100
Subject: [PATCH] [Test] Reduce the number of tests
---
tests/fpgadataflow/test_convert_to_hls_pool_batch.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/tests/fpgadataflow/test_convert_to_hls_pool_batch.py b/tests/fpgadataflow/test_convert_to_hls_pool_batch.py
index f301b8090..aba973051 100644
--- a/tests/fpgadataflow/test_convert_to_hls_pool_batch.py
+++ b/tests/fpgadataflow/test_convert_to_hls_pool_batch.py
@@ -120,13 +120,11 @@ def prepare_inputs(input_tensor):
# output datatype
@pytest.mark.parametrize("odt", [DataType.UINT4, DataType.INT4])
# pool configuration: ( k,stride, pad, ifm_dim )
-@pytest.mark.parametrize(
- "pool_config", [(7, 7, 0, 7), (3, 2, 0, 5), (3, 2, 1, 5), (2, 2, 0, 8)]
-)
+@pytest.mark.parametrize("pool_config", [(7, 7, 0, 7), (3, 2, 1, 5)])
# input channels
-@pytest.mark.parametrize("ifm_ch", [1, 4, 20])
+@pytest.mark.parametrize("ifm_ch", [1, 4])
# number of out channel computed in parallel
-@pytest.mark.parametrize("pe", [1, 4])
+@pytest.mark.parametrize("pe", [1, 2, 4])
# pool type
@pytest.mark.parametrize("op_type", ["QuantAvgPool2d", "MaxPool"])
# execution mode
--
GitLab