From d68d26bd04b19eac633fee47d5c6a6b74d567150 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 17 Dec 2021 23:01:14 +0100
Subject: [PATCH] [Pool] correctly add all headers for codegen
---
src/finn/custom_op/fpgadataflow/pool_batch.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/pool_batch.py b/src/finn/custom_op/fpgadataflow/pool_batch.py
index fea1ce3f6..7d0ad4310 100644
--- a/src/finn/custom_op/fpgadataflow/pool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/pool_batch.py
@@ -201,7 +201,7 @@ class Pool_Batch(HLSCustomOp):
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 "maxpool.h"']
self.code_gen_dict["$GLOBALS$"] += ['#include "pool.hpp"']
def defines(self, var):
--
GitLab