From bbbd6e40f155e7616b3ab8bb916ead7d73ce4592 Mon Sep 17 00:00:00 2001
From: Oscar Gustafsson <oscar.gustafsson@gmail.com>
Date: Fri, 12 May 2023 12:00:51 +0200
Subject: [PATCH] Add assign to ProcessingElement
---
b_asic/architecture.py | 52 ++++++++++++++++++++++++++----------------
1 file changed, 32 insertions(+), 20 deletions(-)
diff --git a/b_asic/architecture.py b/b_asic/architecture.py
index d675dded..480da9c4 100644
--- a/b_asic/architecture.py
+++ b/b_asic/architecture.py
@@ -225,6 +225,14 @@ class Resource(HardwareBlock):
def is_assigned(self) -> bool:
return self._assignment is not None
+ def assign(self, heuristic: str = 'left_edge'):
+ """
+ Perform assignment of processes to resource.
+
+ See the specific resource types for more information.
+ """
+ raise NotImplementedError()
+
@property
def content(self) -> plt.Figure:
"""
@@ -277,16 +285,31 @@ class ProcessingElement(Resource):
self._type_name = op_type.type_name()
self._input_count = ops[0].input_count
self._output_count = ops[0].output_count
- self._assignment = list(
- self._collection.split_on_execution_time(heuristic="left_edge")
- )
- if len(self._assignment) > 1:
- raise ValueError("Cannot map ProcessCollection to single ProcessingElement")
+ self.assign()
@property
def processes(self) -> List[OperatorProcess]:
return [cast(OperatorProcess, p) for p in self._collection]
+ def assign(self, heuristic: str = "left_edge") -> None:
+ """
+ Perform assignment of the processes.
+
+ Parameters
+ ----------
+ heuristic : str, default: 'left_edge'
+ The assignment algorithm.
+
+ * 'left_edge': Left-edge algorithm.
+ * 'graph_color': Graph-coloring based on exclusion graph.
+ """
+ self._assignment = list(
+ self._collection.split_on_execution_time(heuristic=heuristic)
+ )
+ if len(self._assignment) > 1:
+ self._assignment = None
+ raise ValueError("Cannot map ProcessCollection to single ProcessingElement")
+
class Memory(Resource):
"""
@@ -347,26 +370,13 @@ class Memory(Resource):
# Add information about the iterator type
return cast(Iterator[MemoryVariable], iter(self._collection))
- def _assign_ram(self, heuristic: str = "left_edge"):
- """
- Perform RAM-type assignment of MemoryVariables in this Memory.
-
- Parameters
- ----------
- heuristic : {'left_edge', 'graph_color'}
- The underlying heuristic to use when performing RAM assignment.
- """
- self._assignment = list(
- self._collection.split_on_execution_time(heuristic=heuristic)
- )
-
def assign(self, heuristic: str = "left_edge") -> None:
"""
Perform assignment of the memory variables.
Parameters
----------
- heuristic : str
+ heuristic : str, default: 'left_edge'
The assignment algorithm. Depending on memory type the following are
available:
@@ -377,7 +387,9 @@ class Memory(Resource):
* ...
"""
if self._memory_type == "RAM":
- self._assign_ram(heuristic=heuristic)
+ self._assignment = self._collection.split_on_execution_time(
+ heuristic=heuristic
+ )
else: # "register"
raise NotImplementedError()
--
GitLab