From 5c5e65f4475decdd302b3c767ac80871e11c7298 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 14 Jul 2020 15:38:16 +0100
Subject: [PATCH] [Test] Check if SIMD in padding node is set to input channels
after conversion to hls
---
tests/fpgadataflow/test_convert_to_hls_conv_layer.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
index 22c356a58..188f20e22 100644
--- a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
+++ b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
@@ -23,6 +23,7 @@ from finn.transformation.fpgadataflow.prepare_cppsim import PrepareCppSim
from finn.transformation.fpgadataflow.compile_cppsim import CompileCppSim
from finn.transformation.fpgadataflow.set_exec_mode import SetExecMode
from finn.custom_op.im2col import compute_conv_output_dim
+from finn.custom_op.registry import getCustomOp
# conv_config kernel_size,stride, pad
@@ -110,3 +111,8 @@ def test_convert_to_hls_conv_layer(conv_config, exec_mode):
assert oxe.compare_execution(model, new_model, inp_dict)
if kernel_size == 1 and stride > 1 and pad == 0:
assert new_model.graph.node[1].op_type == "DownSampler"
+
+ if pad == 1:
+ padding_node = new_model.get_nodes_by_op_type("FMPadding_Batch")[0]
+ padding_inst = getCustomOp(padding_node)
+ assert padding_inst.get_nodeattr("SIMD") == in_chn
--
GitLab