Skip to content
Snippets Groups Projects
Commit f543a334 authored by auphelia's avatar auphelia
Browse files

[Code Generation - IP] Fixed bug in no activation case (Pragma for thresholds)

parent 7d72077e
No related branches found
No related tags found
No related merge requests found
......@@ -523,11 +523,13 @@ class StreamingFCLayer_Batch(HLSCustomOp):
"""DO_PRAGMA(HLS ARRAY_PARTITION
variable=weights complete dim=2)"""
)
self.code_gen_dict["$PRAGMAS$"].append(
"""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)"""
)
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)"""
)
self.code_gen_dict["$PRAGMAS$"].append(
"""DO_PRAGMA(HLS ARRAY_PARTITION
variable=threshs complete dim=3)"""
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment