diff --git a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py index 7f834cf5ccd27f9f95350b584e9b41aef92d9105..49c7f81078c0b3a4832e052b66e21063619ab543 100644 --- a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py +++ b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py @@ -82,7 +82,7 @@ class ConvolutionInputGenerator(HLSCustomOp): def global_includes(self): self.code_gen_dict["$GLOBALS$"] = ['#include "slidingwindow.h"'] - def defines(self): + def defines(self, var): numReps = 1 self.code_gen_dict["$DEFINES$"] = [ """#define ConvKernelDim1 {}\n #define IFMChannels1 {} diff --git a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py index d331ffe04eb2ea4c445ee4fe1148eed262b23aa7..a316695b3d8691ab66cdef7a87093d7a777ef7ff 100644 --- a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py @@ -72,7 +72,7 @@ class StreamingMaxPool_Batch(HLSCustomOp): def global_includes(self): self.code_gen_dict["$GLOBALS$"] = ['#include "maxpool.h"'] - def defines(self): + def defines(self, var): numReps = 2 self.code_gen_dict["$DEFINES$"] = [ """#define ImgDim {}\n #define PoolDim {}\n @@ -197,3 +197,9 @@ class StreamingMaxPool_Batch(HLSCustomOp): int(self.get_nodeattr("ImgDim") / self.get_nodeattr("PoolDim")), ) ] + + def blackboxfunction(self): + pass + + def pragmas(self): + pass