From ef8d6caeb454051008fa850a8e38958da9b563e6 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 18 May 2021 14:19:39 +0100
Subject: [PATCH] [Test] update convert_to_hls_cnv testcase for new
streamlining
---
tests/fpgadataflow/test_convert_to_hls_layers_cnv.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/fpgadataflow/test_convert_to_hls_layers_cnv.py b/tests/fpgadataflow/test_convert_to_hls_layers_cnv.py
index 20751a587..15bf16079 100644
--- a/tests/fpgadataflow/test_convert_to_hls_layers_cnv.py
+++ b/tests/fpgadataflow/test_convert_to_hls_layers_cnv.py
@@ -115,8 +115,8 @@ def test_convert_to_hls_layers_cnv_w1a1(fused_activation):
thr_nodes = model.get_nodes_by_op_type("Thresholding_Batch")
assert len(thr_nodes) == 8
non_finn_nodes = model.get_non_finn_nodes()
- assert len(non_finn_nodes) == 4
- exp_non_finn_nodes = ["Transpose", "Reshape", "Mul", "Add"]
+ assert len(non_finn_nodes) == 5
+ exp_non_finn_nodes = ["Transpose", "Transpose", "Reshape", "Mul", "Add"]
assert [x.op_type for x in non_finn_nodes] == exp_non_finn_nodes
fc_nodes = model.get_nodes_by_op_type("StreamingFCLayer_Batch")
assert len(fc_nodes) == 9
--
GitLab