diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 6dd55b463cac755a96b32af90b29270e99f412c1..8d3d063c7a35bb86c72f1fcc1886c1821dd10c6a 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -516,20 +516,17 @@ class StreamingFCLayer_Batch(HLSCustomOp): "#pragma HLS INTERFACE ap_ctrl_none port=return" ) self.code_gen_dict["$PRAGMAS$"].append( - """DO_PRAGMA(HLS ARRAY_PARTITION - variable=weights complete dim=1)""" + "DO_PRAGMA(HLS ARRAY_PARTITION variable=weights complete dim=1)" ) + self.code_gen_dict["$PRAGMAS$"].append( - """DO_PRAGMA(HLS ARRAY_PARTITION - variable=weights complete dim=2)""" + "DO_PRAGMA(HLS ARRAY_PARTITION variable=weights complete dim=2)" ) if self.calc_tmem() != 0: # TODO find a better way of checking for no pregenerated thresholds self.code_gen_dict["$PRAGMAS$"].append( - """DO_PRAGMA(HLS ARRAY_PARTITION - variable=threshs complete dim=1)""" + "DO_PRAGMA(HLS ARRAY_PARTITION variable=threshs complete dim=1)" ) self.code_gen_dict["$PRAGMAS$"].append( - """DO_PRAGMA(HLS ARRAY_PARTITION - variable=threshs complete dim=3)""" + "DO_PRAGMA(HLS ARRAY_PARTITION variable=threshs complete dim=3)" )