1 from nmigen
.compat
.sim
import run_simulation
2 from nmigen
.cli
import verilog
, rtlil
3 from nmigen
import Module
, Signal
, Cat
, Array
, Const
, Repl
, Elaboratable
4 from nmigen
.lib
.coding
import Decoder
6 from scoreboard
.group_picker
import PriorityPicker
9 class RegDecode(Elaboratable
):
10 """ decodes registers into unary
14 * :wid: register file width
16 def __init__(self
, wid
):
20 self
.enable_i
= Signal(reset_less
=True) # enable decoders
21 self
.dest_i
= Signal(max=wid
, reset_less
=True) # Dest R# in
22 self
.src1_i
= Signal(max=wid
, reset_less
=True) # oper1 R# in
23 self
.src2_i
= Signal(max=wid
, reset_less
=True) # oper2 R# in
26 self
.dest_o
= Signal(wid
, reset_less
=True) # Dest unary out
27 self
.src1_o
= Signal(wid
, reset_less
=True) # oper1 unary out
28 self
.src2_o
= Signal(wid
, reset_less
=True) # oper2 unary out
30 def elaborate(self
, platform
):
32 m
.submodules
.dest_d
= dest_d
= Decoder(self
.reg_width
)
33 m
.submodules
.src1_d
= src1_d
= Decoder(self
.reg_width
)
34 m
.submodules
.src2_d
= src2_d
= Decoder(self
.reg_width
)
36 # dest decoder: write-pending
37 for d
, i
, o
in [(dest_d
, self
.dest_i
, self
.dest_o
),
38 (src1_d
, self
.src1_i
, self
.src1_o
),
39 (src2_d
, self
.src2_i
, self
.src2_o
)]:
41 m
.d
.comb
+= d
.n
.eq(~self
.enable_i
)
59 class IssueUnitGroup(Elaboratable
):
60 """ Manages a batch of Computation Units all of which can do the same task
62 A priority picker will allocate one instruction in this cycle based
63 on whether the others are busy.
65 insn_i indicates to this module that there is an instruction to be
66 issued which this group can handle
68 busy_i is a vector of signals that indicate, in this cycle, which
69 of the units are currently busy.
71 busy_o indicates whether it is "safe to proceed" i.e. whether
72 there is a unit here that can *be* issued an instruction
74 fn_issue_o indicates, out of the available (non-busy) units,
75 which one may be selected
77 def __init__(self
, n_insns
):
78 """ Set up inputs and outputs for the Group
82 * :n_insns: number of instructions in this issue unit.
84 self
.n_insns
= n_insns
87 self
.insn_i
= Signal(reset_less
=True, name
="insn_i")
88 self
.busy_i
= Signal(n_insns
, reset_less
=True, name
="busy_i")
91 self
.fn_issue_o
= Signal(n_insns
, reset_less
=True, name
="fn_issue_o")
92 self
.busy_o
= Signal(reset_less
=True)
94 def elaborate(self
, platform
):
100 m
.submodules
.pick
= pick
= PriorityPicker(self
.n_insns
)
103 allissue
= Signal(self
.n_insns
, reset_less
=True)
105 m
.d
.comb
+= allissue
.eq(Repl(self
.insn_i
, self
.n_insns
))
106 # Pick one (and only one) of the units to proceed in this cycle
107 m
.d
.comb
+= pick
.i
.eq(~self
.busy_i
& allissue
)
109 # "Safe to issue" condition is basically when all units are not busy
110 m
.d
.comb
+= self
.busy_o
.eq(~
((~self
.busy_i
).bool()))
112 # Picker only raises one signal, therefore it's also the fn_issue
113 m
.d
.comb
+= self
.fn_issue_o
.eq(pick
.o
& Repl(~self
.busy_o
, self
.n_insns
))
120 yield self
.fn_issue_o
127 class IssueUnitArray(Elaboratable
):
128 """ Convenience module that amalgamates the issue and busy signals
130 unit issue_i is to be set externally, at the same time as the
133 def __init__(self
, units
):
135 self
.issue_o
= Signal(reset_less
=True)
138 n_insns
+= len(u
.fn_issue_o
)
139 self
.busy_i
= Signal(n_insns
, reset_less
=True)
140 self
.fn_issue_o
= Signal(n_insns
, reset_less
=True)
141 self
.n_insns
= n_insns
143 def elaborate(self
, platform
):
145 for i
, u
in enumerate(self
.units
):
146 setattr(m
.submodules
, "issue%d" % i
, u
)
152 busy_i
.append(u
.busy_i
)
153 g_issue_o
.append(u
.busy_o
)
154 fn_issue_o
.append(u
.fn_issue_o
)
155 m
.d
.comb
+= self
.issue_o
.eq(~
(Cat(*g_issue_o
).bool()))
156 m
.d
.comb
+= self
.fn_issue_o
.eq(Cat(*fn_issue_o
))
157 m
.d
.comb
+= Cat(*busy_i
).eq(self
.busy_i
)
164 yield self
.fn_issue_o
165 yield from self
.units
169 class IssueUnit(Elaboratable
):
170 """ implements 11.4.14 issue unit, p50
174 * :n_insns: number of instructions in this issue unit.
176 def __init__(self
, n_insns
):
177 self
.n_insns
= n_insns
180 self
.insn_i
= Signal(n_insns
, reset_less
=True, name
="insn_i")
181 self
.busy_i
= Signal(n_insns
, reset_less
=True, name
="busy_i")
184 self
.fn_issue_o
= Signal(n_insns
, reset_less
=True, name
="fn_issue_o")
185 self
.g_issue_o
= Signal(reset_less
=True)
187 def elaborate(self
, platform
):
190 if self
.n_insns
== 0:
194 fu_stall
= Signal(reset_less
=True)
197 for i
in range(self
.n_insns
):
198 ib_l
.append(self
.insn_i
[i
] & self
.busy_i
[i
])
199 m
.d
.comb
+= fu_stall
.eq(Cat(*ib_l
).bool())
200 m
.d
.comb
+= self
.g_issue_o
.eq(~
(fu_stall
))
201 for i
in range(self
.n_insns
):
202 m
.d
.comb
+= self
.fn_issue_o
[i
].eq(self
.g_issue_o
& self
.insn_i
[i
])
209 yield self
.fn_issue_o
216 class IntFPIssueUnit(Elaboratable
):
217 def __init__(self
, n_int_insns
, n_fp_insns
):
218 self
.i
= IssueUnit(n_int_insns
)
219 self
.f
= IssueUnit(n_fp_insns
)
220 self
.issue_o
= Signal(reset_less
=True)
222 def elaborate(self
, platform
):
224 m
.submodules
.intissue
= self
.i
225 m
.submodules
.fpissue
= self
.f
227 m
.d
.comb
+= self
.issue_o
.eq(self
.i
.g_issue_o | self
.f
.g_issue_o
)
237 def issue_unit_sim(dut
):
238 yield dut
.dest_i
.eq(1)
239 yield dut
.issue_i
.eq(1)
241 yield dut
.issue_i
.eq(0)
243 yield dut
.src1_i
.eq(1)
244 yield dut
.issue_i
.eq(1)
248 yield dut
.issue_i
.eq(0)
250 yield dut
.go_rd_i
.eq(1)
252 yield dut
.go_rd_i
.eq(0)
254 yield dut
.go_wr_i
.eq(1)
256 yield dut
.go_wr_i
.eq(0)
259 def test_issue_unit():
260 dut
= IssueUnitGroup(3)
261 vl
= rtlil
.convert(dut
, ports
=dut
.ports())
262 with
open("test_issue_unit_group.il", "w") as f
:
265 dut
= IssueUnit(32, 3)
266 vl
= rtlil
.convert(dut
, ports
=dut
.ports())
267 with
open("test_issue_unit.il", "w") as f
:
270 dut
= IntFPIssueUnit(32, 3, 3)
271 vl
= rtlil
.convert(dut
, ports
=dut
.ports())
272 with
open("test_intfp_issue_unit.il", "w") as f
:
275 run_simulation(dut
, issue_unit_sim(dut
), vcd_name
='test_issue_unit.vcd')
277 if __name__
== '__main__':