xref: /XiangShan/src/main/scala/xiangshan/frontend/PreDecode.scala (revision f7af4c746b893ede5aa64c681f8da182c602efe0)
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
198891a219SYinan Xuimport org.chipsalliance.cde.config.Parameters
2009c6f1ddSLingrui98import freechips.rocketchip.rocket.{RVCDecoder, ExpandedInstruction}
2109c6f1ddSLingrui98import chisel3.{util, _}
2209c6f1ddSLingrui98import chisel3.util._
2309c6f1ddSLingrui98import utils._
243c02ee8fSwakafaimport utility._
2509c6f1ddSLingrui98import xiangshan._
261d8f4dcbSJayimport xiangshan.frontend.icache._
2709c6f1ddSLingrui98import xiangshan.backend.decode.isa.predecode.PreDecodeInst
28330aad7fSGuokai Chenimport java.lang.reflect.Parameter
29*f7af4c74Schengguanghuiimport xiangshan.backend.fu.util.SdtrigExt
3009c6f1ddSLingrui98
3109c6f1ddSLingrui98trait HasPdConst extends HasXSParameter with HasICacheParameters with HasIFUConst{
3209c6f1ddSLingrui98  def isRVC(inst: UInt) = (inst(1,0) =/= 3.U)
3309c6f1ddSLingrui98  def isLink(reg:UInt) = reg === 1.U || reg === 5.U
3409c6f1ddSLingrui98  def brInfo(instr: UInt) = {
3509c6f1ddSLingrui98    val brType::Nil = ListLookup(instr, List(BrType.notCFI), PreDecodeInst.brTable)
3609c6f1ddSLingrui98    val rd = Mux(isRVC(instr), instr(12), instr(11,7))
3709c6f1ddSLingrui98    val rs = Mux(isRVC(instr), Mux(brType === BrType.jal, 0.U, instr(11, 7)), instr(19, 15))
3809c6f1ddSLingrui98    val isCall = (brType === BrType.jal && !isRVC(instr) || brType === BrType.jalr) && isLink(rd) // Only for RV64
3909c6f1ddSLingrui98    val isRet = brType === BrType.jalr && isLink(rs) && !isCall
4009c6f1ddSLingrui98    List(brType, isCall, isRet)
4109c6f1ddSLingrui98  }
4209c6f1ddSLingrui98  def jal_offset(inst: UInt, rvc: Bool): UInt = {
4309c6f1ddSLingrui98    val rvc_offset = Cat(inst(12), inst(8), inst(10, 9), inst(6), inst(7), inst(2), inst(11), inst(5, 3), 0.U(1.W))
4409c6f1ddSLingrui98    val rvi_offset = Cat(inst(31), inst(19, 12), inst(20), inst(30, 21), 0.U(1.W))
4509c6f1ddSLingrui98    val max_width = rvi_offset.getWidth
4609c6f1ddSLingrui98    SignExt(Mux(rvc, SignExt(rvc_offset, max_width), SignExt(rvi_offset, max_width)), XLEN)
4709c6f1ddSLingrui98  }
4809c6f1ddSLingrui98  def br_offset(inst: UInt, rvc: Bool): UInt = {
4909c6f1ddSLingrui98    val rvc_offset = Cat(inst(12), inst(6, 5), inst(2), inst(11, 10), inst(4, 3), 0.U(1.W))
5009c6f1ddSLingrui98    val rvi_offset = Cat(inst(31), inst(7), inst(30, 25), inst(11, 8), 0.U(1.W))
5109c6f1ddSLingrui98    val max_width = rvi_offset.getWidth
5209c6f1ddSLingrui98    SignExt(Mux(rvc, SignExt(rvc_offset, max_width), SignExt(rvi_offset, max_width)), XLEN)
5309c6f1ddSLingrui98  }
5409c6f1ddSLingrui98
5509c6f1ddSLingrui98  def NOP = "h4501".U(16.W)
5609c6f1ddSLingrui98}
5709c6f1ddSLingrui98
5809c6f1ddSLingrui98object BrType {
5909c6f1ddSLingrui98  def notCFI   = "b00".U
6009c6f1ddSLingrui98  def branch  = "b01".U
6109c6f1ddSLingrui98  def jal     = "b10".U
6209c6f1ddSLingrui98  def jalr    = "b11".U
6309c6f1ddSLingrui98  def apply() = UInt(2.W)
6409c6f1ddSLingrui98}
6509c6f1ddSLingrui98
6609c6f1ddSLingrui98object ExcType {  //TODO:add exctype
6709c6f1ddSLingrui98  def notExc = "b000".U
6809c6f1ddSLingrui98  def apply() = UInt(3.W)
6909c6f1ddSLingrui98}
7009c6f1ddSLingrui98
7109c6f1ddSLingrui98class PreDecodeInfo extends Bundle {  // 8 bit
7209c6f1ddSLingrui98  val valid   = Bool()
7309c6f1ddSLingrui98  val isRVC   = Bool()
7409c6f1ddSLingrui98  val brType  = UInt(2.W)
7509c6f1ddSLingrui98  val isCall  = Bool()
7609c6f1ddSLingrui98  val isRet   = Bool()
7709c6f1ddSLingrui98  //val excType = UInt(3.W)
7809c6f1ddSLingrui98  def isBr    = brType === BrType.branch
7909c6f1ddSLingrui98  def isJal   = brType === BrType.jal
8009c6f1ddSLingrui98  def isJalr  = brType === BrType.jalr
8109c6f1ddSLingrui98  def notCFI  = brType === BrType.notCFI
8209c6f1ddSLingrui98}
8309c6f1ddSLingrui98
8409c6f1ddSLingrui98class PreDecodeResp(implicit p: Parameters) extends XSBundle with HasPdConst {
85a4e57ea3SLi Qianruo  val pd = Vec(PredictWidth, new PreDecodeInfo)
86a4e57ea3SLi Qianruo  val hasHalfValid = Vec(PredictWidth, Bool())
8748a62719SJenius  //val expInstr = Vec(PredictWidth, UInt(32.W))
8848a62719SJenius  val instr      = Vec(PredictWidth, UInt(32.W))
89a4e57ea3SLi Qianruo  val jumpOffset = Vec(PredictWidth, UInt(XLEN.W))
90a4e57ea3SLi Qianruo//  val hasLastHalf = Bool()
9172951335SLi Qianruo  val triggered    = Vec(PredictWidth, new TriggerCf)
9209c6f1ddSLingrui98}
9309c6f1ddSLingrui98
9409c6f1ddSLingrui98class PreDecode(implicit p: Parameters) extends XSModule with HasPdConst{
9509c6f1ddSLingrui98  val io = IO(new Bundle() {
9609c6f1ddSLingrui98    val in = Input(new IfuToPreDecode)
9709c6f1ddSLingrui98    val out = Output(new PreDecodeResp)
9809c6f1ddSLingrui98  })
9909c6f1ddSLingrui98
10009c6f1ddSLingrui98  val data          = io.in.data
101a4e57ea3SLi Qianruo//  val lastHalfMatch = io.in.lastHalfMatch
102a4e57ea3SLi Qianruo  val validStart, validEnd = Wire(Vec(PredictWidth, Bool()))
103a4e57ea3SLi Qianruo  val h_validStart, h_validEnd = Wire(Vec(PredictWidth, Bool()))
10409c6f1ddSLingrui98
105330aad7fSGuokai Chen  val validStart_half, validEnd_half = Wire(Vec(PredictWidth, Bool()))
106330aad7fSGuokai Chen  val h_validStart_half, h_validEnd_half = Wire(Vec(PredictWidth, Bool()))
107330aad7fSGuokai Chen
108330aad7fSGuokai Chen  val validStart_halfPlus1, validEnd_halfPlus1 = Wire(Vec(PredictWidth, Bool()))
109330aad7fSGuokai Chen  val h_validStart_halfPlus1, h_validEnd_halfPlus1 = Wire(Vec(PredictWidth, Bool()))
110330aad7fSGuokai Chen
111330aad7fSGuokai Chen  val validStart_diff, validEnd_diff = Wire(Vec(PredictWidth, Bool()))
112330aad7fSGuokai Chen  val h_validStart_diff, h_validEnd_diff = Wire(Vec(PredictWidth, Bool()))
113330aad7fSGuokai Chen
114330aad7fSGuokai Chen  val currentIsRVC = Wire(Vec(PredictWidth, Bool()))
115330aad7fSGuokai Chen
116330aad7fSGuokai Chen  validStart_half.map(_ := false.B)
117330aad7fSGuokai Chen  validEnd_half.map(_ := false.B)
118330aad7fSGuokai Chen  h_validStart_half.map(_ := false.B)
119330aad7fSGuokai Chen  h_validEnd_half.map(_ := false.B)
120330aad7fSGuokai Chen
121330aad7fSGuokai Chen  validStart_halfPlus1.map(_ := false.B)
122330aad7fSGuokai Chen  validEnd_halfPlus1.map(_ := false.B)
123330aad7fSGuokai Chen  h_validStart_halfPlus1.map(_ := false.B)
124330aad7fSGuokai Chen  h_validEnd_halfPlus1.map(_ := false.B)
125330aad7fSGuokai Chen
126330aad7fSGuokai Chen  dontTouch(validStart_half)
127330aad7fSGuokai Chen  dontTouch(validEnd_half)
128330aad7fSGuokai Chen  dontTouch(h_validStart_half)
129330aad7fSGuokai Chen  dontTouch(h_validEnd_half)
130330aad7fSGuokai Chen  dontTouch(validStart_halfPlus1)
131330aad7fSGuokai Chen  dontTouch(validEnd_halfPlus1)
132330aad7fSGuokai Chen  dontTouch(h_validStart_halfPlus1)
133330aad7fSGuokai Chen  dontTouch(h_validEnd_halfPlus1)
134330aad7fSGuokai Chen  dontTouch(validStart_diff)
135330aad7fSGuokai Chen  dontTouch(validEnd_diff)
136330aad7fSGuokai Chen  dontTouch(h_validStart_diff)
137330aad7fSGuokai Chen  dontTouch(h_validEnd_diff)
138330aad7fSGuokai Chen  dontTouch(validStart)
139330aad7fSGuokai Chen  dontTouch(validEnd)
140330aad7fSGuokai Chen  dontTouch(h_validStart)
141330aad7fSGuokai Chen  dontTouch(h_validEnd)
142330aad7fSGuokai Chen
14309c6f1ddSLingrui98  val rawInsts = if (HasCExtension) VecInit((0 until PredictWidth).map(i => Cat(data(i+1), data(i))))
14409c6f1ddSLingrui98  else         VecInit((0 until PredictWidth).map(i => data(i)))
14509c6f1ddSLingrui98
14609c6f1ddSLingrui98  for (i <- 0 until PredictWidth) {
147a4e57ea3SLi Qianruo    val inst           = WireInit(rawInsts(i))
14848a62719SJenius    //val expander       = Module(new RVCExpander)
149330aad7fSGuokai Chen    currentIsRVC(i)   := isRVC(inst)
150a4e57ea3SLi Qianruo    val currentPC      = io.in.pc(i)
15148a62719SJenius    //expander.io.in             := inst
15209c6f1ddSLingrui98
15309c6f1ddSLingrui98    val brType::isCall::isRet::Nil = brInfo(inst)
154330aad7fSGuokai Chen    val jalOffset = jal_offset(inst, currentIsRVC(i))
155330aad7fSGuokai Chen    val brOffset  = br_offset(inst, currentIsRVC(i))
156a4e57ea3SLi Qianruo
157a4e57ea3SLi Qianruo    io.out.hasHalfValid(i)        := h_validStart(i)
158a4e57ea3SLi Qianruo
159a4e57ea3SLi Qianruo    io.out.triggered(i)   := DontCare//VecInit(Seq.fill(10)(false.B))
160a4e57ea3SLi Qianruo
161a4e57ea3SLi Qianruo
162a4e57ea3SLi Qianruo    io.out.pd(i).valid         := validStart(i)
163330aad7fSGuokai Chen    io.out.pd(i).isRVC         := currentIsRVC(i)
164330aad7fSGuokai Chen
165330aad7fSGuokai Chen    // for diff purpose only
16609c6f1ddSLingrui98    io.out.pd(i).brType        := brType
16709c6f1ddSLingrui98    io.out.pd(i).isCall        := isCall
16809c6f1ddSLingrui98    io.out.pd(i).isRet         := isRet
169ddb65c47SLi Qianruo
17048a62719SJenius    //io.out.expInstr(i)         := expander.io.out.bits
17148a62719SJenius    io.out.instr(i)              :=inst
172a4e57ea3SLi Qianruo    io.out.jumpOffset(i)       := Mux(io.out.pd(i).isBr, brOffset, jalOffset)
17309c6f1ddSLingrui98  }
17409c6f1ddSLingrui98
175330aad7fSGuokai Chen  // the first half is always reliable
176330aad7fSGuokai Chen  for (i <- 0 until PredictWidth / 2) {
177330aad7fSGuokai Chen    val lastIsValidEnd =   if (i == 0) { true.B } else { validEnd(i-1) || !HasCExtension.B }
178330aad7fSGuokai Chen    validStart(i)   := (lastIsValidEnd || !HasCExtension.B)
179330aad7fSGuokai Chen    validEnd(i)     := validStart(i) && currentIsRVC(i) || !validStart(i) || !HasCExtension.B
180330aad7fSGuokai Chen
181330aad7fSGuokai Chen    //prepared for last half match
182330aad7fSGuokai Chen    val h_lastIsValidEnd = if (i == 0) { false.B } else { h_validEnd(i-1) || !HasCExtension.B }
183330aad7fSGuokai Chen    h_validStart(i)   := (h_lastIsValidEnd || !HasCExtension.B)
184330aad7fSGuokai Chen    h_validEnd(i)     := h_validStart(i) && currentIsRVC(i) || !h_validStart(i) || !HasCExtension.B
185330aad7fSGuokai Chen  }
186330aad7fSGuokai Chen
187330aad7fSGuokai Chen  for (i <- 0 until PredictWidth) {
188330aad7fSGuokai Chen    val lastIsValidEnd =   if (i == 0) { true.B } else { validEnd_diff(i-1) || !HasCExtension.B }
189330aad7fSGuokai Chen    validStart_diff(i)   := (lastIsValidEnd || !HasCExtension.B)
190330aad7fSGuokai Chen    validEnd_diff(i)     := validStart_diff(i) && currentIsRVC(i) || !validStart_diff(i) || !HasCExtension.B
191330aad7fSGuokai Chen
192330aad7fSGuokai Chen    //prepared for last half match
193330aad7fSGuokai Chen    val h_lastIsValidEnd = if (i == 0) { false.B } else { h_validEnd_diff(i-1) || !HasCExtension.B }
194330aad7fSGuokai Chen    h_validStart_diff(i)   := (h_lastIsValidEnd || !HasCExtension.B)
195330aad7fSGuokai Chen    h_validEnd_diff(i)     := h_validStart_diff(i) && currentIsRVC(i) || !h_validStart_diff(i) || !HasCExtension.B
196330aad7fSGuokai Chen  }
197330aad7fSGuokai Chen
198330aad7fSGuokai Chen  // assume PredictWidth / 2 is a valid start
199330aad7fSGuokai Chen  for (i <- PredictWidth / 2 until PredictWidth) {
200330aad7fSGuokai Chen    val lastIsValidEnd =   if (i == PredictWidth / 2) { true.B } else { validEnd_half(i-1) || !HasCExtension.B }
201330aad7fSGuokai Chen    validStart_half(i)   := (lastIsValidEnd || !HasCExtension.B)
202330aad7fSGuokai Chen    validEnd_half(i)     := validStart_half(i) && currentIsRVC(i) || !validStart_half(i) || !HasCExtension.B
203330aad7fSGuokai Chen
204330aad7fSGuokai Chen    //prepared for last half match
205330aad7fSGuokai Chen    val h_lastIsValidEnd = if (i == PredictWidth / 2) { true.B } else { h_validEnd_half(i-1) || !HasCExtension.B }
206330aad7fSGuokai Chen    h_validStart_half(i)   := (h_lastIsValidEnd || !HasCExtension.B)
207330aad7fSGuokai Chen    h_validEnd_half(i)     := h_validStart_half(i) && currentIsRVC(i) || !h_validStart_half(i) || !HasCExtension.B
208330aad7fSGuokai Chen  }
209330aad7fSGuokai Chen
210330aad7fSGuokai Chen  // assume PredictWidth / 2 + 1 is a valid start (and PredictWidth / 2 is last half of RVI)
211330aad7fSGuokai Chen  for (i <- PredictWidth / 2 + 1 until PredictWidth) {
212330aad7fSGuokai Chen    val lastIsValidEnd =   if (i == PredictWidth / 2 + 1) { true.B } else { validEnd_halfPlus1(i-1) || !HasCExtension.B }
213330aad7fSGuokai Chen    validStart_halfPlus1(i)   := (lastIsValidEnd || !HasCExtension.B)
214330aad7fSGuokai Chen    validEnd_halfPlus1(i)     := validStart_halfPlus1(i) && currentIsRVC(i) || !validStart_halfPlus1(i) || !HasCExtension.B
215330aad7fSGuokai Chen
216330aad7fSGuokai Chen    //prepared for last half match
217330aad7fSGuokai Chen    val h_lastIsValidEnd = if (i == PredictWidth / 2 + 1) { true.B } else { h_validEnd_halfPlus1(i-1) || !HasCExtension.B }
218330aad7fSGuokai Chen    h_validStart_halfPlus1(i)   := (h_lastIsValidEnd || !HasCExtension.B)
219330aad7fSGuokai Chen    h_validEnd_halfPlus1(i)     := h_validStart_halfPlus1(i) && currentIsRVC(i) || !h_validStart_halfPlus1(i) || !HasCExtension.B
220330aad7fSGuokai Chen  }
221330aad7fSGuokai Chen  validStart_halfPlus1(PredictWidth / 2) := false.B // could be true but when true we select half, not halfPlus1
222330aad7fSGuokai Chen  validEnd_halfPlus1(PredictWidth / 2) := true.B
223330aad7fSGuokai Chen
224330aad7fSGuokai Chen  // assume h_PredictWidth / 2 is an end
225330aad7fSGuokai Chen  h_validStart_halfPlus1(PredictWidth / 2) := false.B // could be true but when true we select half, not halfPlus1
226330aad7fSGuokai Chen  h_validEnd_halfPlus1(PredictWidth / 2) := true.B
227330aad7fSGuokai Chen
228330aad7fSGuokai Chen  // if PredictWidth / 2 - 1 is a valid end, PredictWidth / 2 is a valid start
229330aad7fSGuokai Chen  for (i <- PredictWidth / 2 until PredictWidth) {
230330aad7fSGuokai Chen    validStart(i) := Mux(validEnd(PredictWidth / 2 - 1), validStart_half(i), validStart_halfPlus1(i))
231330aad7fSGuokai Chen    validEnd(i) := Mux(validEnd(PredictWidth / 2 - 1), validEnd_half(i), validEnd_halfPlus1(i))
232330aad7fSGuokai Chen    h_validStart(i) := Mux(h_validEnd(PredictWidth / 2 - 1), h_validStart_half(i), h_validStart_halfPlus1(i))
233330aad7fSGuokai Chen    h_validEnd(i) := Mux(h_validEnd(PredictWidth / 2 - 1), h_validEnd_half(i), h_validEnd_halfPlus1(i))
234330aad7fSGuokai Chen  }
235330aad7fSGuokai Chen
236330aad7fSGuokai Chen  val validStartMismatch = Wire(Bool())
237330aad7fSGuokai Chen  val validEndMismatch = Wire(Bool())
238330aad7fSGuokai Chen  val validH_ValidStartMismatch = Wire(Bool())
239330aad7fSGuokai Chen  val validH_ValidEndMismatch = Wire(Bool())
240330aad7fSGuokai Chen
241330aad7fSGuokai Chen  validStartMismatch := validStart.zip(validStart_diff).map{case(a,b) => a =/= b}.reduce(_||_)
242330aad7fSGuokai Chen  validEndMismatch := validEnd.zip(validEnd_diff).map{case(a,b) => a =/= b}.reduce(_||_)
243330aad7fSGuokai Chen  validH_ValidStartMismatch := h_validStart.zip(h_validStart_diff).map{case(a,b) => a =/= b}.reduce(_||_)
244330aad7fSGuokai Chen  validH_ValidEndMismatch := h_validEnd.zip(h_validEnd_diff).map{case(a,b) => a =/= b}.reduce(_||_)
245330aad7fSGuokai Chen
246330aad7fSGuokai Chen  XSError(validStartMismatch, p"validStart mismatch\n")
247330aad7fSGuokai Chen  XSError(validEndMismatch, p"validEnd mismatch\n")
248330aad7fSGuokai Chen  XSError(validH_ValidStartMismatch, p"h_validStart mismatch\n")
249330aad7fSGuokai Chen  XSError(validH_ValidEndMismatch, p"h_validEnd mismatch\n")
250330aad7fSGuokai Chen
251a4e57ea3SLi Qianruo//  io.out.hasLastHalf := !io.out.pd(PredictWidth - 1).isRVC && io.out.pd(PredictWidth - 1).valid
25209c6f1ddSLingrui98
25309c6f1ddSLingrui98  for (i <- 0 until PredictWidth) {
25409c6f1ddSLingrui98    XSDebug(true.B,
25548a62719SJenius      p"instr ${Hexadecimal(io.out.instr(i))}, " +
25609c6f1ddSLingrui98        p"validStart ${Binary(validStart(i))}, " +
25709c6f1ddSLingrui98        p"validEnd ${Binary(validEnd(i))}, " +
25809c6f1ddSLingrui98        p"isRVC ${Binary(io.out.pd(i).isRVC)}, " +
25909c6f1ddSLingrui98        p"brType ${Binary(io.out.pd(i).brType)}, " +
26009c6f1ddSLingrui98        p"isRet ${Binary(io.out.pd(i).isRet)}, " +
26109c6f1ddSLingrui98        p"isCall ${Binary(io.out.pd(i).isCall)}\n"
26209c6f1ddSLingrui98    )
26309c6f1ddSLingrui98  }
26409c6f1ddSLingrui98}
26509c6f1ddSLingrui98
266330aad7fSGuokai Chenclass IfuToF3PreDecode(implicit p: Parameters) extends XSBundle with HasPdConst {
267330aad7fSGuokai Chen  val instr      = Vec(PredictWidth, UInt(32.W))
268330aad7fSGuokai Chen}
269330aad7fSGuokai Chen
270330aad7fSGuokai Chenclass F3PreDecodeResp(implicit p: Parameters) extends XSBundle with HasPdConst {
271330aad7fSGuokai Chen  val pd = Vec(PredictWidth, new PreDecodeInfo)
272330aad7fSGuokai Chen}
273330aad7fSGuokai Chenclass F3Predecoder(implicit p: Parameters) extends XSModule with HasPdConst {
274330aad7fSGuokai Chen  val io = IO(new Bundle() {
275330aad7fSGuokai Chen    val in = Input(new IfuToF3PreDecode)
276330aad7fSGuokai Chen    val out = Output(new F3PreDecodeResp)
277330aad7fSGuokai Chen  })
278330aad7fSGuokai Chen  io.out.pd.zipWithIndex.map{ case (pd,i) =>
279330aad7fSGuokai Chen    pd.valid := DontCare
280330aad7fSGuokai Chen    pd.isRVC := DontCare
281330aad7fSGuokai Chen    pd.brType := brInfo(io.in.instr(i))(0)
282330aad7fSGuokai Chen    pd.isCall := brInfo(io.in.instr(i))(1)
283330aad7fSGuokai Chen    pd.isRet := brInfo(io.in.instr(i))(2)
284330aad7fSGuokai Chen  }
285330aad7fSGuokai Chen
286330aad7fSGuokai Chen}
287330aad7fSGuokai Chen
28809c6f1ddSLingrui98class RVCExpander(implicit p: Parameters) extends XSModule {
28909c6f1ddSLingrui98  val io = IO(new Bundle {
29009c6f1ddSLingrui98    val in = Input(UInt(32.W))
29109c6f1ddSLingrui98    val out = Output(new ExpandedInstruction)
29209c6f1ddSLingrui98  })
29309c6f1ddSLingrui98
29409c6f1ddSLingrui98  if (HasCExtension) {
2953f6effe4SYinan Xu    io.out := new RVCDecoder(io.in, XLEN, useAddiForMv = true).decode
29609c6f1ddSLingrui98  } else {
2973f6effe4SYinan Xu    io.out := new RVCDecoder(io.in, XLEN, useAddiForMv = true).passthrough
29809c6f1ddSLingrui98  }
29909c6f1ddSLingrui98}
300a4e57ea3SLi Qianruo
301a4e57ea3SLi Qianruo/* ---------------------------------------------------------------------
302a4e57ea3SLi Qianruo * Predict result check
303a4e57ea3SLi Qianruo *
304a4e57ea3SLi Qianruo * ---------------------------------------------------------------------
305a4e57ea3SLi Qianruo */
306a4e57ea3SLi Qianruo
307a4e57ea3SLi Qianruoobject FaultType {
308a4e57ea3SLi Qianruo  def noFault         = "b000".U
309a4e57ea3SLi Qianruo  def jalFault        = "b001".U    //not CFI taken or invalid instruction taken
310a4e57ea3SLi Qianruo  def retFault        = "b010".U    //not CFI taken or invalid instruction taken
311a4e57ea3SLi Qianruo  def targetFault     = "b011".U
3125b3c20f7SJinYue  def notCFIFault    = "b100".U    //not CFI taken or invalid instruction taken
3135b3c20f7SJinYue  def invalidTaken    = "b101".U
314a4e57ea3SLi Qianruo  def apply() = UInt(3.W)
315a4e57ea3SLi Qianruo}
316a4e57ea3SLi Qianruo
317a4e57ea3SLi Qianruoclass CheckInfo extends Bundle {  // 8 bit
318a4e57ea3SLi Qianruo  val value  = UInt(3.W)
319a4e57ea3SLi Qianruo  def isjalFault      = value === FaultType.jalFault
320a4e57ea3SLi Qianruo  def isRetFault      = value === FaultType.retFault
321a4e57ea3SLi Qianruo  def istargetFault   = value === FaultType.targetFault
3225b3c20f7SJinYue  def invalidTakenFault    = value === FaultType.invalidTaken
3235b3c20f7SJinYue  def notCFIFault          = value === FaultType.notCFIFault
324a4e57ea3SLi Qianruo}
325a4e57ea3SLi Qianruo
326a4e57ea3SLi Qianruoclass PredCheckerResp(implicit p: Parameters) extends XSBundle with HasPdConst {
3275995c9e7SJenius  //to Ibuffer write port  (stage 1)
3285995c9e7SJenius  val stage1Out = new Bundle{
329a4e57ea3SLi Qianruo    val fixedRange  = Vec(PredictWidth, Bool())
330a4e57ea3SLi Qianruo    val fixedTaken  = Vec(PredictWidth, Bool())
3315995c9e7SJenius  }
3325995c9e7SJenius  //to Ftq write back port (stage 2)
3335995c9e7SJenius  val stage2Out = new Bundle{
334a4e57ea3SLi Qianruo    val fixedTarget = Vec(PredictWidth, UInt(VAddrBits.W))
335d10ddd67SGuokai Chen    val jalTarget = Vec(PredictWidth, UInt(VAddrBits.W))
336a4e57ea3SLi Qianruo    val fixedMissPred = Vec(PredictWidth,  Bool())
337a4e57ea3SLi Qianruo    val faultType   = Vec(PredictWidth, new CheckInfo)
338a4e57ea3SLi Qianruo  }
3395995c9e7SJenius}
340a4e57ea3SLi Qianruo
341a4e57ea3SLi Qianruo
342a4e57ea3SLi Qianruoclass PredChecker(implicit p: Parameters) extends XSModule with HasPdConst {
343a4e57ea3SLi Qianruo  val io = IO( new Bundle{
344a4e57ea3SLi Qianruo    val in = Input(new IfuToPredChecker)
345a4e57ea3SLi Qianruo    val out = Output(new PredCheckerResp)
346a4e57ea3SLi Qianruo  })
347a4e57ea3SLi Qianruo
348a4e57ea3SLi Qianruo  val (takenIdx, predTaken)     = (io.in.ftqOffset.bits, io.in.ftqOffset.valid)
349a4e57ea3SLi Qianruo  val predTarget                = (io.in.target)
350a4e57ea3SLi Qianruo  val (instrRange, instrValid)  = (io.in.instrRange, io.in.instrValid)
351a4e57ea3SLi Qianruo  val (pds, pc, jumpOffset)     = (io.in.pds, io.in.pc, io.in.jumpOffset)
352a4e57ea3SLi Qianruo
353a4e57ea3SLi Qianruo  val jalFaultVec, retFaultVec, targetFault, notCFITaken, invalidTaken = Wire(Vec(PredictWidth, Bool()))
354a4e57ea3SLi Qianruo
355a4e57ea3SLi Qianruo  /** remask fault may appear together with other faults, but other faults are exclusive
356a4e57ea3SLi Qianruo    * so other f ault mast use fixed mask to keep only one fault would be found and redirect to Ftq
357a4e57ea3SLi Qianruo    * we first detecct remask fault and then use fixedRange to do second check
358a4e57ea3SLi Qianruo    **/
359a4e57ea3SLi Qianruo
3605995c9e7SJenius  //Stage 1: detect remask fault
361a4e57ea3SLi Qianruo  /** first check: remask Fault */
362a4e57ea3SLi Qianruo  jalFaultVec         := VecInit(pds.zipWithIndex.map{case(pd, i) => pd.isJal && instrRange(i) && instrValid(i) && (takenIdx > i.U && predTaken || !predTaken) })
363a4e57ea3SLi Qianruo  retFaultVec         := VecInit(pds.zipWithIndex.map{case(pd, i) => pd.isRet && instrRange(i) && instrValid(i) && (takenIdx > i.U && predTaken || !predTaken) })
364a4e57ea3SLi Qianruo  val remaskFault      = VecInit((0 until PredictWidth).map(i => jalFaultVec(i) || retFaultVec(i)))
365a4e57ea3SLi Qianruo  val remaskIdx        = ParallelPriorityEncoder(remaskFault.asUInt)
366a4e57ea3SLi Qianruo  val needRemask       = ParallelOR(remaskFault)
367a4e57ea3SLi Qianruo  val fixedRange       = instrRange.asUInt & (Fill(PredictWidth, !needRemask) | Fill(PredictWidth, 1.U(1.W)) >> ~remaskIdx)
368a4e57ea3SLi Qianruo
3695995c9e7SJenius  io.out.stage1Out.fixedRange := fixedRange.asTypeOf((Vec(PredictWidth, Bool())))
370a4e57ea3SLi Qianruo
3715995c9e7SJenius  io.out.stage1Out.fixedTaken := VecInit(pds.zipWithIndex.map{case(pd, i) => instrValid (i) && fixedRange(i) && (pd.isRet || pd.isJal || takenIdx === i.U && predTaken && !pd.notCFI)  })
372a4e57ea3SLi Qianruo
373a4e57ea3SLi Qianruo  /** second check: faulse prediction fault and target fault */
374a4e57ea3SLi Qianruo  notCFITaken  := VecInit(pds.zipWithIndex.map{case(pd, i) => fixedRange(i) && instrValid(i) && i.U === takenIdx && pd.notCFI && predTaken })
375a4e57ea3SLi Qianruo  invalidTaken := VecInit(pds.zipWithIndex.map{case(pd, i) => fixedRange(i) && !instrValid(i)  && i.U === takenIdx  && predTaken })
376a4e57ea3SLi Qianruo
377a483ee06SGuokai Chen  val jumpTargets          = VecInit(pds.zipWithIndex.map{case(pd,i) => (pc(i) + jumpOffset(i)).asTypeOf(UInt(VAddrBits.W))})
378a4e57ea3SLi Qianruo  val seqTargets = VecInit((0 until PredictWidth).map(i => pc(i) + Mux(pds(i).isRVC || !instrValid(i), 2.U, 4.U ) ))
379a4e57ea3SLi Qianruo
3805995c9e7SJenius  //Stage 2: detect target fault
3815995c9e7SJenius  /** target calculation: in the next stage  */
3825995c9e7SJenius  val fixedRangeNext = RegNext(fixedRange)
3835995c9e7SJenius  val instrValidNext = RegNext(instrValid)
3845995c9e7SJenius  val takenIdxNext   = RegNext(takenIdx)
3855995c9e7SJenius  val predTakenNext  = RegNext(predTaken)
3865995c9e7SJenius  val predTargetNext = RegNext(predTarget)
3875995c9e7SJenius  val jumpTargetsNext = RegNext(jumpTargets)
3885995c9e7SJenius  val seqTargetsNext = RegNext(seqTargets)
3895995c9e7SJenius  val pdsNext = RegNext(pds)
3905995c9e7SJenius  val jalFaultVecNext = RegNext(jalFaultVec)
3915995c9e7SJenius  val retFaultVecNext = RegNext(retFaultVec)
3925995c9e7SJenius  val notCFITakenNext = RegNext(notCFITaken)
3935995c9e7SJenius  val invalidTakenNext = RegNext(invalidTaken)
394a4e57ea3SLi Qianruo
3955995c9e7SJenius  targetFault      := VecInit(pdsNext.zipWithIndex.map{case(pd,i) => fixedRangeNext(i) && instrValidNext(i) && (pd.isJal || pd.isBr) && takenIdxNext === i.U && predTakenNext  && (predTargetNext =/= jumpTargetsNext(i))})
3965995c9e7SJenius
3975995c9e7SJenius
3985995c9e7SJenius  io.out.stage2Out.faultType.zipWithIndex.map{case(faultType, i) => faultType.value := Mux(jalFaultVecNext(i) , FaultType.jalFault ,
3995995c9e7SJenius                                                                             Mux(retFaultVecNext(i), FaultType.retFault ,
4005995c9e7SJenius                                                                             Mux(targetFault(i), FaultType.targetFault ,
4015995c9e7SJenius                                                                             Mux(notCFITakenNext(i) , FaultType.notCFIFault,
4025995c9e7SJenius                                                                             Mux(invalidTakenNext(i), FaultType.invalidTaken,  FaultType.noFault)))))}
4035995c9e7SJenius
4045995c9e7SJenius  io.out.stage2Out.fixedMissPred.zipWithIndex.map{case(missPred, i ) => missPred := jalFaultVecNext(i) || retFaultVecNext(i) || notCFITakenNext(i) || invalidTakenNext(i) || targetFault(i)}
4055995c9e7SJenius  io.out.stage2Out.fixedTarget.zipWithIndex.map{case(target, i) => target := Mux(jalFaultVecNext(i) || targetFault(i), jumpTargetsNext(i),  seqTargetsNext(i) )}
406d10ddd67SGuokai Chen  io.out.stage2Out.jalTarget.zipWithIndex.map{case(target, i) => target := jumpTargetsNext(i) }
407a4e57ea3SLi Qianruo
408a4e57ea3SLi Qianruo}
409a4e57ea3SLi Qianruo
410*f7af4c74Schengguanghuiclass FrontendTrigger(implicit p: Parameters) extends XSModule with SdtrigExt {
411a4e57ea3SLi Qianruo  val io = IO(new Bundle(){
412a4e57ea3SLi Qianruo    val frontendTrigger = Input(new FrontendTdataDistributeIO)
413a4e57ea3SLi Qianruo    val triggered     = Output(Vec(PredictWidth, new TriggerCf))
414a4e57ea3SLi Qianruo
415a4e57ea3SLi Qianruo    val pds           = Input(Vec(PredictWidth, new PreDecodeInfo))
416a4e57ea3SLi Qianruo    val pc            = Input(Vec(PredictWidth, UInt(VAddrBits.W)))
417a4e57ea3SLi Qianruo    val data          = if(HasCExtension) Input(Vec(PredictWidth + 1, UInt(16.W)))
418a4e57ea3SLi Qianruo                        else Input(Vec(PredictWidth, UInt(32.W)))
419a4e57ea3SLi Qianruo  })
420a4e57ea3SLi Qianruo
421a4e57ea3SLi Qianruo  val data          = io.data
422a4e57ea3SLi Qianruo
423a4e57ea3SLi Qianruo  val rawInsts = if (HasCExtension) VecInit((0 until PredictWidth).map(i => Cat(data(i+1), data(i))))
424a4e57ea3SLi Qianruo  else         VecInit((0 until PredictWidth).map(i => data(i)))
425a4e57ea3SLi Qianruo
426*f7af4c74Schengguanghui  val tdata = RegInit(VecInit(Seq.fill(TriggerNum)(0.U.asTypeOf(new MatchTriggerIO))))
427*f7af4c74Schengguanghui  when(io.frontendTrigger.tUpdate.valid) {
428*f7af4c74Schengguanghui    tdata(io.frontendTrigger.tUpdate.bits.addr) := io.frontendTrigger.tUpdate.bits.tdata
429a4e57ea3SLi Qianruo  }
430*f7af4c74Schengguanghui  val triggerEnableVec = RegInit(VecInit(Seq.fill(TriggerNum)(false.B))) // From CSR, controlled by priv mode, etc.
431*f7af4c74Schengguanghui  triggerEnableVec := io.frontendTrigger.tEnableVec
432*f7af4c74Schengguanghui  XSDebug(triggerEnableVec.asUInt.orR, "Debug Mode: At least one frontend trigger is enabled\n")
433a4e57ea3SLi Qianruo
434*f7af4c74Schengguanghui  val triggerTimingVec = VecInit(tdata.map(_.timing))
435*f7af4c74Schengguanghui  val triggerChainVec = VecInit(tdata.map(_.chain))
436*f7af4c74Schengguanghui
437*f7af4c74Schengguanghui  for (i <- 0 until TriggerNum) { PrintTriggerInfo(triggerEnableVec(i), tdata(i)) }
438a4e57ea3SLi Qianruo
439a4e57ea3SLi Qianruo  for (i <- 0 until PredictWidth) {
440a4e57ea3SLi Qianruo    val currentPC = io.pc(i)
441a4e57ea3SLi Qianruo    val currentIsRVC = io.pds(i).isRVC
442a4e57ea3SLi Qianruo    val inst = WireInit(rawInsts(i))
443*f7af4c74Schengguanghui    val triggerHitVec = Wire(Vec(TriggerNum, Bool()))
444*f7af4c74Schengguanghui    val triggerCanFireVec = Wire(Vec(TriggerNum, Bool()))
445a4e57ea3SLi Qianruo
446*f7af4c74Schengguanghui    for (j <- 0 until TriggerNum) {
447*f7af4c74Schengguanghui      triggerHitVec(j) := Mux(
448*f7af4c74Schengguanghui        tdata(j).select,
449*f7af4c74Schengguanghui        TriggerCmp(Mux(currentIsRVC, inst(15, 0), inst), tdata(j).tdata2, tdata(j).matchType, triggerEnableVec(j)),
450*f7af4c74Schengguanghui        TriggerCmp(currentPC, tdata(j).tdata2, tdata(j).matchType, triggerEnableVec(j))
451*f7af4c74Schengguanghui      )
452a4e57ea3SLi Qianruo    }
453a4e57ea3SLi Qianruo
454*f7af4c74Schengguanghui    TriggerCheckCanFire(TriggerNum, triggerCanFireVec, triggerHitVec, triggerTimingVec, triggerChainVec)
455*f7af4c74Schengguanghui
456*f7af4c74Schengguanghui    // only hit, no matter fire or not
457a4e57ea3SLi Qianruo    io.triggered(i).frontendHit := triggerHitVec
458*f7af4c74Schengguanghui    // can fire, exception will be handled at rob enq
459*f7af4c74Schengguanghui    io.triggered(i).frontendCanFire := triggerCanFireVec
460*f7af4c74Schengguanghui    io.triggered(i).frontendTiming  := triggerTimingVec.zip(triggerEnableVec).map{ case(timing, en) => timing && en}
461*f7af4c74Schengguanghui    io.triggered(i).frontendChain  := triggerChainVec.zip(triggerEnableVec).map{ case(chain, en) => chain && en}
462*f7af4c74Schengguanghui    XSDebug(io.triggered(i).getFrontendCanFire, p"Debug Mode: Predecode Inst No. ${i} has trigger fire vec ${io.triggered(i).frontendCanFire}\n")
463a4e57ea3SLi Qianruo  }
464*f7af4c74Schengguanghui  io.triggered.foreach(_.backendCanFire := VecInit(Seq.fill(TriggerNum)(false.B)))
465*f7af4c74Schengguanghui  io.triggered.foreach(_.backendHit := VecInit(Seq.fill(TriggerNum)(false.B)))
466a4e57ea3SLi Qianruo}
467