109c6f1ddSLingrui98/*************************************************************************************** 209c6f1ddSLingrui98* Copyright (c) 2020-2021 Institute of Computing Technology, Chinese Academy of Sciences 309c6f1ddSLingrui98* Copyright (c) 2020-2021 Peng Cheng Laboratory 409c6f1ddSLingrui98* 509c6f1ddSLingrui98* XiangShan is licensed under Mulan PSL v2. 609c6f1ddSLingrui98* You can use this software according to the terms and conditions of the Mulan PSL v2. 709c6f1ddSLingrui98* You may obtain a copy of Mulan PSL v2 at: 809c6f1ddSLingrui98* http://license.coscl.org.cn/MulanPSL2 909c6f1ddSLingrui98* 1009c6f1ddSLingrui98* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, 1109c6f1ddSLingrui98* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, 1209c6f1ddSLingrui98* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. 1309c6f1ddSLingrui98* 1409c6f1ddSLingrui98* See the Mulan PSL v2 for more details. 1509c6f1ddSLingrui98***************************************************************************************/ 1609c6f1ddSLingrui98 1709c6f1ddSLingrui98package xiangshan.frontend 1809c6f1ddSLingrui98 1909c6f1ddSLingrui98import chipsalliance.rocketchip.config.Parameters 2009c6f1ddSLingrui98import chisel3._ 2109c6f1ddSLingrui98import chisel3.stage.{ChiselGeneratorAnnotation, ChiselStage} 2209c6f1ddSLingrui98import chisel3.util._ 2309c6f1ddSLingrui98import xiangshan._ 2409c6f1ddSLingrui98import utils._ 253c02ee8fSwakafaimport utility._ 2609c6f1ddSLingrui98import chisel3.experimental.chiselName 2709c6f1ddSLingrui98 2809c6f1ddSLingrui98import scala.math.min 29adc0b8dfSGuokai Chenimport scala.{Tuple2 => &} 30eeb5ff92SLingrui98import os.copy 3109c6f1ddSLingrui98 3209c6f1ddSLingrui98 3309c6f1ddSLingrui98trait FTBParams extends HasXSParameter with HasBPUConst { 34b37e4b45SLingrui98 val numEntries = FtbSize 35b37e4b45SLingrui98 val numWays = FtbWays 3609c6f1ddSLingrui98 val numSets = numEntries/numWays // 512 3709c6f1ddSLingrui98 val tagSize = 20 3809c6f1ddSLingrui98 39eeb5ff92SLingrui98 40eeb5ff92SLingrui98 4109c6f1ddSLingrui98 val TAR_STAT_SZ = 2 4209c6f1ddSLingrui98 def TAR_FIT = 0.U(TAR_STAT_SZ.W) 4309c6f1ddSLingrui98 def TAR_OVF = 1.U(TAR_STAT_SZ.W) 4409c6f1ddSLingrui98 def TAR_UDF = 2.U(TAR_STAT_SZ.W) 4509c6f1ddSLingrui98 46bf358e08SLingrui98 def BR_OFFSET_LEN = 12 47bf358e08SLingrui98 def JMP_OFFSET_LEN = 20 4809c6f1ddSLingrui98} 4909c6f1ddSLingrui98 50b30c10d6SLingrui98class FtbSlot(val offsetLen: Int, val subOffsetLen: Option[Int] = None)(implicit p: Parameters) extends XSBundle with FTBParams { 51b30c10d6SLingrui98 if (subOffsetLen.isDefined) { 52b30c10d6SLingrui98 require(subOffsetLen.get <= offsetLen) 53b30c10d6SLingrui98 } 54eeb5ff92SLingrui98 val offset = UInt(log2Ceil(PredictWidth).W) 55eeb5ff92SLingrui98 val lower = UInt(offsetLen.W) 56eeb5ff92SLingrui98 val tarStat = UInt(TAR_STAT_SZ.W) 57eeb5ff92SLingrui98 val sharing = Bool() 5809c6f1ddSLingrui98 val valid = Bool() 5909c6f1ddSLingrui98 60d2b20d1aSTang Haojin val sc = Bool() // set by sc in s3, perf use only 61d2b20d1aSTang Haojin 62eeb5ff92SLingrui98 def setLowerStatByTarget(pc: UInt, target: UInt, isShare: Boolean) = { 63eeb5ff92SLingrui98 def getTargetStatByHigher(pc_higher: UInt, target_higher: UInt) = 64eeb5ff92SLingrui98 Mux(target_higher > pc_higher, TAR_OVF, 65eeb5ff92SLingrui98 Mux(target_higher < pc_higher, TAR_UDF, TAR_FIT)) 66eeb5ff92SLingrui98 def getLowerByTarget(target: UInt, offsetLen: Int) = target(offsetLen, 1) 67b30c10d6SLingrui98 val offLen = if (isShare) this.subOffsetLen.get else this.offsetLen 68eeb5ff92SLingrui98 val pc_higher = pc(VAddrBits-1, offLen+1) 69eeb5ff92SLingrui98 val target_higher = target(VAddrBits-1, offLen+1) 70eeb5ff92SLingrui98 val stat = getTargetStatByHigher(pc_higher, target_higher) 71eeb5ff92SLingrui98 val lower = ZeroExt(getLowerByTarget(target, offLen), this.offsetLen) 72eeb5ff92SLingrui98 this.lower := lower 73eeb5ff92SLingrui98 this.tarStat := stat 74eeb5ff92SLingrui98 this.sharing := isShare.B 75eeb5ff92SLingrui98 } 7609c6f1ddSLingrui98 77b30c10d6SLingrui98 def getTarget(pc: UInt, last_stage: Option[Tuple2[UInt, Bool]] = None) = { 78b30c10d6SLingrui98 def getTarget(offLen: Int)(pc: UInt, lower: UInt, stat: UInt, 79b30c10d6SLingrui98 last_stage: Option[Tuple2[UInt, Bool]] = None) = { 80b30c10d6SLingrui98 val h = pc(VAddrBits - 1, offLen + 1) 81b30c10d6SLingrui98 val higher = Wire(UInt((VAddrBits - offLen - 1).W)) 82b30c10d6SLingrui98 val higher_plus_one = Wire(UInt((VAddrBits - offLen - 1).W)) 83b30c10d6SLingrui98 val higher_minus_one = Wire(UInt((VAddrBits-offLen-1).W)) 8447c003a9SEaston Man 8547c003a9SEaston Man // Switch between previous stage pc and current stage pc 8647c003a9SEaston Man // Give flexibility for timing 87b30c10d6SLingrui98 if (last_stage.isDefined) { 88b30c10d6SLingrui98 val last_stage_pc = last_stage.get._1 89b30c10d6SLingrui98 val last_stage_pc_h = last_stage_pc(VAddrBits-1, offLen+1) 90b30c10d6SLingrui98 val stage_en = last_stage.get._2 91b30c10d6SLingrui98 higher := RegEnable(last_stage_pc_h, stage_en) 92b30c10d6SLingrui98 higher_plus_one := RegEnable(last_stage_pc_h+1.U, stage_en) 93b30c10d6SLingrui98 higher_minus_one := RegEnable(last_stage_pc_h-1.U, stage_en) 94b30c10d6SLingrui98 } else { 95b30c10d6SLingrui98 higher := h 96b30c10d6SLingrui98 higher_plus_one := h + 1.U 97b30c10d6SLingrui98 higher_minus_one := h - 1.U 98b30c10d6SLingrui98 } 99eeb5ff92SLingrui98 val target = 100eeb5ff92SLingrui98 Cat( 101b30c10d6SLingrui98 Mux1H(Seq( 102b30c10d6SLingrui98 (stat === TAR_OVF, higher_plus_one), 103b30c10d6SLingrui98 (stat === TAR_UDF, higher_minus_one), 104b30c10d6SLingrui98 (stat === TAR_FIT, higher), 105b30c10d6SLingrui98 )), 106eeb5ff92SLingrui98 lower(offLen-1, 0), 0.U(1.W) 107eeb5ff92SLingrui98 ) 108eeb5ff92SLingrui98 require(target.getWidth == VAddrBits) 109eeb5ff92SLingrui98 require(offLen != 0) 110eeb5ff92SLingrui98 target 111eeb5ff92SLingrui98 } 112b30c10d6SLingrui98 if (subOffsetLen.isDefined) 113eeb5ff92SLingrui98 Mux(sharing, 114b30c10d6SLingrui98 getTarget(subOffsetLen.get)(pc, lower, tarStat, last_stage), 115b30c10d6SLingrui98 getTarget(offsetLen)(pc, lower, tarStat, last_stage) 116eeb5ff92SLingrui98 ) 117eeb5ff92SLingrui98 else 118b30c10d6SLingrui98 getTarget(offsetLen)(pc, lower, tarStat, last_stage) 119eeb5ff92SLingrui98 } 120eeb5ff92SLingrui98 def fromAnotherSlot(that: FtbSlot) = { 121eeb5ff92SLingrui98 require( 122b30c10d6SLingrui98 this.offsetLen > that.offsetLen && this.subOffsetLen.map(_ == that.offsetLen).getOrElse(true) || 123eeb5ff92SLingrui98 this.offsetLen == that.offsetLen 124eeb5ff92SLingrui98 ) 125eeb5ff92SLingrui98 this.offset := that.offset 126eeb5ff92SLingrui98 this.tarStat := that.tarStat 127b30c10d6SLingrui98 this.sharing := (this.offsetLen > that.offsetLen && that.offsetLen == this.subOffsetLen.get).B 128eeb5ff92SLingrui98 this.valid := that.valid 129eeb5ff92SLingrui98 this.lower := ZeroExt(that.lower, this.offsetLen) 130eeb5ff92SLingrui98 } 131eeb5ff92SLingrui98 132eeb5ff92SLingrui98} 133eeb5ff92SLingrui98 134eeb5ff92SLingrui98class FTBEntry(implicit p: Parameters) extends XSBundle with FTBParams with BPUUtils { 135eeb5ff92SLingrui98 136eeb5ff92SLingrui98 137eeb5ff92SLingrui98 val valid = Bool() 138eeb5ff92SLingrui98 139eeb5ff92SLingrui98 val brSlots = Vec(numBrSlot, new FtbSlot(BR_OFFSET_LEN)) 140eeb5ff92SLingrui98 141b30c10d6SLingrui98 val tailSlot = new FtbSlot(JMP_OFFSET_LEN, Some(BR_OFFSET_LEN)) 14209c6f1ddSLingrui98 14309c6f1ddSLingrui98 // Partial Fall-Through Address 144a60a2901SLingrui98 val pftAddr = UInt(log2Up(PredictWidth).W) 14509c6f1ddSLingrui98 val carry = Bool() 14609c6f1ddSLingrui98 14709c6f1ddSLingrui98 val isCall = Bool() 14809c6f1ddSLingrui98 val isRet = Bool() 14909c6f1ddSLingrui98 val isJalr = Bool() 15009c6f1ddSLingrui98 151f4ebc4b2SLingrui98 val last_may_be_rvi_call = Bool() 15209c6f1ddSLingrui98 15309c6f1ddSLingrui98 val always_taken = Vec(numBr, Bool()) 15409c6f1ddSLingrui98 155eeb5ff92SLingrui98 def getSlotForBr(idx: Int): FtbSlot = { 156b37e4b45SLingrui98 require(idx <= numBr-1) 157b37e4b45SLingrui98 (idx, numBr) match { 158b37e4b45SLingrui98 case (i, n) if i == n-1 => this.tailSlot 159eeb5ff92SLingrui98 case _ => this.brSlots(idx) 16009c6f1ddSLingrui98 } 16109c6f1ddSLingrui98 } 162eeb5ff92SLingrui98 def allSlotsForBr = { 163eeb5ff92SLingrui98 (0 until numBr).map(getSlotForBr(_)) 16409c6f1ddSLingrui98 } 16509c6f1ddSLingrui98 def setByBrTarget(brIdx: Int, pc: UInt, target: UInt) = { 166eeb5ff92SLingrui98 val slot = getSlotForBr(brIdx) 167b37e4b45SLingrui98 slot.setLowerStatByTarget(pc, target, brIdx == numBr-1) 16809c6f1ddSLingrui98 } 16909c6f1ddSLingrui98 def setByJmpTarget(pc: UInt, target: UInt) = { 170eeb5ff92SLingrui98 this.tailSlot.setLowerStatByTarget(pc, target, false) 17109c6f1ddSLingrui98 } 17209c6f1ddSLingrui98 173b30c10d6SLingrui98 def getTargetVec(pc: UInt, last_stage: Option[Tuple2[UInt, Bool]] = None) = { 174b30c10d6SLingrui98 VecInit((brSlots :+ tailSlot).map(_.getTarget(pc, last_stage))) 175bf358e08SLingrui98 } 17609c6f1ddSLingrui98 177eeb5ff92SLingrui98 def getOffsetVec = VecInit(brSlots.map(_.offset) :+ tailSlot.offset) 17809c6f1ddSLingrui98 def isJal = !isJalr 17947c003a9SEaston Man def getFallThrough(pc: UInt, last_stage_entry: Option[Tuple2[FTBEntry, Bool]] = None) = { 18047c003a9SEaston Man if (last_stage_entry.isDefined) { 18147c003a9SEaston Man var stashed_carry = RegEnable(last_stage_entry.get._1.carry, last_stage_entry.get._2) 18247c003a9SEaston Man getFallThroughAddr(pc, stashed_carry, pftAddr) 18347c003a9SEaston Man } else { 18447c003a9SEaston Man getFallThroughAddr(pc, carry, pftAddr) 18547c003a9SEaston Man } 18647c003a9SEaston Man } 18747c003a9SEaston Man 188eeb5ff92SLingrui98 def hasBr(offset: UInt) = 189eeb5ff92SLingrui98 brSlots.map{ s => s.valid && s.offset <= offset}.reduce(_||_) || 190b37e4b45SLingrui98 (tailSlot.valid && tailSlot.offset <= offset && tailSlot.sharing) 19109c6f1ddSLingrui98 192eeb5ff92SLingrui98 def getBrMaskByOffset(offset: UInt) = 193b37e4b45SLingrui98 brSlots.map{ s => s.valid && s.offset <= offset } :+ 194b37e4b45SLingrui98 (tailSlot.valid && tailSlot.offset <= offset && tailSlot.sharing) 195eeb5ff92SLingrui98 196eeb5ff92SLingrui98 def getBrRecordedVec(offset: UInt) = { 197eeb5ff92SLingrui98 VecInit( 198b37e4b45SLingrui98 brSlots.map(s => s.valid && s.offset === offset) :+ 199b37e4b45SLingrui98 (tailSlot.valid && tailSlot.offset === offset && tailSlot.sharing) 200eeb5ff92SLingrui98 ) 20109c6f1ddSLingrui98 } 20209c6f1ddSLingrui98 203eeb5ff92SLingrui98 def brIsSaved(offset: UInt) = getBrRecordedVec(offset).reduce(_||_) 204eeb5ff92SLingrui98 205eeb5ff92SLingrui98 def brValids = { 206eeb5ff92SLingrui98 VecInit( 207b37e4b45SLingrui98 brSlots.map(_.valid) :+ (tailSlot.valid && tailSlot.sharing) 208eeb5ff92SLingrui98 ) 209eeb5ff92SLingrui98 } 210eeb5ff92SLingrui98 211eeb5ff92SLingrui98 def noEmptySlotForNewBr = { 212b37e4b45SLingrui98 VecInit(brSlots.map(_.valid) :+ tailSlot.valid).reduce(_&&_) 213eeb5ff92SLingrui98 } 214eeb5ff92SLingrui98 215eeb5ff92SLingrui98 def newBrCanNotInsert(offset: UInt) = { 216b37e4b45SLingrui98 val lastSlotForBr = tailSlot 217eeb5ff92SLingrui98 lastSlotForBr.valid && lastSlotForBr.offset < offset 218eeb5ff92SLingrui98 } 219eeb5ff92SLingrui98 220eeb5ff92SLingrui98 def jmpValid = { 221b37e4b45SLingrui98 tailSlot.valid && !tailSlot.sharing 222eeb5ff92SLingrui98 } 223eeb5ff92SLingrui98 224eeb5ff92SLingrui98 def brOffset = { 225b37e4b45SLingrui98 VecInit(brSlots.map(_.offset) :+ tailSlot.offset) 226eeb5ff92SLingrui98 } 227eeb5ff92SLingrui98 22809c6f1ddSLingrui98 def display(cond: Bool): Unit = { 22909c6f1ddSLingrui98 XSDebug(cond, p"-----------FTB entry----------- \n") 23009c6f1ddSLingrui98 XSDebug(cond, p"v=${valid}\n") 23109c6f1ddSLingrui98 for(i <- 0 until numBr) { 232eeb5ff92SLingrui98 XSDebug(cond, p"[br$i]: v=${allSlotsForBr(i).valid}, offset=${allSlotsForBr(i).offset}," + 233eeb5ff92SLingrui98 p"lower=${Hexadecimal(allSlotsForBr(i).lower)}\n") 23409c6f1ddSLingrui98 } 235eeb5ff92SLingrui98 XSDebug(cond, p"[tailSlot]: v=${tailSlot.valid}, offset=${tailSlot.offset}," + 236eeb5ff92SLingrui98 p"lower=${Hexadecimal(tailSlot.lower)}, sharing=${tailSlot.sharing}}\n") 23709c6f1ddSLingrui98 XSDebug(cond, p"pftAddr=${Hexadecimal(pftAddr)}, carry=$carry\n") 23809c6f1ddSLingrui98 XSDebug(cond, p"isCall=$isCall, isRet=$isRet, isjalr=$isJalr\n") 239f4ebc4b2SLingrui98 XSDebug(cond, p"last_may_be_rvi_call=$last_may_be_rvi_call\n") 24009c6f1ddSLingrui98 XSDebug(cond, p"------------------------------- \n") 24109c6f1ddSLingrui98 } 24209c6f1ddSLingrui98 24309c6f1ddSLingrui98} 24409c6f1ddSLingrui98 24509c6f1ddSLingrui98class FTBEntryWithTag(implicit p: Parameters) extends XSBundle with FTBParams with BPUUtils { 24609c6f1ddSLingrui98 val entry = new FTBEntry 24709c6f1ddSLingrui98 val tag = UInt(tagSize.W) 24809c6f1ddSLingrui98 def display(cond: Bool): Unit = { 249eeb5ff92SLingrui98 entry.display(cond) 250eeb5ff92SLingrui98 XSDebug(cond, p"tag is ${Hexadecimal(tag)}\n------------------------------- \n") 25109c6f1ddSLingrui98 } 25209c6f1ddSLingrui98} 25309c6f1ddSLingrui98 25409c6f1ddSLingrui98class FTBMeta(implicit p: Parameters) extends XSBundle with FTBParams { 255bb09c7feSzoujr val writeWay = UInt(log2Ceil(numWays).W) 25609c6f1ddSLingrui98 val hit = Bool() 2571bc6e9c8SLingrui98 val pred_cycle = if (!env.FPGAPlatform) Some(UInt(64.W)) else None 25809c6f1ddSLingrui98} 25909c6f1ddSLingrui98 26009c6f1ddSLingrui98object FTBMeta { 26109c6f1ddSLingrui98 def apply(writeWay: UInt, hit: Bool, pred_cycle: UInt)(implicit p: Parameters): FTBMeta = { 26209c6f1ddSLingrui98 val e = Wire(new FTBMeta) 26309c6f1ddSLingrui98 e.writeWay := writeWay 26409c6f1ddSLingrui98 e.hit := hit 2651bc6e9c8SLingrui98 e.pred_cycle.map(_ := pred_cycle) 26609c6f1ddSLingrui98 e 26709c6f1ddSLingrui98 } 26809c6f1ddSLingrui98} 26909c6f1ddSLingrui98 270c6bf0bffSzoujr// class UpdateQueueEntry(implicit p: Parameters) extends XSBundle with FTBParams { 271c6bf0bffSzoujr// val pc = UInt(VAddrBits.W) 272c6bf0bffSzoujr// val ftb_entry = new FTBEntry 273c6bf0bffSzoujr// val hit = Bool() 274c6bf0bffSzoujr// val hit_way = UInt(log2Ceil(numWays).W) 275c6bf0bffSzoujr// } 276c6bf0bffSzoujr// 277c6bf0bffSzoujr// object UpdateQueueEntry { 278c6bf0bffSzoujr// def apply(pc: UInt, fe: FTBEntry, hit: Bool, hit_way: UInt)(implicit p: Parameters): UpdateQueueEntry = { 279c6bf0bffSzoujr// val e = Wire(new UpdateQueueEntry) 280c6bf0bffSzoujr// e.pc := pc 281c6bf0bffSzoujr// e.ftb_entry := fe 282c6bf0bffSzoujr// e.hit := hit 283c6bf0bffSzoujr// e.hit_way := hit_way 284c6bf0bffSzoujr// e 285c6bf0bffSzoujr// } 286c6bf0bffSzoujr// } 287c6bf0bffSzoujr 2881ca0e4f3SYinan Xuclass FTB(implicit p: Parameters) extends BasePredictor with FTBParams with BPUUtils 2891ca0e4f3SYinan Xu with HasCircularQueuePtrHelper with HasPerfEvents { 29009c6f1ddSLingrui98 override val meta_size = WireInit(0.U.asTypeOf(new FTBMeta)).getWidth 29109c6f1ddSLingrui98 29209c6f1ddSLingrui98 val ftbAddr = new TableAddr(log2Up(numSets), 1) 29309c6f1ddSLingrui98 29409c6f1ddSLingrui98 class FTBBank(val numSets: Int, val nWays: Int) extends XSModule with BPUUtils { 29509c6f1ddSLingrui98 val io = IO(new Bundle { 2965371700eSzoujr val s1_fire = Input(Bool()) 29709c6f1ddSLingrui98 29809c6f1ddSLingrui98 // when ftb hit, read_hits.valid is true, and read_hits.bits is OH of hit way 29909c6f1ddSLingrui98 // when ftb not hit, read_hits.valid is false, and read_hits is OH of allocWay 300bb09c7feSzoujr // val read_hits = Valid(Vec(numWays, Bool())) 3011c8d9e26Szoujr val req_pc = Flipped(DecoupledIO(UInt(VAddrBits.W))) 3021c8d9e26Szoujr val read_resp = Output(new FTBEntry) 303bb09c7feSzoujr val read_hits = Valid(UInt(log2Ceil(numWays).W)) 30409c6f1ddSLingrui98 3051c8d9e26Szoujr val u_req_pc = Flipped(DecoupledIO(UInt(VAddrBits.W))) 3061c8d9e26Szoujr val update_hits = Valid(UInt(log2Ceil(numWays).W)) 3071c8d9e26Szoujr val update_access = Input(Bool()) 30809c6f1ddSLingrui98 30909c6f1ddSLingrui98 val update_pc = Input(UInt(VAddrBits.W)) 31009c6f1ddSLingrui98 val update_write_data = Flipped(Valid(new FTBEntryWithTag)) 311c6bf0bffSzoujr val update_write_way = Input(UInt(log2Ceil(numWays).W)) 312c6bf0bffSzoujr val update_write_alloc = Input(Bool()) 31309c6f1ddSLingrui98 }) 31409c6f1ddSLingrui98 3151c8d9e26Szoujr // Extract holdRead logic to fix bug that update read override predict read result 3166fe623afSLingrui98 val ftb = Module(new SRAMTemplate(new FTBEntryWithTag, set = numSets, way = numWays, shouldReset = true, holdRead = false, singlePort = true)) 317ccd953deSSteve Gou val ftb_r_entries = ftb.io.r.resp.data.map(_.entry) 31809c6f1ddSLingrui98 3191c8d9e26Szoujr val pred_rdata = HoldUnless(ftb.io.r.resp.data, RegNext(io.req_pc.valid && !io.update_access)) 3201c8d9e26Szoujr ftb.io.r.req.valid := io.req_pc.valid || io.u_req_pc.valid // io.s0_fire 3211c8d9e26Szoujr ftb.io.r.req.bits.setIdx := Mux(io.u_req_pc.valid, ftbAddr.getIdx(io.u_req_pc.bits), ftbAddr.getIdx(io.req_pc.bits)) // s0_idx 3221c8d9e26Szoujr 3231c8d9e26Szoujr assert(!(io.req_pc.valid && io.u_req_pc.valid)) 32409c6f1ddSLingrui98 32509c6f1ddSLingrui98 io.req_pc.ready := ftb.io.r.req.ready 3261c8d9e26Szoujr io.u_req_pc.ready := ftb.io.r.req.ready 32709c6f1ddSLingrui98 32809c6f1ddSLingrui98 val req_tag = RegEnable(ftbAddr.getTag(io.req_pc.bits)(tagSize-1, 0), io.req_pc.valid) 329ac3f6f25Szoujr val req_idx = RegEnable(ftbAddr.getIdx(io.req_pc.bits), io.req_pc.valid) 33009c6f1ddSLingrui98 3311c8d9e26Szoujr val u_req_tag = RegEnable(ftbAddr.getTag(io.u_req_pc.bits)(tagSize-1, 0), io.u_req_pc.valid) 33209c6f1ddSLingrui98 3331c8d9e26Szoujr val read_entries = pred_rdata.map(_.entry) 3341c8d9e26Szoujr val read_tags = pred_rdata.map(_.tag) 3351c8d9e26Szoujr 3361c8d9e26Szoujr val total_hits = VecInit((0 until numWays).map(b => read_tags(b) === req_tag && read_entries(b).valid && io.s1_fire)) 33709c6f1ddSLingrui98 val hit = total_hits.reduce(_||_) 338bb09c7feSzoujr // val hit_way_1h = VecInit(PriorityEncoderOH(total_hits)) 339ab890bfeSLingrui98 val hit_way = OHToUInt(total_hits) 34009c6f1ddSLingrui98 3411c8d9e26Szoujr val u_total_hits = VecInit((0 until numWays).map(b => 3421c8d9e26Szoujr ftb.io.r.resp.data(b).tag === u_req_tag && ftb.io.r.resp.data(b).entry.valid && RegNext(io.update_access))) 3431c8d9e26Szoujr val u_hit = u_total_hits.reduce(_||_) 3441c8d9e26Szoujr // val hit_way_1h = VecInit(PriorityEncoderOH(total_hits)) 345ab890bfeSLingrui98 val u_hit_way = OHToUInt(u_total_hits) 3461c8d9e26Szoujr 347ccd953deSSteve Gou // assert(PopCount(total_hits) === 1.U || PopCount(total_hits) === 0.U) 348ccd953deSSteve Gou // assert(PopCount(u_total_hits) === 1.U || PopCount(u_total_hits) === 0.U) 349ccd953deSSteve Gou for (n <- 1 to numWays) { 350ccd953deSSteve Gou XSPerfAccumulate(f"ftb_pred_${n}_way_hit", PopCount(total_hits) === n.U) 351ccd953deSSteve Gou XSPerfAccumulate(f"ftb_update_${n}_way_hit", PopCount(u_total_hits) === n.U) 352ccd953deSSteve Gou } 35309c6f1ddSLingrui98 354ac3f6f25Szoujr val replacer = ReplacementPolicy.fromString(Some("setplru"), numWays, numSets) 355c6bf0bffSzoujr // val allocWriteWay = replacer.way(req_idx) 35609c6f1ddSLingrui98 357ac3f6f25Szoujr val touch_set = Seq.fill(1)(Wire(UInt(log2Ceil(numSets).W))) 358ac3f6f25Szoujr val touch_way = Seq.fill(1)(Wire(Valid(UInt(log2Ceil(numWays).W)))) 359ac3f6f25Szoujr 360a788562dSSteve Gou val write_set = Wire(UInt(log2Ceil(numSets).W)) 361a788562dSSteve Gou val write_way = Wire(Valid(UInt(log2Ceil(numWays).W))) 362ac3f6f25Szoujr 363a788562dSSteve Gou val read_set = Wire(UInt(log2Ceil(numSets).W)) 364a788562dSSteve Gou val read_way = Wire(Valid(UInt(log2Ceil(numWays).W))) 365a788562dSSteve Gou 366a788562dSSteve Gou read_set := req_idx 367a788562dSSteve Gou read_way.valid := hit 368a788562dSSteve Gou read_way.bits := hit_way 369a788562dSSteve Gou 37021bd6001SEaston Man // Read replacer access is postponed for 1 cycle 37121bd6001SEaston Man // this helps timing 37221bd6001SEaston Man touch_set(0) := Mux(write_way.valid, write_set, RegNext(read_set)) 37321bd6001SEaston Man touch_way(0).valid := write_way.valid || RegNext(read_way.valid) 37421bd6001SEaston Man touch_way(0).bits := Mux(write_way.valid, write_way.bits, RegNext(read_way.bits)) 375ac3f6f25Szoujr 376c6bf0bffSzoujr replacer.access(touch_set, touch_way) 377c6bf0bffSzoujr 37821bd6001SEaston Man // Select the update allocate way 37921bd6001SEaston Man // Selection logic: 38021bd6001SEaston Man // 1. if any entries within the same index is not valid, select it 38121bd6001SEaston Man // 2. if all entries is valid, use replacer 38202f21c16SLingrui98 def allocWay(valids: UInt, idx: UInt): UInt = { 38309c6f1ddSLingrui98 if (numWays > 1) { 38409c6f1ddSLingrui98 val w = Wire(UInt(log2Up(numWays).W)) 38509c6f1ddSLingrui98 val valid = WireInit(valids.andR) 3865371700eSzoujr w := Mux(valid, replacer.way(idx), PriorityEncoder(~valids)) 38709c6f1ddSLingrui98 w 38809c6f1ddSLingrui98 } else { 38902f21c16SLingrui98 val w = WireInit(0.U(log2Up(numWays).W)) 39009c6f1ddSLingrui98 w 39109c6f1ddSLingrui98 } 39209c6f1ddSLingrui98 } 39309c6f1ddSLingrui98 394ab890bfeSLingrui98 io.read_resp := Mux1H(total_hits, read_entries) // Mux1H 39509c6f1ddSLingrui98 io.read_hits.valid := hit 3965371700eSzoujr io.read_hits.bits := hit_way 39709c6f1ddSLingrui98 3981c8d9e26Szoujr io.update_hits.valid := u_hit 3991c8d9e26Szoujr io.update_hits.bits := u_hit_way 4001c8d9e26Szoujr 40109c6f1ddSLingrui98 // Update logic 40209c6f1ddSLingrui98 val u_valid = io.update_write_data.valid 40309c6f1ddSLingrui98 val u_data = io.update_write_data.bits 40409c6f1ddSLingrui98 val u_idx = ftbAddr.getIdx(io.update_pc) 40502f21c16SLingrui98 val allocWriteWay = allocWay(RegNext(VecInit(ftb_r_entries.map(_.valid))).asUInt, u_idx) 40602f21c16SLingrui98 val u_way = Mux(io.update_write_alloc, allocWriteWay, io.update_write_way) 40702f21c16SLingrui98 val u_mask = UIntToOH(u_way) 408c6bf0bffSzoujr 409c6bf0bffSzoujr for (i <- 0 until numWays) { 41002f21c16SLingrui98 XSPerfAccumulate(f"ftb_replace_way$i", u_valid && io.update_write_alloc && u_way === i.U) 41102f21c16SLingrui98 XSPerfAccumulate(f"ftb_replace_way${i}_has_empty", u_valid && io.update_write_alloc && !ftb_r_entries.map(_.valid).reduce(_&&_) && u_way === i.U) 4125371700eSzoujr XSPerfAccumulate(f"ftb_hit_way$i", hit && !io.update_access && hit_way === i.U) 413c6bf0bffSzoujr } 41409c6f1ddSLingrui98 41509c6f1ddSLingrui98 ftb.io.w.apply(u_valid, u_data, u_idx, u_mask) 416eeb5ff92SLingrui98 417a788562dSSteve Gou // for replacer 418f4e1af07SLingrui98 write_set := u_idx 419f4e1af07SLingrui98 write_way.valid := u_valid 420f4e1af07SLingrui98 write_way.bits := Mux(io.update_write_alloc, allocWriteWay, io.update_write_way) 421a788562dSSteve Gou 422eeb5ff92SLingrui98 // print hit entry info 423ab890bfeSLingrui98 Mux1H(total_hits, ftb.io.r.resp.data).display(true.B) 42409c6f1ddSLingrui98 } // FTBBank 42509c6f1ddSLingrui98 42609c6f1ddSLingrui98 val ftbBank = Module(new FTBBank(numSets, numWays)) 42709c6f1ddSLingrui98 428adc0b8dfSGuokai Chen ftbBank.io.req_pc.valid := io.s0_fire(0) 429adc0b8dfSGuokai Chen ftbBank.io.req_pc.bits := s0_pc_dup(0) 43009c6f1ddSLingrui98 431adc0b8dfSGuokai Chen val btb_enable_dup = dup(RegNext(io.ctrl.btb_enable)) 432adc0b8dfSGuokai Chen val s2_ftb_entry_dup = io.s1_fire.map(f => RegEnable(ftbBank.io.read_resp, f)) 433adc0b8dfSGuokai Chen val s3_ftb_entry_dup = io.s2_fire.zip(s2_ftb_entry_dup).map {case (f, e) => RegEnable(e, f)} 434adc0b8dfSGuokai Chen 4356ee06c7aSSteve Gou val s1_hit = ftbBank.io.read_hits.valid && io.ctrl.btb_enable 436*c89b4642SGuokai Chen val s2_hit_dup = io.s1_fire.map(f => RegEnable(s1_hit, 0.B, f)) 437*c89b4642SGuokai Chen val s3_hit_dup = io.s2_fire.zip(s2_hit_dup).map {case (f, h) => RegEnable(h, 0.B, f)} 43809c6f1ddSLingrui98 val writeWay = ftbBank.io.read_hits.bits 43909c6f1ddSLingrui98 44009c6f1ddSLingrui98 // io.out.bits.resp := RegEnable(io.in.bits.resp_in(0), 0.U.asTypeOf(new BranchPredictionResp), io.s1_fire) 441c2d1ec7dSLingrui98 io.out := io.in.bits.resp_in(0) 44209c6f1ddSLingrui98 443adc0b8dfSGuokai Chen io.out.s2.full_pred.zip(s2_hit_dup).map {case (fp, h) => fp.hit := h} 444adc0b8dfSGuokai Chen io.out.s2.pc := s2_pc_dup 445adc0b8dfSGuokai Chen for (full_pred & s2_ftb_entry & s2_pc & s1_pc & s1_fire <- 446adc0b8dfSGuokai Chen io.out.s2.full_pred zip s2_ftb_entry_dup zip s2_pc_dup zip s1_pc_dup zip io.s1_fire) { 44747c003a9SEaston Man full_pred.fromFtbEntry(s2_ftb_entry, 44847c003a9SEaston Man s2_pc, 44947c003a9SEaston Man // Previous stage meta for better timing 45047c003a9SEaston Man Some(s1_pc, s1_fire), 45147c003a9SEaston Man Some(ftbBank.io.read_resp, s1_fire) 45247c003a9SEaston Man ) 453adc0b8dfSGuokai Chen } 45409c6f1ddSLingrui98 455adc0b8dfSGuokai Chen io.out.s3.full_pred.zip(s3_hit_dup).map {case (fp, h) => fp.hit := h} 456adc0b8dfSGuokai Chen io.out.s3.pc := s3_pc_dup 457adc0b8dfSGuokai Chen for (full_pred & s3_ftb_entry & s3_pc & s2_pc & s2_fire <- 458adc0b8dfSGuokai Chen io.out.s3.full_pred zip s3_ftb_entry_dup zip s3_pc_dup zip s2_pc_dup zip io.s2_fire) 459adc0b8dfSGuokai Chen full_pred.fromFtbEntry(s3_ftb_entry, s3_pc, Some((s2_pc, s2_fire))) 460cb4f77ceSLingrui98 461adc0b8dfSGuokai Chen io.out.last_stage_ftb_entry := s3_ftb_entry_dup(0) 462adc0b8dfSGuokai Chen io.out.last_stage_meta := RegEnable(RegEnable(FTBMeta(writeWay.asUInt(), s1_hit, GTimer()).asUInt(), io.s1_fire(0)), io.s2_fire(0)) 46309c6f1ddSLingrui98 46409c6f1ddSLingrui98 // always taken logic 46509c6f1ddSLingrui98 for (i <- 0 until numBr) { 466adc0b8dfSGuokai Chen for (out_fp & in_fp & s2_hit & s2_ftb_entry <- 467adc0b8dfSGuokai Chen io.out.s2.full_pred zip io.in.bits.resp_in(0).s2.full_pred zip s2_hit_dup zip s2_ftb_entry_dup) 468adc0b8dfSGuokai Chen out_fp.br_taken_mask(i) := in_fp.br_taken_mask(i) || s2_hit && s2_ftb_entry.always_taken(i) 469adc0b8dfSGuokai Chen for (out_fp & in_fp & s3_hit & s3_ftb_entry <- 470adc0b8dfSGuokai Chen io.out.s3.full_pred zip io.in.bits.resp_in(0).s3.full_pred zip s3_hit_dup zip s3_ftb_entry_dup) 471adc0b8dfSGuokai Chen out_fp.br_taken_mask(i) := in_fp.br_taken_mask(i) || s3_hit && s3_ftb_entry.always_taken(i) 47209c6f1ddSLingrui98 } 47309c6f1ddSLingrui98 47409c6f1ddSLingrui98 // Update logic 47502f21c16SLingrui98 val update = io.update.bits 476c6bf0bffSzoujr 47709c6f1ddSLingrui98 val u_meta = update.meta.asTypeOf(new FTBMeta) 47802f21c16SLingrui98 val u_valid = io.update.valid && !io.update.bits.old_entry 479bb09c7feSzoujr 48002f21c16SLingrui98 val delay2_pc = DelayN(update.pc, 2) 48102f21c16SLingrui98 val delay2_entry = DelayN(update.ftb_entry, 2) 482bb09c7feSzoujr 48302f21c16SLingrui98 484c6bf0bffSzoujr val update_now = u_valid && u_meta.hit 48502f21c16SLingrui98 val update_need_read = u_valid && !u_meta.hit 48602f21c16SLingrui98 // stall one more cycle because we use a whole cycle to do update read tag hit 48702f21c16SLingrui98 io.s1_ready := ftbBank.io.req_pc.ready && !(update_need_read) && !RegNext(update_need_read) 488c6bf0bffSzoujr 48902f21c16SLingrui98 ftbBank.io.u_req_pc.valid := update_need_read 4901c8d9e26Szoujr ftbBank.io.u_req_pc.bits := update.pc 491bb09c7feSzoujr 492bb09c7feSzoujr 49309c6f1ddSLingrui98 49409c6f1ddSLingrui98 val ftb_write = Wire(new FTBEntryWithTag) 49502f21c16SLingrui98 ftb_write.entry := Mux(update_now, update.ftb_entry, delay2_entry) 49602f21c16SLingrui98 ftb_write.tag := ftbAddr.getTag(Mux(update_now, update.pc, delay2_pc))(tagSize-1, 0) 49709c6f1ddSLingrui98 49802f21c16SLingrui98 val write_valid = update_now || DelayN(u_valid && !u_meta.hit, 2) 499c6bf0bffSzoujr 500c6bf0bffSzoujr ftbBank.io.update_write_data.valid := write_valid 50109c6f1ddSLingrui98 ftbBank.io.update_write_data.bits := ftb_write 50202f21c16SLingrui98 ftbBank.io.update_pc := Mux(update_now, update.pc, delay2_pc) 50302f21c16SLingrui98 ftbBank.io.update_write_way := Mux(update_now, u_meta.writeWay, RegNext(ftbBank.io.update_hits.bits)) // use it one cycle later 50402f21c16SLingrui98 ftbBank.io.update_write_alloc := Mux(update_now, false.B, RegNext(!ftbBank.io.update_hits.valid)) // use it one cycle later 5051c8d9e26Szoujr ftbBank.io.update_access := u_valid && !u_meta.hit 506adc0b8dfSGuokai Chen ftbBank.io.s1_fire := io.s1_fire(0) 50709c6f1ddSLingrui98 508adc0b8dfSGuokai Chen XSDebug("req_v=%b, req_pc=%x, ready=%b (resp at next cycle)\n", io.s0_fire(0), s0_pc_dup(0), ftbBank.io.req_pc.ready) 509adc0b8dfSGuokai Chen XSDebug("s2_hit=%b, hit_way=%b\n", s2_hit_dup(0), writeWay.asUInt) 510eeb5ff92SLingrui98 XSDebug("s2_br_taken_mask=%b, s2_real_taken_mask=%b\n", 511adc0b8dfSGuokai Chen io.in.bits.resp_in(0).s2.full_pred(0).br_taken_mask.asUInt, io.out.s2.full_pred(0).real_slot_taken_mask().asUInt) 512adc0b8dfSGuokai Chen XSDebug("s2_target=%x\n", io.out.s2.getTarget(0)) 51309c6f1ddSLingrui98 514adc0b8dfSGuokai Chen s2_ftb_entry_dup(0).display(true.B) 51509c6f1ddSLingrui98 516adc0b8dfSGuokai Chen XSPerfAccumulate("ftb_read_hits", RegNext(io.s0_fire(0)) && s1_hit) 517adc0b8dfSGuokai Chen XSPerfAccumulate("ftb_read_misses", RegNext(io.s0_fire(0)) && !s1_hit) 51809c6f1ddSLingrui98 51902f21c16SLingrui98 XSPerfAccumulate("ftb_commit_hits", io.update.valid && u_meta.hit) 52002f21c16SLingrui98 XSPerfAccumulate("ftb_commit_misses", io.update.valid && !u_meta.hit) 52109c6f1ddSLingrui98 52209c6f1ddSLingrui98 XSPerfAccumulate("ftb_update_req", io.update.valid) 52309c6f1ddSLingrui98 XSPerfAccumulate("ftb_update_ignored", io.update.valid && io.update.bits.old_entry) 52409c6f1ddSLingrui98 XSPerfAccumulate("ftb_updated", u_valid) 525cd365d4cSrvcoresjw 5264813e060SLingrui98 override val perfEvents = Seq( 527adc0b8dfSGuokai Chen ("ftb_commit_hits ", io.update.valid && u_meta.hit), 528adc0b8dfSGuokai Chen ("ftb_commit_misses ", io.update.valid && !u_meta.hit), 529cd365d4cSrvcoresjw ) 5301ca0e4f3SYinan Xu generatePerfEvent() 53109c6f1ddSLingrui98} 532