diff --git a/pyrtl/passes.py b/pyrtl/passes.py index 0b684303..35f365e5 100644 --- a/pyrtl/passes.py +++ b/pyrtl/passes.py @@ -69,10 +69,9 @@ def __setitem__(self, key, item): self.dict[key] = item def find_producer(self, item): - if item in self.dict: - return self.find_producer(self.dict[item]) - else: - return item + while item in self.dict: + item = self.dict[item] + return item def _remove_wire_nets(block, skip_sanity_check=False):