From 87be8851b984e4399589839c21eadc10023a1188 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 23 Jan 2020 15:17:11 +0000
Subject: [PATCH] [TLastMarker] fix out stream name and inst name
---
src/finn/custom_op/fpgadataflow/tlastmarker.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/tlastmarker.py b/src/finn/custom_op/fpgadataflow/tlastmarker.py
index 77b03b018..ac73a2580 100644
--- a/src/finn/custom_op/fpgadataflow/tlastmarker.py
+++ b/src/finn/custom_op/fpgadataflow/tlastmarker.py
@@ -63,7 +63,7 @@ class TLastMarker(HLSCustomOp):
"OutDType t;",
"t.set_data(in0.read());",
"t.set_last(i==(NumIters-1));",
- "output.write(t);",
+ "out.write(t);",
"}"
]
@@ -76,8 +76,8 @@ class TLastMarker(HLSCustomOp):
def blackboxfunction(self):
self.code_gen_dict["$BLACKBOXFUNCTION$"] = [
- """void TLastMarker(hls::stream<ap_uint<StreamWidth> > &in0,
- hls::stream<OutDType> &out)"""
+ """void %s(hls::stream<ap_uint<StreamWidth> > &in0,
+ hls::stream<OutDType> &out)""" % self.onnx_node.name
]
def pragmas(self):
--
GitLab