diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py index 2b9935812ecd8b1b4075df5efab2e9f5f0998ba9..66128de4dfdd048f22fa02d10f8c3e094d43a178 100644 --- a/tests/end2end/test_end2end_bnn_pynq.py +++ b/tests/end2end/test_end2end_bnn_pynq.py @@ -343,6 +343,8 @@ class TestEnd2End: @pytest.mark.vivado @pytest.mark.parametrize("kind", ["zynq", "alveo"]) def test_ipgen(self, topology, wbits, abits, kind): + if kind == "alveo" and ("VITIS_PATH" not in os.environ): + pytest.skip("VITIS_PATH not set") prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold") model = load_test_checkpoint_or_skip(prev_chkpt_name) test_fpga_part = get_build_env(kind, target_clk_ns)["part"] @@ -355,6 +357,8 @@ class TestEnd2End: @pytest.mark.vivado @pytest.mark.parametrize("kind", ["zynq", "alveo"]) def test_ipstitch_rtlsim(self, topology, wbits, abits, kind): + if kind == "alveo" and ("VITIS_PATH" not in os.environ): + pytest.skip("VITIS_PATH not set") prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "ipgen_" + kind) model = load_test_checkpoint_or_skip(prev_chkpt_name) test_fpga_part = get_build_env(kind, target_clk_ns)["part"] @@ -375,7 +379,9 @@ class TestEnd2End: "rtlsim_trace", "%s_w%da%d.vcd" % (topology, wbits, abits) ) os.environ["RTLSIM_TRACE_DEPTH"] = "3" - rtlsim_chkpt = get_checkpoint_name(topology, wbits, abits, "ipstitch_rtlsim_" + kind) + rtlsim_chkpt = get_checkpoint_name( + topology, wbits, abits, "ipstitch_rtlsim_" + kind + ) model.save(rtlsim_chkpt) parent_chkpt = get_checkpoint_name(topology, wbits, abits, "dataflow_parent") (input_tensor_npy, output_tensor_npy) = get_golden_io_pair( @@ -391,7 +397,11 @@ class TestEnd2End: @pytest.mark.vivado @pytest.mark.parametrize("kind", ["zynq", "alveo"]) def test_throughput_rtlsim(self, topology, wbits, abits, kind): - prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "ipstitch_rtlsim_" + kind) + if kind == "alveo" and ("VITIS_PATH" not in os.environ): + pytest.skip("VITIS_PATH not set") + prev_chkpt_name = get_checkpoint_name( + topology, wbits, abits, "ipstitch_rtlsim_" + kind + ) model = load_test_checkpoint_or_skip(prev_chkpt_name) n_nodes = len(model.graph.node) perf_est = model.analysis(dataflow_performance)