diff --git a/src/finn/custom_op/fpgadataflow/pool_batch.py b/src/finn/custom_op/fpgadataflow/pool_batch.py index 946611094048d32c1d5b8e9d3327f9d9b0ef8a32..fea1ce3f6fae9d243e4dddb5c6324a0c51e6eaf3 100644 --- a/src/finn/custom_op/fpgadataflow/pool_batch.py +++ b/src/finn/custom_op/fpgadataflow/pool_batch.py @@ -200,6 +200,7 @@ class Pool_Batch(HLSCustomOp): return info_messages def global_includes(self): + self.code_gen_dict["$GLOBALS$"] = ['#include "activations.hpp"'] self.code_gen_dict["$GLOBALS$"] = ['#include "maxpool.h"'] self.code_gen_dict["$GLOBALS$"] += ['#include "pool.hpp"']