From 9d7de20465b2833200cb4a10f1d5c661928a60a4 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 21 Apr 2020 10:48:46 +0100
Subject: [PATCH] [Test] respect mem_mode in test_end2end_cnv_w1a1
---
tests/end2end/test_end2end_cnv_w1a1.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py
index 1a59191a0..75dc6c84f 100644
--- a/tests/end2end/test_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_end2end_cnv_w1a1.py
@@ -112,8 +112,8 @@ def test_end2end_cnv_w1a1_streamline():
def test_end2end_cnv_w1a1_convert_to_hls_layers():
model = ModelWrapper(build_dir + "/end2end_cnv_w1a1_streamlined.onnx")
- model = model.transform(to_hls.InferBinaryStreamingFCLayer())
- model = model.transform(to_hls.InferQuantizedStreamingFCLayer())
+ model = model.transform(to_hls.InferBinaryStreamingFCLayer(mem_mode))
+ model = model.transform(to_hls.InferQuantizedStreamingFCLayer(mem_mode))
model = model.transform(to_hls.InferConvInpGen())
model = model.transform(to_hls.InferStreamingMaxPool())
model = model.transform(MoveReshape())
--
GitLab