Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

examples/dataflow/structuring: test Cast

This commit is contained in:
Sebastien Bourdeauducq 2012-06-20 18:29:03 +02:00
parent 6fac3f027f
commit 9a38c47048
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
examples/dataflow/structuring.py
View File

@ -8,7 +8,6 @@ pack_factor = 5
def source_gen(): def source_gen():
for i in range(80): for i in range(80):
#print("==> " + str(i))
yield Token("source", {"value": i}) yield Token("source", {"value": i})
def sink_gen(): def sink_gen():
@ -19,15 +18,23 @@ def sink_gen():
def main(): def main():
base_layout = [("value", BV(32))] base_layout = [("value", BV(32))]
packed_layout = structuring.pack_layout(base_layout, pack_factor)
rawbits_layout = [("value", BV(32*pack_factor))]
source = ActorNode(SimActor(source_gen(), ("source", Source, base_layout))) source = ActorNode(SimActor(source_gen(), ("source", Source, base_layout)))
packer = ActorNode(structuring.Pack(base_layout, pack_factor))
unpacker = ActorNode(structuring.Unpack(pack_factor, base_layout))
sink = ActorNode(SimActor(sink_gen(), ("sink", Sink, base_layout))) sink = ActorNode(SimActor(sink_gen(), ("sink", Sink, base_layout)))
# A tortuous way of passing integer tokens.
packer = ActorNode(structuring.Pack(base_layout, pack_factor))
to_raw = ActorNode(structuring.Cast(packed_layout, rawbits_layout))
from_raw = ActorNode(structuring.Cast(rawbits_layout, packed_layout))
unpacker = ActorNode(structuring.Unpack(pack_factor, base_layout))
g = DataFlowGraph() g = DataFlowGraph()
g.add_connection(source, packer) g.add_connection(source, packer)
g.add_connection(packer, unpacker) g.add_connection(packer, to_raw)
g.add_connection(to_raw, from_raw)
g.add_connection(from_raw, unpacker)
g.add_connection(unpacker, sink) g.add_connection(unpacker, sink)
comp = CompositeActor(g) comp = CompositeActor(g)