From 7b90ab65cb535fba55da627b561f82e1a63a5daa Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 24 Sep 2021 14:26:30 +0200
Subject: [PATCH] [Partition] fix dataflow partition location
---
.../fpgadataflow/create_dataflow_partition.py | 9 ++++++---
src/finn/transformation/fpgadataflow/make_zynq_proj.py | 2 +-
src/finn/transformation/fpgadataflow/vitis_build.py | 2 +-
3 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/src/finn/transformation/fpgadataflow/create_dataflow_partition.py b/src/finn/transformation/fpgadataflow/create_dataflow_partition.py
index 9e4637976..9b2577bc2 100644
--- a/src/finn/transformation/fpgadataflow/create_dataflow_partition.py
+++ b/src/finn/transformation/fpgadataflow/create_dataflow_partition.py
@@ -31,7 +31,7 @@ from finn.custom_op.registry import getCustomOp
from finn.transformation.base import Transformation
from finn.transformation.create_generic_partitions import PartitionFromLambda
from finn.transformation.fpgadataflow.externalize_params import ExternalizeParams
-from finn.util.basic import get_by_name
+from finn.util.basic import get_by_name, make_build_dir
class CreateDataflowPartition(Transformation):
@@ -41,9 +41,12 @@ class CreateDataflowPartition(Transformation):
that indicates the filename for the second graph that only contains
dataflow nodes. No action is taken if there are no dataflow nodes."""
- def __init__(self, partition_model_dir="dataflow_partition"):
+ def __init__(self, partition_model_dir=None):
super().__init__()
- self.partition_model_dir = partition_model_dir
+ if partition_model_dir is None:
+ self.partition_model_dir = make_build_dir("dataflow_partition_")
+ else:
+ self.partition_model_dir = partition_model_dir
def apply(self, model):
def filter_fc_extw(x):
diff --git a/src/finn/transformation/fpgadataflow/make_zynq_proj.py b/src/finn/transformation/fpgadataflow/make_zynq_proj.py
index 0fce73f8c..84d587b6c 100644
--- a/src/finn/transformation/fpgadataflow/make_zynq_proj.py
+++ b/src/finn/transformation/fpgadataflow/make_zynq_proj.py
@@ -308,7 +308,7 @@ class ZynqBuild(Transformation):
platform,
period_ns,
enable_debug=False,
- partition_model_dir="dataflow_partition",
+ partition_model_dir=None,
):
super().__init__()
self.fpga_part = pynq_part_map[platform]
diff --git a/src/finn/transformation/fpgadataflow/vitis_build.py b/src/finn/transformation/fpgadataflow/vitis_build.py
index 3a29082d6..a28653214 100644
--- a/src/finn/transformation/fpgadataflow/vitis_build.py
+++ b/src/finn/transformation/fpgadataflow/vitis_build.py
@@ -378,7 +378,7 @@ class VitisBuild(Transformation):
enable_debug=False,
floorplan_file=None,
enable_link=True,
- partition_model_dir="dataflow_partition",
+ partition_model_dir=None,
):
super().__init__()
self.fpga_part = fpga_part
--
GitLab