usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

back.rtlil: only translate switch tests once.

Browse source 
This seems to affect synthesis with Yosys but only marginally.
It is mostly a speed and readability improvement.
This commit is contained in:
whitequark 2018-12-23 07:17:33 +00:00
parent 4e49772f67
commit 45a474788c
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
nmigen/back/rtlil.py
View file

@ -529,6 +529,8 @@ class _StatementCompiler(xfrm.StatementVisitor):
self._group = None self._group = None
self._case = None self._case = None
self._test_cache = {}
@contextmanager @contextmanager
def case(self, switch, value): def case(self, switch, value):
try: try:
@ -556,7 +558,11 @@ class _StatementCompiler(xfrm.StatementVisitor):
self._case.assign(self.lhs_compiler(stmt.lhs), rhs_sigspec) self._case.assign(self.lhs_compiler(stmt.lhs), rhs_sigspec)
def on_Switch(self, stmt): def on_Switch(self, stmt):
with self._case.switch(self.rhs_compiler(stmt.test)) as switch: if stmt not in self._test_cache:
self._test_cache[stmt] = self.rhs_compiler(stmt.test)
test_sigspec = self._test_cache[stmt]
with self._case.switch(test_sigspec) as switch:
for value, stmts in stmt.cases.items(): for value, stmts in stmt.cases.items():
with self.case(switch, value): with self.case(switch, value):
self.on_statements(stmts) self.on_statements(stmts)