From bdb2e7ee9abae9cca3529924513e71a53901ec48 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 12 Aug 2020 00:51:04 +0200
Subject: [PATCH] [Test] call AnnotateCycles in end2end cnv-w1a1 test
---
tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
index 1a1b21afa..8d0000472 100644
--- a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
@@ -63,6 +63,7 @@ from finn.transformation.move_reshape import RemoveCNVtoFCFlatten
from finn.transformation.lower_convs_to_matmul import LowerConvsToMatMul
from finn.transformation.streamline.reorder import MakeMaxPoolNHWC
from finn.transformation.infer_data_layouts import InferDataLayouts
+from finn.transformation.fpgadataflow.annotate_cycles import AnnotateCycles
build_dir = "/tmp/" + os.environ["FINN_INST_NAME"]
@@ -116,6 +117,7 @@ def test_end2end_zynqbuild_cnv_w1a1_convert_to_hls_layers():
model = model.transform(to_hls.InferQuantizedStreamingFCLayer(mem_mode))
model = model.transform(to_hls.InferConvInpGen())
model = model.transform(to_hls.InferStreamingMaxPool())
+ model = model.transform(GiveUniqueNodeNames())
model = model.transform(RemoveCNVtoFCFlatten())
model = model.transform(InferDataLayouts())
model.save(build_dir + "/end2end_zynqbuild_cnv_w1a1_hls_layers.onnx")
@@ -165,6 +167,7 @@ def test_end2end_zynqbuild_cnv_w1a1_fold():
swg_inst.set_nodeattr("SIMD", simd)
swg_inst.set_nodeattr("inFIFODepth", swg_idepth[i])
model = model.transform(AnnotateResources("estimate"))
+ model = model.transform(AnnotateCycles())
model.save(build_dir + "/end2end_zynqbuild_cnv_w1a1_folded.onnx")
--
GitLab