From 6a0a98934ea7db218959ea58f9dbef35a2f84aa1 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 17 Apr 2020 00:18:21 +0100
Subject: [PATCH] [HLSCustomOp] also emit pragmas for npysim codegen

---
 src/finn/custom_op/fpgadataflow/__init__.py  | 1 +
 src/finn/custom_op/fpgadataflow/templates.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/finn/custom_op/fpgadataflow/__init__.py b/src/finn/custom_op/fpgadataflow/__init__.py
index 4231be7c5..4fc69985f 100644
--- a/src/finn/custom_op/fpgadataflow/__init__.py
+++ b/src/finn/custom_op/fpgadataflow/__init__.py
@@ -166,6 +166,7 @@ class HLSCustomOp(CustomOp):
         self.defines("npysim")
         self.read_npy_data()
         self.strm_decl()
+        self.pragmas()
         self.docompute()
         self.dataoutstrm()
         self.save_as_npy()
diff --git a/src/finn/custom_op/fpgadataflow/templates.py b/src/finn/custom_op/fpgadataflow/templates.py
index c34167feb..5323aac2e 100644
--- a/src/finn/custom_op/fpgadataflow/templates.py
+++ b/src/finn/custom_op/fpgadataflow/templates.py
@@ -42,6 +42,7 @@ $GLOBALS$
 $DEFINES$
 
 int main(){
+$PRAGMAS$
 
 $STREAMDECLARATIONS$
 
-- 
GitLab