From 1b1a5d88b5a629258537b166596437f3541bbc62 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 27 Jun 2022 10:38:54 +0100
Subject: [PATCH] [CustomOp] Expose drain of checksum layer for ip generation
---
src/finn/custom_op/fpgadataflow/checksum.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/checksum.py b/src/finn/custom_op/fpgadataflow/checksum.py
index 9d08757ec..9a492aa7d 100644
--- a/src/finn/custom_op/fpgadataflow/checksum.py
+++ b/src/finn/custom_op/fpgadataflow/checksum.py
@@ -300,7 +300,7 @@ class CheckSum(HLSCustomOp):
def blackboxfunction(self):
self.code_gen_dict["$BLACKBOXFUNCTION$"] = [
"""using T = ap_uint<WORD_SIZE>;\n void {}(hls::stream<T> &in0,
- hls::stream<T> &out, ap_uint<32> &chk, ap_uint<1> drain)""".format(
+ hls::stream<T> &out, ap_uint<32> &chk, ap_uint<1> &drain)""".format(
self.onnx_node.name
)
]
@@ -315,6 +315,9 @@ class CheckSum(HLSCustomOp):
self.code_gen_dict["$PRAGMAS$"].append(
"#pragma HLS interface s_axilite port=chk bundle=checksum"
)
+ self.code_gen_dict["$PRAGMAS$"].append(
+ "#pragma HLS interface s_axilite port=drain bundle=checksum"
+ )
self.code_gen_dict["$PRAGMAS$"].append(
"#pragma HLS interface ap_ctrl_none port=return"
)
--
GitLab