From 9dba93d461204ea42367c97f41b3737b099a9589 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 9 Jan 2020 16:26:21 +0000
Subject: [PATCH] [HLSCustomOp] Added new parameters to functions to allow
using the same function during npysim and ipgen
---
src/finn/custom_op/fpgadataflow/__init__.py | 13 ++++++++-----
1 file changed, 8 insertions(+), 5 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/__init__.py b/src/finn/custom_op/fpgadataflow/__init__.py
index 874d53956..d8806ec29 100644
--- a/src/finn/custom_op/fpgadataflow/__init__.py
+++ b/src/finn/custom_op/fpgadataflow/__init__.py
@@ -37,8 +37,10 @@ class HLSCustomOp(CustomOp):
node = self.onnx_node
# generate top cpp file for ip generation
+ path = self.get_nodeattr("code_gen_dir_ipgen")
+ self.generate_params(model, path)
self.global_includes()
- self.defines()
+ self.defines("ipgen")
self.blackboxfunction()
self.pragmas()
self.docompute()
@@ -86,9 +88,10 @@ class HLSCustomOp(CustomOp):
def code_generation_npysim(self, model):
node = self.onnx_node
- self.generate_params(model)
+ path = self.get_nodeattr("code_gen_dir_npysim")
+ self.generate_params(model, path)
self.global_includes()
- self.defines()
+ self.defines("npysim")
self.read_npy_data()
self.strm_decl()
self.docompute()
@@ -170,7 +173,7 @@ compilation transformations?
# load output npy file
self.npy_to_dynamic_output(context)
- def generate_params(self, model):
+ def generate_params(self, model, path):
pass
@abstractmethod
@@ -178,7 +181,7 @@ compilation transformations?
pass
@abstractmethod
- def defines(self):
+ def defines(self, var):
pass
@abstractmethod
--
GitLab