From ddde137e207d3faf4ed77fd4a512cbcb81e2c795 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 31 Jan 2023 22:05:30 +0000
Subject: [PATCH] [Test] Fix check for padding layers in conv test
---
tests/fpgadataflow/test_convert_to_hls_conv_layer.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
index 8c9f110c3..5c5363ca2 100644
--- a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
+++ b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
@@ -175,8 +175,11 @@ def test_convert_to_hls_conv_layer(conv_config, depthwise, use_rtl_swg, exec_mod
assert np.isclose(exp_cycles, cycles_rtlsim, atol=11)
assert exp_cycles != 0
- if pad == 1:
- padding_node = new_model.get_nodes_by_op_type("FMPadding_Batch")[0]
+ if pad:
+ if use_rtl_swg:
+ padding_node = new_model.get_nodes_by_op_type("FMPadding_rtl")[0]
+ else:
+ 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