diff --git a/tests/fpgadataflow/test_convert_to_hls_pool_batch.py b/tests/fpgadataflow/test_convert_to_hls_pool_batch.py index 809147be295ce43400ea830a89bcb6beb0538d72..f301b809008f7a754b97793485ff63e5e7d4be3d 100644 --- a/tests/fpgadataflow/test_convert_to_hls_pool_batch.py +++ b/tests/fpgadataflow/test_convert_to_hls_pool_batch.py @@ -115,13 +115,8 @@ def prepare_inputs(input_tensor): return {"inp": input_tensor} -# Note: QuantAvgPool2d with idt = DataType.UINT4 and odt = DataType.UINT8 -# (And in general seems to be a problem when odt.bitwidth() > idt.bitwidth()) -# passes cppsim but fails rtlsim(Verilator). Cosim with same parameters in -# Vivado_HLS passes. - # input datatype -@pytest.mark.parametrize("idt", [DataType.UINT4, DataType.INT4]) +@pytest.mark.parametrize("idt", [DataType.UINT4, DataType.INT4, DataType.INT8]) # output datatype @pytest.mark.parametrize("odt", [DataType.UINT4, DataType.INT4]) # pool configuration: ( k,stride, pad, ifm_dim )