Skip to content
Snippets Groups Projects
Commit 28b7dfe2 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] bring back test markers into end2end

parent 1527a1c8
No related branches found
No related tags found
No related merge requests found
...@@ -277,6 +277,8 @@ class TestEnd2End: ...@@ -277,6 +277,8 @@ class TestEnd2End:
model = folding_fxn(model) model = folding_fxn(model)
model.save(get_checkpoint_name(topology, wbits, abits, "fold")) model.save(get_checkpoint_name(topology, wbits, abits, "fold"))
@pytest.mark.slow
@pytest.mark.vivado
def test_cppsim(self, topology, wbits, abits): def test_cppsim(self, topology, wbits, abits):
prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold") prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold")
model = load_test_checkpoint_or_skip(prev_chkpt_name) model = load_test_checkpoint_or_skip(prev_chkpt_name)
...@@ -292,6 +294,8 @@ class TestEnd2End: ...@@ -292,6 +294,8 @@ class TestEnd2End:
y = execute_parent(parent_chkpt, cppsim_chkpt, input_tensor_npy) y = execute_parent(parent_chkpt, cppsim_chkpt, input_tensor_npy)
assert np.isclose(y, output_tensor_npy).all() assert np.isclose(y, output_tensor_npy).all()
@pytest.mark.slow
@pytest.mark.vivado
def test_ipgen(self, topology, wbits, abits): def test_ipgen(self, topology, wbits, abits):
prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold") prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold")
model = load_test_checkpoint_or_skip(prev_chkpt_name) model = load_test_checkpoint_or_skip(prev_chkpt_name)
...@@ -301,6 +305,8 @@ class TestEnd2End: ...@@ -301,6 +305,8 @@ class TestEnd2End:
model = model.transform(HLSSynthIP()) model = model.transform(HLSSynthIP())
model.save(get_checkpoint_name(topology, wbits, abits, "ipgen")) model.save(get_checkpoint_name(topology, wbits, abits, "ipgen"))
@pytest.mark.slow
@pytest.mark.vivado
def test_ipstitch_rtlsim(self, topology, wbits, abits): def test_ipstitch_rtlsim(self, topology, wbits, abits):
prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "ipgen") prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "ipgen")
model = load_test_checkpoint_or_skip(prev_chkpt_name) model = load_test_checkpoint_or_skip(prev_chkpt_name)
...@@ -335,6 +341,9 @@ class TestEnd2End: ...@@ -335,6 +341,9 @@ class TestEnd2End:
warnings.warn("Estimated & rtlsim performance: " + str(perf)) warnings.warn("Estimated & rtlsim performance: " + str(perf))
assert np.isclose(y, output_tensor_npy).all() assert np.isclose(y, output_tensor_npy).all()
@pytest.mark.slow
@pytest.mark.vivado
@pytest.mark.vitis
@pytest.mark.parametrize("kind", ["zynq", "alveo"]) @pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_build(self, topology, wbits, abits, kind): def test_build(self, topology, wbits, abits, kind):
if kind == "alveo" and ("VITIS_PATH" not in os.environ): if kind == "alveo" and ("VITIS_PATH" not in os.environ):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment