diff --git a/b_asic/architecture.py b/b_asic/architecture.py index dc78118a135ca79273735357943a3f6daebd6e2b..cde168556ab59da33c7f369174cfde7a8e7c02c8 100644 --- a/b_asic/architecture.py +++ b/b_asic/architecture.py @@ -764,19 +764,55 @@ of :class:`~b_asic.architecture.ProcessingElement` d_out[i][v] += 1 return [dict(d) for d in d_in], [dict(d) for d in d_out] - def resource_from_name(self, name: str): + def resource_from_name(self, name: str) -> Resource: + """ + Get :class:`Resource` based on name. + + Parameters + ---------- + name : str + Name of the resource. + + Returns + ------- + :class:`Resource` + + """ re = {p.entity_name: p for p in chain(self.memories, self.processing_elements)} return re[name] + def remove_resource( + self, + resource: Union[str, Resource], + ) -> None: + """ + Remove an empty :class:`Resource` from the architecture. + + Parameters + ---------- + resource : :class:`b_asic.architecture.Resource` or str + The resource or the resource name to remove. + """ + if isinstance(resource, str): + resource = self.resource_from_name(resource) + + if resource.collection: + raise ValueError("Resource must be empty") + + if resource in self.memories: + self.memories.remove(resource) + else: + self.processing_elements.remove(resource) + def move_process( self, proc: Union[str, Process], re_from: Union[str, Resource], re_to: Union[str, Resource], assign: bool = False, - ): + ) -> None: """ - Move a :class:`b_asic.process.Process` from one resource to another. + Move a :class:`b_asic.process.Process` from one :class:`Resource` to another. Both the resource moved from and will become unassigned after a process has been moved, unless *assign* is set to True. diff --git a/examples/fivepointwinograddft.py b/examples/fivepointwinograddft.py index b1a0186dfb66e83d2b00d334919e12f8e4dfecab..6d00c8dd93a253d4463dfbb8f04f4ccebd46b85b 100644 --- a/examples/fivepointwinograddft.py +++ b/examples/fivepointwinograddft.py @@ -196,7 +196,12 @@ memories[0].assign() memories[1].assign() memories[4].assign() -for memory in memories: +# %% +# Memory 4 is now empty, so remove it. + +arch.remove_resource('memory4') + +for memory in arch.memories: memory.show_content(title=f"Improved {memory.entity_name}") arch