xref: /XiangShan/src/main/scala/xiangshan/frontend/IFU.scala (revision 6e3ea132be870a2e7573e122ef35630c7b184003)
1package xiangshan.frontend
2
3import chisel3._
4import chisel3.util._
5import device.RAMHelper
6import xiangshan._
7import utils._
8
9trait HasIFUConst { this: XSModule =>
10  val resetVector = 0x80000000L//TODO: set reset vec
11  val groupAlign = log2Up(FetchWidth * 4)
12  def groupPC(pc: UInt): UInt = Cat(pc(VAddrBits-1, groupAlign), 0.U(groupAlign.W))
13  def snpc(pc: UInt): UInt = pc + (1 << groupAlign).U
14
15}
16
17class IFUIO extends XSBundle
18{
19    val fetchPacket = DecoupledIO(new FetchPacket)
20    val redirectInfo = Input(new RedirectInfo)
21    val icacheReq = DecoupledIO(new FakeIcacheReq)
22    val icacheResp = Flipped(DecoupledIO(new FakeIcacheResp))
23}
24
25class FakeBPU extends XSModule{
26  val io = IO(new Bundle() {
27    val redirectInfo = Input(new RedirectInfo)
28    val in = new Bundle { val pc = Flipped(Valid(UInt(VAddrBits.W))) }
29    val btbOut = ValidIO(new BranchPrediction)
30    val tageOut = Decoupled(new BranchPrediction)
31    val predecode = Flipped(ValidIO(new Predecode))
32  })
33
34  io.btbOut.valid := true.B
35  io.btbOut.bits <> DontCare
36  io.btbOut.bits.redirect := GTimer() === 1.U
37  io.btbOut.bits.target := "h080001234".U
38  io.tageOut.valid := false.B
39  io.tageOut.bits <> DontCare
40}
41
42
43class IFU extends XSModule with HasIFUConst
44{
45    val io = IO(new IFUIO)
46    val bpu = if(EnableBPU) Module(new BPU) else Module(new FakeBPU)
47
48    //-------------------------
49    //      IF1  PC update
50    //-------------------------
51    //local
52    val if1_npc = WireInit(0.U(VAddrBits.W))
53    val if1_valid = !reset.asBool
54    val if1_pc = RegInit(resetVector.U(VAddrBits.W))
55    //next
56    val if2_ready = WireInit(false.B)
57    val if2_snpc = snpc(if1_pc) //TODO: calculate snpc according to mask of current fetch packet
58    val needflush = WireInit(false.B)
59    // when an RVI instruction is predicted as taken and it crosses over two fetch packets,
60    // IFU should not take this branch but fetch the latter half of the instruction sequentially,
61    // and take the jump target in the next fetch cycle
62    val if2_lateJumpLatch = WireInit(false.B)
63    val if2_lateJumpTarget = RegInit(0.U(VAddrBits.W))
64    val if4_lateJumpLatch = WireInit(false.B)
65    val if4_lateJumpTarget = RegInit(0.U(VAddrBits.W))
66
67    //pipe fire
68    val if1_fire = if1_valid && if2_ready || needflush
69    val if1_pcUpdate = if1_fire || needflush
70
71    bpu.io.in.pc.valid := if1_fire
72    bpu.io.in.pc.bits := if1_npc
73    bpu.io.redirectInfo := io.redirectInfo
74
75    XSDebug("[IF1]if1_valid:%d  ||  if1_npc:0x%x  || if1_pcUpdate:%d if1_pc:0x%x  || if2_ready:%d",if1_valid,if1_npc,if1_pcUpdate,if1_pc,if2_ready)
76    XSDebug(false,if1_fire,"------IF1->fire!!!")
77    XSDebug(false,true.B,"\n")
78
79    //-------------------------
80    //      IF2  btb response
81    //           icache visit
82    //-------------------------
83    //local
84    val if2_valid = RegEnable(next=if1_valid,init=false.B,enable=if1_fire)
85    val if2_pc = if1_pc
86    val if2_btb_taken = bpu.io.btbOut.valid && bpu.io.btbOut.bits.redirect
87    val if2_btb_lateJump = WireInit(false.B)
88    val if2_btb_insMask = Mux(if2_btb_taken, bpu.io.btbOut.bits.instrValid.asUInt, Fill(FetchWidth*2, 1.U(1.W))) // TODO: FIX THIS
89    val if2_btb_target = Mux(if2_btb_lateJump, if2_snpc, bpu.io.btbOut.bits.target)
90
91    if2_lateJumpLatch := BoolStopWatch(if2_btb_lateJump, if1_fire, startHighPriority = true)
92    // since late jump target should be taken after the latter half of late jump instr is fetched, we need to latch this target
93    when (if2_btb_lateJump) {
94      if2_lateJumpTarget := bpu.io.btbOut.bits.target
95    }
96
97    //next
98    val if3_ready = WireInit(false.B)
99
100    //pipe fire
101    val if2_fire = if2_valid && if3_ready && io.icacheReq.fire()
102    if2_ready := (if2_fire) || !if2_valid
103
104    io.icacheReq.valid := if2_valid
105    io.icacheReq.bits.addr := if2_pc
106
107    when(RegNext(reset.asBool) && !reset.asBool){
108    //when((GTimer() === 501.U)){ //TODO:this is ugly
109      XSDebug("RESET....\n")
110      if1_npc := resetVector.U(VAddrBits.W)
111    }.elsewhen (if2_fire) {
112      if1_npc := Mux(if4_lateJumpLatch, if4_lateJumpTarget, Mux(if2_lateJumpLatch, if2_lateJumpTarget, if2_snpc))
113    }.otherwise {
114      if1_npc := if1_pc
115    }
116
117    when(if1_pcUpdate)
118    {
119      if1_pc := if1_npc
120    }
121
122    // when if2 fire and if2 redirects, update npc
123    when(if2_fire && if2_btb_taken)
124    {
125      if1_npc := if2_btb_target
126    }
127
128    bpu.io.in.pc.valid := if1_fire && !if2_btb_lateJump
129
130    XSDebug("[IF2]if2_valid:%d  ||  if2_pc:0x%x   || if3_ready:%d                                        ",if2_valid,if2_pc,if3_ready)
131    XSDebug(false,if2_fire,"------IF2->fire!!!")
132    XSDebug(false,true.B,"\n")
133    XSDebug("[IF2-Icache-Req] icache_in_valid:%d  icache_in_ready:%d\n",io.icacheReq.valid,io.icacheReq.ready)
134    XSDebug("[IF2-BPU-out]if2_btbTaken:%d || if2_btb_insMask:%b || if2_btb_target:0x%x \n",if2_btb_taken,if2_btb_insMask.asUInt,if2_btb_target)
135    //-------------------------
136    //      IF3  icache hit check
137    //-------------------------
138    //local
139    val if3_valid = RegEnable(next=if2_valid,init=false.B,enable=if2_fire)
140    val if3_pc = RegEnable(if2_pc,if2_fire)
141    val if3_npc = RegEnable(if1_npc, if2_fire)
142    val if3_btb_target = RegEnable(Mux(if2_lateJumpLatch, if2_lateJumpTarget, Mux(if2_btb_lateJump, bpu.io.btbOut.bits.target, if2_btb_target)), if2_fire)
143    val if3_btb_taken = RegEnable(Mux(if2_lateJumpLatch, true.B, if2_btb_taken), if2_fire)
144    val if3_btb_insMask = RegEnable(Mux(if2_lateJumpLatch, 1.U((FetchWidth*2).W), if2_btb_insMask), if2_fire)
145    val if3_btb_lateJump = RegEnable(if2_btb_lateJump, if2_fire)
146
147    //next
148    val if4_ready = WireInit(false.B)
149
150    //pipe fire
151    val if3_fire = if3_valid && if4_ready
152    if3_ready := if3_fire  || !if3_valid
153
154
155    XSDebug("[IF3]if3_valid:%d  ||  if3_pc:0x%x   if3_npc:0x%x || if4_ready:%d                    ",if3_valid,if3_pc,if3_npc,if4_ready)
156    XSDebug("[IF3]if3_btb_taken:%d if3_btb_insMask:%b if3_btb_lateJump:%d if3_btb_target:0x%x\n",
157      if3_btb_taken, if3_btb_insMask, if3_btb_lateJump, if3_btb_target)
158    XSDebug(false,if3_fire,"------IF3->fire!!!")
159    XSDebug(false,true.B,"\n")
160
161    //-------------------------
162    //      IF4  icache response
163    //           RAS result
164    //           taget generate
165    //-------------------------
166    val if4_valid = RegEnable(next=if3_valid,init=false.B,enable=if3_fire)
167    val if4_pc = RegEnable(if3_pc,if3_fire)
168    val if4_npc = RegEnable(if3_npc,if3_fire)
169    val if4_btb_target = RegEnable(if3_btb_target,if3_fire)
170    val if4_btb_taken = RegEnable(if3_btb_taken,if3_fire)
171    val if4_btb_insMask = RegEnable(if3_btb_insMask, if3_fire)
172    val if4_btb_lateJump = RegEnable(if3_btb_lateJump, if3_fire)
173    val if4_tage_taken = bpu.io.tageOut.valid && bpu.io.tageOut.bits.redirect
174    val if4_tage_lateJump = if4_tage_taken && bpu.io.tageOut.bits.lateJump && !io.redirectInfo.flush()
175    val if4_tage_insMask = bpu.io.tageOut.bits.instrValid
176    val if4_snpc = if4_pc + (PopCount(if4_tage_insMask) << 1.U)
177    val if4_tage_target = Mux(if4_tage_lateJump, if4_snpc, bpu.io.tageOut.bits.target)
178
179    if2_btb_lateJump := if2_btb_taken && bpu.io.btbOut.bits.lateJump && !io.redirectInfo.flush() && !if4_tage_taken
180
181    if4_lateJumpLatch := BoolStopWatch(if4_tage_lateJump, if1_fire, startHighPriority = true)
182    when (if4_tage_lateJump) {
183      if4_lateJumpTarget := bpu.io.tageOut.bits.target
184    }
185
186    bpu.io.in.pc.valid := if1_fire && !if2_btb_lateJump && !if4_tage_lateJump
187
188    XSDebug("[IF4]if4_valid:%d  ||  if4_pc:0x%x   if4_npc:0x%x\n",if4_valid,if4_pc,if4_npc)
189    XSDebug("[IF4]          if4_btb_taken:%d  if4_btb_lateJump:%d  if4_btb_insMask:%b  if4_btb_target:0x%x\n",if4_btb_taken, if4_btb_lateJump, if4_btb_insMask.asUInt, if4_btb_target)
190    XSDebug("[IF4-TAGE-out]if4_tage_taken:%d if4_tage_lateJump:%d if4_tage_insMask:%b if4_tage_target:0x%x\n",if4_tage_taken,if4_tage_lateJump,if4_tage_insMask.asUInt,if4_tage_target)
191    XSDebug("[IF4-ICACHE-RESP]icacheResp.valid:%d   icacheResp.ready:%d\n",io.icacheResp.valid,io.icacheResp.ready)
192
193    when(io.icacheResp.fire() && if4_tage_taken &&if4_valid)
194    {
195      if1_npc := if4_tage_target
196    }
197
198    //redirect: miss predict
199    when(io.redirectInfo.flush()){
200      if1_npc := io.redirectInfo.redirect.target
201    }
202    XSDebug(io.redirectInfo.flush(),"[IFU-REDIRECT] target:0x%x  \n",io.redirectInfo.redirect.target.asUInt)
203
204
205    //flush pipline
206    // if(EnableBPD){needflush := (if4_valid && if4_tage_taken) || io.redirectInfo.flush() }
207    // else {needflush := io.redirectInfo.flush()}
208    needflush := (if4_valid && if4_tage_taken && io.icacheResp.fire()) || io.redirectInfo.flush()
209    when(needflush){
210      // if2_valid := false.B
211      if3_valid := false.B
212      if4_valid := false.B
213    }
214    //flush ICache
215    io.icacheReq.bits.flush := needflush
216
217    //Output -> iBuffer
218    //io.fetchPacket <> DontCare
219    if4_ready := io.fetchPacket.ready && (io.icacheResp.valid || !if4_valid) && (GTimer() > 500.U)
220    io.fetchPacket.valid := if4_valid && !io.redirectInfo.flush()
221    io.fetchPacket.bits.instrs := io.icacheResp.bits.icacheOut
222    /*
223    if(EnableBPU){
224      io.fetchPacket.bits.mask := Mux(if4_tage_taken, Fill(FetchWidth*2, 1.U(1.W)) & if4_tage_insMask.asUInt,
225                                  Mux(if4_btb_taken,  Fill(FetchWidth*2, 1.U(1.W)) & if4_btb_insMask.asUInt,
226                                                      Fill(FetchWidth*2, 1.U(1.W)))
227      )
228    }
229    else{
230      io.fetchPacket.bits.mask := Fill(FetchWidth*2, 1.U(1.W)) //TODO : consider cross cacheline fetch
231    }
232    */
233    io.fetchPacket.bits.mask := Mux(if4_lateJumpLatch,  1.U((FetchWidth*2).W),
234                                Mux(if4_tage_taken,     Fill(FetchWidth*2, 1.U(1.W)) & if4_tage_insMask.asUInt,
235                                                        Fill(FetchWidth*2, 1.U(1.W)) & if4_btb_insMask.asUInt))
236    io.fetchPacket.bits.pc := if4_pc
237
238    XSDebug(io.fetchPacket.fire,"[IFU-Out-FetchPacket] starPC:0x%x   GroupPC:0x%xn\n",if4_pc.asUInt,groupPC(if4_pc).asUInt)
239    XSDebug(io.fetchPacket.fire,"[IFU-Out-FetchPacket] instrmask %b\n",io.fetchPacket.bits.mask.asUInt)
240    for(i <- 0 until (FetchWidth*2)) {
241      when (if4_btb_taken && !if4_tage_taken && i.U === OHToUInt(HighestBit(if4_btb_insMask.asUInt, FetchWidth*2))) {
242        io.fetchPacket.bits.pnpc(i) := if4_btb_target
243        if (i != 0) {
244          when (!io.icacheResp.bits.predecode.isRVC(i) && !if4_btb_lateJump) {
245            io.fetchPacket.bits.pnpc(i-1) := if4_btb_target
246          }
247        }
248      }.elsewhen (if4_tage_taken && i.U === OHToUInt(HighestBit(if4_tage_insMask.asUInt, FetchWidth*2))) {
249        io.fetchPacket.bits.pnpc(i) := Mux(if4_tage_lateJump, bpu.io.tageOut.bits.target, if4_tage_target)
250        if (i != 0) {
251          when (!io.icacheResp.bits.predecode.isRVC(i) && !if4_tage_lateJump) {
252            io.fetchPacket.bits.pnpc(i-1) := if4_tage_target
253          }
254        }
255      }.otherwise {
256        io.fetchPacket.bits.pnpc(i) := if4_pc + (i.U << 1.U) + Mux(io.icacheResp.bits.predecode.isRVC(i), 2.U, 4.U)
257      }
258      XSDebug(io.fetchPacket.fire,"[IFU-Out-FetchPacket] instruction %x    pnpc:0x%x\n",
259        Mux((i.U)(0), io.fetchPacket.bits.instrs(i>>1)(31,16), io.fetchPacket.bits.instrs(i>>1)(15,0)),
260        io.fetchPacket.bits.pnpc(i))
261    }
262    io.fetchPacket.bits.hist := bpu.io.tageOut.bits.hist
263    // io.fetchPacket.bits.btbVictimWay := bpu.io.tageOut.bits.btbVictimWay
264    io.fetchPacket.bits.predCtr := bpu.io.tageOut.bits.predCtr
265    io.fetchPacket.bits.btbHit := bpu.io.tageOut.bits.btbHit
266    io.fetchPacket.bits.tageMeta := bpu.io.tageOut.bits.tageMeta
267    io.fetchPacket.bits.rasSp := bpu.io.tageOut.bits.rasSp
268    io.fetchPacket.bits.rasTopCtr := bpu.io.tageOut.bits.rasTopCtr
269    bpu.io.tageOut.ready := io.fetchPacket.ready
270
271    //to BPU
272    bpu.io.predecode.valid := io.icacheResp.fire() && if4_valid
273    bpu.io.predecode.bits <> io.icacheResp.bits.predecode
274    //TODO: consider RVC && consider cross cacheline fetch
275    bpu.io.predecode.bits.mask := Fill(FetchWidth*2, 1.U(1.W))
276    bpu.io.predecode.bits.isRVC := 0.U.asTypeOf(Vec(FetchWidth*2, Bool()))
277    bpu.io.redirectInfo := io.redirectInfo
278    io.icacheResp.ready := io.fetchPacket.ready && (GTimer() > 500.U)
279
280}