diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py index 7dd45cbc732a50f8f41c1932601308f0dfd77c20..47d69b3e9d4527c9d66569987271004ebbfeb8da 100644 --- a/tests/end2end/test_end2end_cnv_w1a1.py +++ b/tests/end2end/test_end2end_cnv_w1a1.py @@ -167,6 +167,7 @@ def test_end2end_cnv_w1a1_fold_and_tlastmarker(): model.save(build_dir + "/end2end_cnv_w1a1_folded.onnx") +@pytest.mark.slow def test_end2end_cnv_w1a1_gen_hls_ip(): model = ModelWrapper(build_dir + "/end2end_cnv_w1a1_folded.onnx") model = model.transform(PrepareIP(test_fpga_part, target_clk_ns)) @@ -263,6 +264,7 @@ def test_end2end_cnv_w1a1_make_pynq_proj(): model.save(build_dir + "/end2end_cnv_w1a1_pynq_project.onnx") +@pytest.mark.slow def test_end2end_cnv_w1a1_synth_pynq_project(): model = ModelWrapper(build_dir + "/end2end_cnv_w1a1_pynq_project.onnx") model = model.transform(SynthPYNQProject()) diff --git a/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py b/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py index 74cd46549f45b7512a03da450e011c4f2e80e16e..df53524ffb22696067d15d9d4306ded785415956 100644 --- a/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py +++ b/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py @@ -153,6 +153,7 @@ def test_end2end_tfc_w1a1_fold_and_tlastmarker(): model.save(build_dir + "/end2end_tfc_w1a1_folded.onnx") +@pytest.mark.slow def test_end2end_tfc_w1a1_gen_hls_ip(): model = ModelWrapper(build_dir + "/end2end_tfc_w1a1_folded.onnx") model = model.transform(PrepareIP(test_fpga_part, target_clk_ns)) @@ -241,6 +242,7 @@ def test_end2end_tfc_w1a1_make_pynq_proj(): model.save(build_dir + "/end2end_tfc_w1a1_pynq_project.onnx") +@pytest.mark.slow def test_end2end_tfc_w1a1_synth_pynq_project(): model = ModelWrapper(build_dir + "/end2end_tfc_w1a1_pynq_project.onnx") model = model.transform(SynthPYNQProject()) diff --git a/tests/end2end/test_end2end_tfc_w1a2.py b/tests/end2end/test_end2end_tfc_w1a2.py index 5ee2942845c41f4c6705b4ee3ecee89154d9faa9..716a72a6965c145c8ab618777397beab8f8f41bd 100644 --- a/tests/end2end/test_end2end_tfc_w1a2.py +++ b/tests/end2end/test_end2end_tfc_w1a2.py @@ -145,6 +145,7 @@ def test_end2end_tfc_w1a2_fold_and_tlastmarker(): model.save(build_dir + "/end2end_tfc_w1a2_folded.onnx") +@pytest.mark.slow def test_end2end_tfc_w1a2_gen_hls_ip(): model = ModelWrapper(build_dir + "/end2end_tfc_w1a2_folded.onnx") model = model.transform(PrepareIP(test_fpga_part, target_clk_ns)) @@ -233,6 +234,7 @@ def test_end2end_tfc_w1a2_make_pynq_proj(): model.save(build_dir + "/end2end_tfc_w1a2_pynq_project.onnx") +@pytest.mark.slow def test_end2end_tfc_w1a2_synth_pynq_project(): model = ModelWrapper(build_dir + "/end2end_tfc_w1a2_pynq_project.onnx") model = model.transform(SynthPYNQProject()) diff --git a/tests/end2end/test_end2end_tfc_w2a2.py b/tests/end2end/test_end2end_tfc_w2a2.py index 2477318efd1e02b0865dadb40bad1a74ac8ea0b4..f11d065a9d7966cfd10e64cb3feea0a402deb61b 100644 --- a/tests/end2end/test_end2end_tfc_w2a2.py +++ b/tests/end2end/test_end2end_tfc_w2a2.py @@ -145,6 +145,7 @@ def test_end2end_tfc_w2a2_fold_and_tlastmarker(): model.save(build_dir + "/end2end_tfc_w2a2_folded.onnx") +@pytest.mark.slow def test_end2end_tfc_w2a2_gen_hls_ip(): model = ModelWrapper(build_dir + "/end2end_tfc_w2a2_folded.onnx") model = model.transform(PrepareIP(test_fpga_part, target_clk_ns)) @@ -233,6 +234,7 @@ def test_end2end_tfc_w2a2_make_pynq_proj(): model.save(build_dir + "/end2end_tfc_w2a2_pynq_project.onnx") +@pytest.mark.slow def test_end2end_tfc_w2a2_synth_pynq_project(): model = ModelWrapper(build_dir + "/end2end_tfc_w2a2_pynq_project.onnx") model = model.transform(SynthPYNQProject()) diff --git a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py index 02a9acae5e0e90d2a8dfa7d4d4afb03aa11f4239..826a9a09b0696a1b9c9b14f61175fcb34b21b70e 100644 --- a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py +++ b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py @@ -137,6 +137,7 @@ def prepare_inputs(input_tensor): @pytest.mark.parametrize("exec_mode", ["cppsim", "rtlsim"]) # input channel parallelism ("SIMD") @pytest.mark.parametrize("simd", [1, 2]) +@pytest.mark.slow def test_fpgadataflow_slidingwindow(idt, k, ifm_dim, ifm_ch, stride, exec_mode, simd): ofm_dim = int(((ifm_dim - k) / stride) + 1) diff --git a/tests/fpgadataflow/test_fpgadataflow_dwc.py b/tests/fpgadataflow/test_fpgadataflow_dwc.py index a230fb4201673e3bf0a31cf9ec82f21250fd9e40..4380ab50dc018e22df15ec58a39ff47075064de1 100644 --- a/tests/fpgadataflow/test_fpgadataflow_dwc.py +++ b/tests/fpgadataflow/test_fpgadataflow_dwc.py @@ -55,6 +55,7 @@ def prepare_inputs(input_tensor, dt): @pytest.mark.parametrize("OUTWidth", [2, 4]) # finn_dtype @pytest.mark.parametrize("finn_dtype", [DataType.BIPOLAR, DataType.INT2]) +@pytest.mark.slow def test_fpgadataflow_dwc_rtlsim(Shape, INWidth, OUTWidth, finn_dtype): # generate input data diff --git a/tests/fpgadataflow/test_fpgadataflow_fclayer.py b/tests/fpgadataflow/test_fpgadataflow_fclayer.py index 416d96d5dbfa1125d878eb8339ae38f5d572d1ce..e709598e9e97ab5015876c56ca4d670c813fdead 100644 --- a/tests/fpgadataflow/test_fpgadataflow_fclayer.py +++ b/tests/fpgadataflow/test_fpgadataflow_fclayer.py @@ -149,6 +149,7 @@ def prepare_inputs(input_tensor, idt, wdt): @pytest.mark.parametrize("mw", [16]) # HLS matrix height (output features) @pytest.mark.parametrize("mh", [16]) +@pytest.mark.slow def test_fpgadataflow_fclayer_cppsim(mem_mode, idt, wdt, act, nf, sf, mw, mh): if nf == -1: nf = mh @@ -234,6 +235,7 @@ def test_fpgadataflow_fclayer_cppsim(mem_mode, idt, wdt, act, nf, sf, mw, mh): @pytest.mark.parametrize("mw", [16]) # HLS matrix height (output features) @pytest.mark.parametrize("mh", [16]) +@pytest.mark.slow def test_fpgadataflow_fclayer_rtlsim(mem_mode, idt, wdt, act, nf, sf, mw, mh): if nf == -1: nf = mh diff --git a/tests/fpgadataflow/test_fpgadataflow_fifo.py b/tests/fpgadataflow/test_fpgadataflow_fifo.py index 9158a0b0e72017b2468627e4f30fd3432c418d38..9c68f523aa513d342c08ed2dc2163d73a964d00a 100644 --- a/tests/fpgadataflow/test_fpgadataflow_fifo.py +++ b/tests/fpgadataflow/test_fpgadataflow_fifo.py @@ -76,6 +76,7 @@ def prepare_inputs(input_tensor, dt): @pytest.mark.parametrize("depth", [16]) # finn_dtype @pytest.mark.parametrize("finn_dtype", [DataType.BIPOLAR]) # , DataType.INT2]) +@pytest.mark.slow def test_fpgadataflow_fifo_rtlsim(Shape, folded_shape, depth, finn_dtype): # generate input data diff --git a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py index 30b86d639ae52143320dfdfeb25488bae865b4d2..1aaf71ffa5d89e66d0349501d6aafe529fa2a0b4 100644 --- a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py +++ b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py @@ -284,6 +284,7 @@ def test_fpgadataflow_ipstitch_pynq_projgen(): model.save(ip_stitch_model_dir + "/test_fpgadataflow_pynq_projgen.onnx") +@pytest.mark.slow def test_fpgadataflow_ipstitch_pynq_synth(): model = ModelWrapper(ip_stitch_model_dir + "/test_fpgadataflow_pynq_projgen.onnx") model = model.transform(SynthPYNQProject()) diff --git a/tests/fpgadataflow/test_layer_streaming_maxpool_batch.py b/tests/fpgadataflow/test_layer_streaming_maxpool_batch.py index ac4ab33469c7720c3d7b9f30f5d13be888e1439d..958694d1ce5c9e5daa32c0626f41447a11f8e822 100644 --- a/tests/fpgadataflow/test_layer_streaming_maxpool_batch.py +++ b/tests/fpgadataflow/test_layer_streaming_maxpool_batch.py @@ -121,6 +121,7 @@ def prepare_inputs(input_tensor): @pytest.mark.parametrize("ifm_ch", [1, 2]) # , 2, 3, 4]) # execution mode @pytest.mark.parametrize("exec_mode", ["rtlsim", "cppsim"]) +@pytest.mark.slow def test_fpgadataflow_streamingmaxpool(idt, k, ifm_dim, ifm_ch, exec_mode): stride = k ofm_dim = int(((ifm_dim - k) / stride) + 1)