src/scoreboard/fu_reg_matrix.py

   1 from nmigen.compat.sim import run_simulation
   2 from nmigen.cli import verilog, rtlil
   3 from nmigen import Module, Signal, Elaboratable, Array, Cat
   4
   5 #from nmutil.latch import SRLatch
   6 from scoreboard.dependence_cell import DependenceCell
   7 from scoreboard.fu_wr_pending import FU_RW_Pend
   8 from scoreboard.reg_select import Reg_Rsv
   9
  10 """
  11
  12  6600 Dependency Table Matrix inputs / outputs
  13  ---------------------------------------------
  14
  15                 d s1 s2 i  d s1 s2 i  d s1 s2 i  d s1 s2 i
  16                 | |   | |  | |   | |  | |   | |  | |   | |
  17                 v v   v v  v v   v v  v v   v v  v v   v v
  18  go_rd/go_wr -> dm-r0-fu0  dm-r1-fu0  dm-r2-fu0  dm-r3-fu0 -> wr/rd-pend
  19  go_rd/go_wr -> dm-r0-fu1  dm-r1-fu1  dm-r2-fu1  dm-r3-fu1 -> wr/rd-pend
  20  go_rd/go_wr -> dm-r0-fu2  dm-r1-fu2  dm-r2-fu2  dm-r3-fu2 -> wr/rd-pend
  21                  |  |  |    |  |  |    |  |  |    |  |  |
  22                  v  v  v    v  v  v    v  v  v    v  v  v
  23                  d  s1 s2   d  s1 s2   d  s1 s2   d  s1 s2
  24                  reg sel    reg sel    reg sel    reg sel
  25
  26 """
  27
  28 class FURegDepMatrix(Elaboratable):
  29     """ implements 11.4.7 mitch alsup FU-to-Reg Dependency Matrix, p26
  30     """
  31     def __init__(self, n_fu_row, n_reg_col):
  32         self.n_fu_row = n_fu_row                  # Y (FUs)   ^v
  33         self.n_reg_col = n_reg_col                # X (Regs)  <>
  34         self.dest_i = Signal(n_reg_col, reset_less=True)     # Dest in (top)
  35         self.src1_i = Signal(n_reg_col, reset_less=True)     # oper1 in (top)
  36         self.src2_i = Signal(n_reg_col, reset_less=True)     # oper2 in (top)
  37
  38         self.issue_i = Signal(n_fu_row, reset_less=True)    # Issue in (top)
  39         self.go_wr_i = Signal(n_fu_row, reset_less=True) # Go Write in (left)
  40         self.go_rd_i = Signal(n_fu_row, reset_less=True)  # Go Read in (left)
  41
  42         # for Register File Select Lines (horizontal), per-reg
  43         self.dest_rsel_o = Signal(n_reg_col, reset_less=True) # dest reg (bot)
  44         self.src1_rsel_o = Signal(n_reg_col, reset_less=True) # src1 reg (bot)
  45         self.src2_rsel_o = Signal(n_reg_col, reset_less=True) # src2 reg (bot)
  46
  47         # for Function Unit "forward progress" (vertical), per-FU
  48         self.wr_pend_o = Signal(n_fu_row, reset_less=True) # wr pending (right)
  49         self.rd_pend_o = Signal(n_fu_row, reset_less=True) # rd pending (right)
  50         self.rd_src1_pend_o = Signal(n_fu_row, reset_less=True) # src1 pending
  51         self.rd_src2_pend_o = Signal(n_fu_row, reset_less=True) # src2 pending
  52
  53     def elaborate(self, platform):
  54         m = Module()
  55
  56         # ---
  57         # matrix of dependency cells
  58         # ---
  59         dm = Array(Array(DependenceCell() for r in range(self.n_fu_row)) \
  60                                           for f in range(self.n_reg_col))
  61         for rn in range(self.n_reg_col):
  62             for fu in range(self.n_fu_row):
  63                 setattr(m.submodules, "dm_r%d_fu%d" % (rn, fu), dm[rn][fu])
  64
  65         # ---
  66         # array of Function Unit Pending vectors
  67         # ---
  68         fupend = Array(FU_RW_Pend(self.n_reg_col) for f in range(self.n_fu_row))
  69         for fu in range(self.n_fu_row):
  70             setattr(m.submodules, "fu_fu%d" % (fu), fupend[fu])
  71
  72         # ---
  73         # array of Register Reservation vectors
  74         # ---
  75         regrsv = Array(Reg_Rsv(self.n_fu_row) for r in range(self.n_reg_col))
  76         for rn in range(self.n_reg_col):
  77             setattr(m.submodules, "rr_r%d" % (rn), regrsv[rn])
  78
  79         # ---
  80         # connect Function Unit vector
  81         # ---
  82         wr_pend = []
  83         rd_pend = []
  84         rd_src1_pend = []
  85         rd_src2_pend = []
  86         for fu in range(self.n_fu_row):
  87             fup = fupend[fu]
  88             dest_fwd_o = []
  89             src1_fwd_o = []
  90             src2_fwd_o = []
  91             for rn in range(self.n_reg_col):
  92                 dc = dm[rn][fu]
  93                 # accumulate cell fwd outputs for dest/src1/src2
  94                 dest_fwd_o.append(dc.dest_fwd_o)
  95                 src1_fwd_o.append(dc.src1_fwd_o)
  96                 src2_fwd_o.append(dc.src2_fwd_o)
  97             # connect cell fwd outputs to FU Vector in [Cat is gooood]
  98             m.d.comb += [fup.dest_fwd_i.eq(Cat(*dest_fwd_o)),
  99                          fup.src1_fwd_i.eq(Cat(*src1_fwd_o)),
 100                          fup.src2_fwd_i.eq(Cat(*src2_fwd_o))
 101                         ]
 102             # accumulate FU Vector outputs
 103             wr_pend.append(fup.reg_wr_pend_o)
 104             rd_pend.append(fup.reg_rd_pend_o)
 105             rd_src1_pend.append(fup.reg_rd_src1_pend_o)
 106             rd_src2_pend.append(fup.reg_rd_src2_pend_o)
 107
 108         # ... and output them from this module (vertical, width=FUs)
 109         m.d.comb += self.wr_pend_o.eq(Cat(*wr_pend))
 110         m.d.comb += self.rd_pend_o.eq(Cat(*rd_pend))
 111         m.d.comb += self.rd_src1_pend_o.eq(Cat(*rd_src1_pend))
 112         m.d.comb += self.rd_src2_pend_o.eq(Cat(*rd_src2_pend))
 113
 114         print ("wr pend len", len(wr_pend))
 115
 116         # ---
 117         # connect Reg Selection vector
 118         # ---
 119         dest_rsel = []
 120         src1_rsel = []
 121         src2_rsel = []
 122         for rn in range(self.n_reg_col):
 123             rsv = regrsv[rn]
 124             dest_rsel_o = []
 125             src1_rsel_o = []
 126             src2_rsel_o = []
 127             for fu in range(self.n_fu_row):
 128                 dc = dm[rn][fu]
 129                 # accumulate cell reg-select outputs dest/src1/src2
 130                 dest_rsel_o.append(dc.dest_rsel_o)
 131                 src1_rsel_o.append(dc.src1_rsel_o)
 132                 src2_rsel_o.append(dc.src2_rsel_o)
 133             # connect cell reg-select outputs to Reg Vector In
 134             m.d.comb += [rsv.dest_rsel_i.eq(Cat(*dest_rsel_o)),
 135                          rsv.src1_rsel_i.eq(Cat(*src1_rsel_o)),
 136                          rsv.src2_rsel_i.eq(Cat(*src2_rsel_o)),
 137                         ]
 138             # accumulate Reg-Sel Vector outputs
 139             dest_rsel.append(rsv.dest_rsel_o)
 140             src1_rsel.append(rsv.src1_rsel_o)
 141             src2_rsel.append(rsv.src2_rsel_o)
 142             print ("dest_rsel_rsv len", len(rsv.dest_rsel_o))
 143
 144         # ... and output them from this module (horizontal, width=REGs)
 145         m.d.comb += self.dest_rsel_o.eq(Cat(*dest_rsel))
 146         m.d.comb += self.src1_rsel_o.eq(Cat(*src1_rsel))
 147         m.d.comb += self.src2_rsel_o.eq(Cat(*src2_rsel))
 148
 149         print ("dest rsel len", len(dest_rsel), self.dest_rsel_o)
 150         # ---
 151         # connect Dependency Matrix dest/src1/src2/issue to module d/s/s/i
 152         # ---
 153         for fu in range(self.n_fu_row):
 154             dest_i = []
 155             src1_i = []
 156             src2_i = []
 157             for rn in range(self.n_reg_col):
 158                 dc = dm[rn][fu]
 159                 # accumulate cell inputs dest/src1/src2
 160                 dest_i.append(dc.dest_i)
 161                 src1_i.append(dc.src1_i)
 162                 src2_i.append(dc.src2_i)
 163             # wire up inputs from module to row cell inputs (Cat is gooood)
 164             m.d.comb += [Cat(*dest_i).eq(self.dest_i),
 165                          Cat(*src1_i).eq(self.src1_i),
 166                          Cat(*src2_i).eq(self.src2_i),
 167                         ]
 168
 169         # ---
 170         # connect Dep issue_i/go_rd_i/go_wr_i to module issue_i/go_rd/go_wr
 171         # ---
 172         for rn in range(self.n_reg_col):
 173             go_rd_i = []
 174             go_wr_i = []
 175             issue_i = []
 176             for fu in range(self.n_fu_row):
 177                 dc = dm[rn][fu]
 178                 # accumulate cell fwd outputs for dest/src1/src2
 179                 go_rd_i.append(dc.go_rd_i)
 180                 go_wr_i.append(dc.go_wr_i)
 181                 issue_i.append(dc.issue_i)
 182             # wire up inputs from module to row cell inputs (Cat is gooood)
 183             m.d.comb += [Cat(*go_rd_i).eq(self.go_rd_i),
 184                          Cat(*go_wr_i).eq(self.go_wr_i),
 185                          Cat(*issue_i).eq(self.issue_i),
 186                         ]
 187
 188         return m
 189
 190     def __iter__(self):
 191         yield self.dest_i
 192         yield self.src1_i
 193         yield self.src2_i
 194         yield self.issue_i
 195         yield self.go_wr_i
 196         yield self.go_rd_i
 197         yield self.dest_rsel_o
 198         yield self.src1_rsel_o
 199         yield self.src2_rsel_o
 200         yield self.wr_pend_o
 201         yield self.rd_pend_o
 202         yield self.rd_src1_pend_o
 203         yield self.rd_src2_pend_o
 204
 205     def ports(self):
 206         return list(self)
 207
 208 def d_matrix_sim(dut):
 209     """ XXX TODO
 210     """
 211     yield dut.dest_i.eq(1)
 212     yield dut.issue_i.eq(1)
 213     yield
 214     yield dut.issue_i.eq(0)
 215     yield
 216     yield dut.src1_i.eq(1)
 217     yield dut.issue_i.eq(1)
 218     yield
 219     yield dut.issue_i.eq(0)
 220     yield
 221     yield dut.go_rd_i.eq(1)
 222     yield
 223     yield dut.go_rd_i.eq(0)
 224     yield
 225     yield dut.go_wr_i.eq(1)
 226     yield
 227     yield dut.go_wr_i.eq(0)
 228     yield
 229
 230 def test_d_matrix():
 231     dut = FURegDepMatrix(n_fu_row=3, n_reg_col=4)
 232     vl = rtlil.convert(dut, ports=dut.ports())
 233     with open("test_fu_reg_matrix.il", "w") as f:
 234         f.write(vl)
 235
 236     run_simulation(dut, d_matrix_sim(dut), vcd_name='test_fu_reg_matrix.vcd')
 237
 238 if __name__ == '__main__':
 239     test_d_matrix()