diff --git a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py index 4c1f5051d161f181e3523382494b129c1007cb77..7f834cf5ccd27f9f95350b584e9b41aef92d9105 100644 --- a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py +++ b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py @@ -171,14 +171,14 @@ class ConvolutionInputGenerator(HLSCustomOp): def blackboxfunction(self): self.code_gen_dict["$BLACKBOXFUNCTION$"] = [ - """void {}(hls::stream<ap_uint<SIMD1*Input_precision1>> &in, + """void {}(hls::stream<ap_uint<SIMD1*Input_precision1>> &in0, hls::stream<ap_uint<SIMD1*Input_precision1>> &out)""".format( self.onnx_node.name ) ] def pragmas(self): - self.code_gen_dict["$PRAGMAS$"] = ["#pragma HLS INTERFACE axis port=in"] + self.code_gen_dict["$PRAGMAS$"] = ["#pragma HLS INTERFACE axis port=in0"] self.code_gen_dict["$PRAGMAS$"].append("#pragma HLS INTERFACE axis port=out") self.code_gen_dict["$PRAGMAS$"].append( "#pragma HLS INTERFACE ap_ctrl_none port=return"