From 793832de5e84b11794b20d98129ba24f30880aa2 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 2 Dec 2019 12:20:02 +0000
Subject: [PATCH] [Compilation Trafo] added function to get the library
includes for the compilation
---
.../compilation_transformation.py | 30 +++++++++++++++++++
1 file changed, 30 insertions(+)
create mode 100644 src/finn/transformation/compilation_transformation.py
diff --git a/src/finn/transformation/compilation_transformation.py b/src/finn/transformation/compilation_transformation.py
new file mode 100644
index 000000000..7205eab9b
--- /dev/null
+++ b/src/finn/transformation/compilation_transformation.py
@@ -0,0 +1,30 @@
+# import os
+# import tempfile as tmp
+
+# import finn.custom_op.registry as registry
+from finn.transformation import Transformation
+
+
+class Compilation(Transformation):
+ """Compilation for all nodes in model"""
+
+ def get_includes(self, node):
+ # step by step addition of include paths to ensure easy extension
+ include_paths = []
+ include_paths.append("/workspace/finn/src/finn/data/cpp")
+ include_paths.append("/workspace/cnpy/")
+ include_paths.append("/workspace/finn-hlslib")
+ include_paths.append("/workspace/vivado-hlslib")
+
+ def apply(self, model):
+ for node in model.graph.node:
+ includes = self.get_includes(node)
+ # bash_compile = """g++ -o {}/execute_{} {}/execute_{}.cpp
+ # /workspace/cnpy/cnpy.cpp -I/workspace/finn/src/finn/data/cpp -I/workspace/cnpy/
+ # -I/workspace/finn-hlslib -I/workspace/vivado-hlslib
+ # --std=c++11 -lz""".format(
+ # self.tmp_dir, node.op_type, self.tmp_dir, node.op_type
+ # )
+ # process_compile = subprocess.Popen(bash_compile.split(), stdout=subprocess.PIPE)
+ # process_compile.communicate()
+ return (model, False)
--
GitLab