diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 3cff5a6af1cf47536d83dfaa94749f737dc539f0..a2a8f13ae11001d591d70553a30849ce103e862b 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -747,14 +747,13 @@ class StreamingFCLayer_Batch(HLSCustomOp): export_wdt = DataType.BINARY else: export_wdt = wdt - wdtype_hls_str = export_wdt.get_hls_datatype_str() self.code_gen_dict["$DOCOMPUTE$"] = [ - """Matrix_Vector_Activate_Stream_Batch<MW1, MH1, SIMD1, PE1, {}, {}, {}, {} > - (in0, out, weights, {}, numReps, {});""".format( + """Matrix_Vector_Activate_Stream_Batch<MW1, MH1, SIMD1, PE1, {}, {}, + {}, {}>(in0, out, weights, {}, numReps, {});""".format( + export_wdt.bitwidth(), tmpl_args["TSrcI"], tmpl_args["TDstI"], tmpl_args["TWeightI"], - wdtype_hls_str, threshs, self.get_nodeattr("resType"), )