Skip to content
Snippets Groups Projects
Commit d68d26bd authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Pool] correctly add all headers for codegen

parent ddb57552
No related branches found
No related tags found
No related merge requests found
......@@ -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):
......
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