Hardware/litex
1
0
Fork 0
mirror of https://github.com/enjoy-digital/litex.git synced 2025-01-04 09:52:26 -05:00
Code Issues Projects Releases Packages Wiki Activity

transform/unroll_sync: support generator function

Browse source
This commit is contained in:
Sebastien Bourdeauducq 2012-10-15 19:42:30 +02:00
parent 9efc581bee
commit fecab5518b
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
migen/transform/unroll.py
View file

@ -80,7 +80,7 @@ def _variable_for(s, n):
name = "v"
return Signal(s.bv, name=name, variable=True)
def unroll_sync(sync, inputs, outputs):
def unroll_sync(statements, inputs, outputs):
assert(inputs or outputs)
if inputs:
sd_in = _list_step_dicts(inputs)
@ -106,7 +106,11 @@ def unroll_sync(sync, inputs, outputs):
# replace signals with intermediate variables and copy statements
io_var_dict.update(di)
r += _replace(sync, io_var_dict, do_var)
if isinstance(statements, list):
sl = statements
else:
sl = statements(n)
r += _replace(sl, io_var_dict, do_var)
# assign to output signals
r += [v.eq(do_var[k]) for k, v in do.items()]