From 5ae1e77de6e79be5e85de4b277cbdaed0f399021 Mon Sep 17 00:00:00 2001
From: JianyiCheng <jc9016@ic.ac.uk>
Date: Sat, 11 Sep 2021 16:44:22 +0100
Subject: [PATCH] Added pragma insertion for array partitioning

---
 python/utils/polybench.py | 1 +
 scripts/pb-flow           | 1 +
 2 files changed, 2 insertions(+)

diff --git a/python/utils/polybench.py b/python/utils/polybench.py
index bf5e49e0db3..ed559c6ffb2 100644
--- a/python/utils/polybench.py
+++ b/python/utils/polybench.py
@@ -982,6 +982,7 @@ def vitis_opt(self):
             '-xlnnames="{}"'.format(",".join(xln_names)),
             "-strip-attr",
             "-xlnunroll",
+            "-xlnarraypartition",
         ]
 
         self.run_command(
diff --git a/scripts/pb-flow b/scripts/pb-flow
index 2bd8fc29e53..fa460d5cc61 100755
--- a/scripts/pb-flow
+++ b/scripts/pb-flow
@@ -165,6 +165,7 @@ function opt_llvm_for_vitis() {
     -xlnanno -xlntop="${top_func}" \
     -strip-attr \
     -xlnunroll \
+    -xlnarraypartition \
     > "${dst_file}"
   
   echo "${dst_file}"