From abbd98dba7c135906e41a4d0d1f62d27e7513f12 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 22 Jan 2020 23:13:49 +0000
Subject: [PATCH] [Transform] remove PYNQ project during cleanup
---
src/finn/transformation/fpgadataflow/cleanup.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/finn/transformation/fpgadataflow/cleanup.py b/src/finn/transformation/fpgadataflow/cleanup.py
index cb27a3be4..741b69d57 100644
--- a/src/finn/transformation/fpgadataflow/cleanup.py
+++ b/src/finn/transformation/fpgadataflow/cleanup.py
@@ -13,11 +13,16 @@ class CleanUp(Transformation):
super().__init__()
def apply(self, model):
+ # delete PYNQ project, if any
+ vivado_pynq_proj_dir = model.get_metadata_prop("vivado_pynq_proj")
+ if vivado_pynq_proj_dir is not None and os.path.isdir(vivado_pynq_proj_dir):
+ shutil.rmtree(vivado_pynq_proj_dir)
+ model.set_metadata_prop("vivado_pynq_proj", "")
# delete IP stitching project, if any
ipstitch_path = model.get_metadata_prop("vivado_stitch_proj")
if ipstitch_path is not None and os.path.isdir(ipstitch_path):
shutil.rmtree(ipstitch_path)
- model.set_metadata_prop("vivado_stitch_proj", "")
+ model.set_metadata_prop("vivado_stitch_proj", "")
for node in model.graph.node:
op_type = node.op_type
if node.domain == "finn":
--
GitLab