xref: /XiangShan/src/main/scala/xiangshan/frontend/icache/IPrefetch.scala (revision 8c57174eee702a858d5291c93f8ae2dae2755096)
17052722fSJay/***************************************************************************************
27052722fSJay  * Copyright (c) 2020-2021 Institute of Computing Technology, Chinese Academy of Sciences
37052722fSJay  * Copyright (c) 2020-2021 Peng Cheng Laboratory
47052722fSJay  *
57052722fSJay  * XiangShan is licensed under Mulan PSL v2.
67052722fSJay  * You can use this software according to the terms and conditions of the Mulan PSL v2.
77052722fSJay  * You may obtain a copy of Mulan PSL v2 at:
87052722fSJay  *          http://license.coscl.org.cn/MulanPSL2
97052722fSJay  *
107052722fSJay  * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
117052722fSJay  * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
127052722fSJay  * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
137052722fSJay  *
147052722fSJay  * See the Mulan PSL v2 for more details.
157052722fSJay  ***************************************************************************************/
167052722fSJay
177052722fSJaypackage xiangshan.frontend.icache
187052722fSJay
198891a219SYinan Xuimport org.chipsalliance.cde.config.Parameters
207052722fSJayimport chisel3._
217052722fSJayimport chisel3.util._
227d45a146SYinan Xuimport difftest._
237052722fSJayimport freechips.rocketchip.tilelink._
247052722fSJayimport utils._
257052722fSJayimport xiangshan.cache.mmu._
267052722fSJayimport xiangshan.frontend._
27d2b20d1aSTang Haojinimport xiangshan.backend.fu.{PMPReqBundle, PMPRespBundle}
28d2b20d1aSTang Haojinimport huancun.PreferCacheKey
29b92c5693STang Haojinimport xiangshan.XSCoreParamsKey
302c9f4a9fSxu_zhimport xiangshan.SoftIfetchPrefetchBundle
31b1ded4e8Sguohongyuimport utility._
327052722fSJay
337052722fSJayabstract class IPrefetchBundle(implicit p: Parameters) extends ICacheBundle
347052722fSJayabstract class IPrefetchModule(implicit p: Parameters) extends ICacheModule
357052722fSJay
362c9f4a9fSxu_zhclass IPrefetchReq(implicit p: Parameters) extends IPrefetchBundle {
372c9f4a9fSxu_zh  val startAddr     : UInt   = UInt(VAddrBits.W)
382c9f4a9fSxu_zh  val nextlineStart : UInt   = UInt(VAddrBits.W)
392c9f4a9fSxu_zh  val ftqIdx        : FtqPtr = new FtqPtr
402c9f4a9fSxu_zh  val isSoftPrefetch: Bool   = Bool()
412c9f4a9fSxu_zh  def crossCacheline: Bool   = startAddr(blockOffBits - 1) === 1.U
422c9f4a9fSxu_zh
432c9f4a9fSxu_zh  def fromFtqICacheInfo(info: FtqICacheInfo): IPrefetchReq = {
442c9f4a9fSxu_zh    this.startAddr := info.startAddr
452c9f4a9fSxu_zh    this.nextlineStart := info.nextlineStart
462c9f4a9fSxu_zh    this.ftqIdx := info.ftqIdx
472c9f4a9fSxu_zh    this.isSoftPrefetch := false.B
482c9f4a9fSxu_zh    this
492c9f4a9fSxu_zh  }
502c9f4a9fSxu_zh
512c9f4a9fSxu_zh  def fromSoftPrefetch(req: SoftIfetchPrefetchBundle): IPrefetchReq = {
522c9f4a9fSxu_zh    this.startAddr := req.vaddr
532c9f4a9fSxu_zh    this.nextlineStart := req.vaddr + (1 << blockOffBits).U
542c9f4a9fSxu_zh    this.ftqIdx := DontCare
552c9f4a9fSxu_zh    this.isSoftPrefetch := true.B
562c9f4a9fSxu_zh    this
572c9f4a9fSxu_zh  }
582c9f4a9fSxu_zh}
592c9f4a9fSxu_zh
6088895b11Sxu_zhclass IPrefetchIO(implicit p: Parameters) extends IPrefetchBundle {
61b92f8445Sssszwic  // control
62b92f8445Sssszwic  val csr_pf_enable     = Input(Bool())
63f80535c3Sxu_zh  val csr_parity_enable = Input(Bool())
64b92f8445Sssszwic  val flush             = Input(Bool())
6558c354d0Sssszwic
662c9f4a9fSxu_zh  val req               = Flipped(Decoupled(new IPrefetchReq))
672c9f4a9fSxu_zh  val flushFromBpu      = Flipped(new BpuFlushInfo)
68b92f8445Sssszwic  val itlb              = Vec(PortNumber, new TlbRequestIO)
69b92f8445Sssszwic  val pmp               = Vec(PortNumber, new ICachePMPBundle)
70b92f8445Sssszwic  val metaRead          = new ICacheMetaReqBundle
71b92f8445Sssszwic  val MSHRReq           = DecoupledIO(new ICacheMissReq)
72b92f8445Sssszwic  val MSHRResp          = Flipped(ValidIO(new ICacheMissResp))
73b92f8445Sssszwic  val wayLookupWrite    = DecoupledIO(new WayLookupInfo)
747052722fSJay}
757052722fSJay
767052722fSJayclass IPrefetchPipe(implicit p: Parameters) extends  IPrefetchModule
777052722fSJay{
7888895b11Sxu_zh  val io: IPrefetchIO = IO(new IPrefetchIO)
797052722fSJay
80b92f8445Sssszwic  val (toITLB,  fromITLB) = (io.itlb.map(_.req), io.itlb.map(_.resp))
81b92f8445Sssszwic  val (toPMP,  fromPMP)   = (io.pmp.map(_.req), io.pmp.map(_.resp))
82b92f8445Sssszwic  val (toMeta,  fromMeta) = (io.metaRead.toIMeta,  io.metaRead.fromIMeta)
83b92f8445Sssszwic  val (toMSHR, fromMSHR)  = (io.MSHRReq, io.MSHRResp)
84b92f8445Sssszwic  val toWayLookup = io.wayLookupWrite
857052722fSJay
86b92f8445Sssszwic  val s0_fire, s1_fire, s2_fire             = WireInit(false.B)
87b92f8445Sssszwic  val s0_discard, s2_discard                = WireInit(false.B)
88b92f8445Sssszwic  val s0_ready, s1_ready, s2_ready          = WireInit(false.B)
89b92f8445Sssszwic  val s0_flush, s1_flush, s2_flush          = WireInit(false.B)
90b92f8445Sssszwic  val from_bpu_s0_flush, from_bpu_s1_flush  = WireInit(false.B)
917052722fSJay
92cb6e5d3cSssszwic  /**
93cb6e5d3cSssszwic    ******************************************************************************
94cb6e5d3cSssszwic    * IPrefetch Stage 0
95b92f8445Sssszwic    * - 1. receive ftq req
96b92f8445Sssszwic    * - 2. send req to ITLB
97b92f8445Sssszwic    * - 3. send req to Meta SRAM
98cb6e5d3cSssszwic    ******************************************************************************
99cb6e5d3cSssszwic    */
1002c9f4a9fSxu_zh  val s0_valid  = io.req.valid
101cb6e5d3cSssszwic
102b92f8445Sssszwic  /**
103b92f8445Sssszwic    ******************************************************************************
104b92f8445Sssszwic    * receive ftq req
105b92f8445Sssszwic    ******************************************************************************
106b92f8445Sssszwic    */
1072c9f4a9fSxu_zh  val s0_req_vaddr    = VecInit(Seq(io.req.bits.startAddr, io.req.bits.nextlineStart))
1082c9f4a9fSxu_zh  val s0_req_ftqIdx   = io.req.bits.ftqIdx
1092c9f4a9fSxu_zh  val s0_isSoftPrefetch = io.req.bits.isSoftPrefetch
1102c9f4a9fSxu_zh  val s0_doubleline   = io.req.bits.crossCacheline
11188895b11Sxu_zh  val s0_req_vSetIdx  = s0_req_vaddr.map(get_idx)
1127052722fSJay
1132c9f4a9fSxu_zh  from_bpu_s0_flush := !s0_isSoftPrefetch && (io.flushFromBpu.shouldFlushByStage2(s0_req_ftqIdx) ||
1142c9f4a9fSxu_zh                                              io.flushFromBpu.shouldFlushByStage3(s0_req_ftqIdx))
115b92f8445Sssszwic  s0_flush := io.flush || from_bpu_s0_flush || s1_flush
1167052722fSJay
117b92f8445Sssszwic  val s0_can_go = s1_ready && toITLB(0).ready && toITLB(1).ready && toMeta.ready
1182c9f4a9fSxu_zh  io.req.ready := s0_can_go
1197052722fSJay
120b92f8445Sssszwic  s0_fire := s0_valid && s0_can_go && !s0_flush
121cb6e5d3cSssszwic
122cb6e5d3cSssszwic  /**
123cb6e5d3cSssszwic    ******************************************************************************
124cb6e5d3cSssszwic    * IPrefetch Stage 1
125b92f8445Sssszwic    * - 1. Receive resp from ITLB
126b92f8445Sssszwic    * - 2. Receive resp from IMeta and check
127b92f8445Sssszwic    * - 3. Monitor the requests from missUnit to write to SRAM.
128b92f8445Sssszwic    * - 4. Wirte wayLookup
129cb6e5d3cSssszwic    ******************************************************************************
130cb6e5d3cSssszwic    */
131b92f8445Sssszwic  val s1_valid = generatePipeControl(lastFire = s0_fire, thisFire = s1_fire, thisFlush = s1_flush, lastFlush = false.B)
132cb6e5d3cSssszwic
133b92f8445Sssszwic  val s1_req_vaddr    = RegEnable(s0_req_vaddr, 0.U.asTypeOf(s0_req_vaddr), s0_fire)
1342c9f4a9fSxu_zh  val s1_isSoftPrefetch = RegEnable(s0_isSoftPrefetch, 0.U.asTypeOf(s0_isSoftPrefetch), s0_fire)
135b92f8445Sssszwic  val s1_doubleline   = RegEnable(s0_doubleline, 0.U.asTypeOf(s0_doubleline), s0_fire)
136b92f8445Sssszwic  val s1_req_ftqIdx   = RegEnable(s0_req_ftqIdx, 0.U.asTypeOf(s0_req_ftqIdx), s0_fire)
13788895b11Sxu_zh  val s1_req_vSetIdx  = VecInit(s1_req_vaddr.map(get_idx))
1387052722fSJay
139b92f8445Sssszwic  val m_idle :: m_itlbResend :: m_metaResend :: m_enqWay :: m_enterS2 :: Nil = Enum(5)
140b92f8445Sssszwic  val state = RegInit(m_idle)
141b92f8445Sssszwic  val next_state = WireDefault(state)
142b92f8445Sssszwic  val s0_fire_r = RegNext(s0_fire)
143b92f8445Sssszwic  dontTouch(state)
144b92f8445Sssszwic  dontTouch(next_state)
145b92f8445Sssszwic  state := next_state
1467052722fSJay
147b92f8445Sssszwic  /**
148b92f8445Sssszwic    ******************************************************************************
149b92f8445Sssszwic    * resend itlb req if miss
150b92f8445Sssszwic    ******************************************************************************
151b92f8445Sssszwic    */
152b92f8445Sssszwic  val s1_wait_itlb  = RegInit(VecInit(Seq.fill(PortNumber)(false.B)))
153b92f8445Sssszwic  (0 until PortNumber).foreach { i =>
154b92f8445Sssszwic    when(s1_flush) {
155b92f8445Sssszwic      s1_wait_itlb(i) := false.B
156b92f8445Sssszwic    }.elsewhen(RegNext(s0_fire) && fromITLB(i).bits.miss) {
157b92f8445Sssszwic      s1_wait_itlb(i) := true.B
158b92f8445Sssszwic    }.elsewhen(s1_wait_itlb(i) && !fromITLB(i).bits.miss) {
159b92f8445Sssszwic      s1_wait_itlb(i) := false.B
160b92f8445Sssszwic    }
161b92f8445Sssszwic  }
162b92f8445Sssszwic  val s1_need_itlb    = VecInit(Seq((RegNext(s0_fire) || s1_wait_itlb(0)) && fromITLB(0).bits.miss,
163b92f8445Sssszwic                                    (RegNext(s0_fire) || s1_wait_itlb(1)) && fromITLB(1).bits.miss && s1_doubleline))
164b92f8445Sssszwic  val tlb_valid_pulse = VecInit(Seq((RegNext(s0_fire) || s1_wait_itlb(0)) && !fromITLB(0).bits.miss,
165b92f8445Sssszwic                                    (RegNext(s0_fire) || s1_wait_itlb(1)) && !fromITLB(1).bits.miss && s1_doubleline))
166b92f8445Sssszwic  val tlb_valid_latch = VecInit((0 until PortNumber).map(i => ValidHoldBypass(tlb_valid_pulse(i), s1_fire, flush=s1_flush)))
167b92f8445Sssszwic  val itlb_finish     = tlb_valid_latch(0) && (!s1_doubleline || tlb_valid_latch(1))
1687052722fSJay
169b92f8445Sssszwic  for (i <- 0 until PortNumber) {
170b92f8445Sssszwic    toITLB(i).valid             := s1_need_itlb(i) || (s0_valid && (if(i == 0) true.B else s0_doubleline))
171b92f8445Sssszwic    toITLB(i).bits              := DontCare
172b92f8445Sssszwic    toITLB(i).bits.size         := 3.U
173b92f8445Sssszwic    toITLB(i).bits.vaddr        := Mux(s1_need_itlb(i), s1_req_vaddr(i), s0_req_vaddr(i))
174b92f8445Sssszwic    toITLB(i).bits.debug.pc     := Mux(s1_need_itlb(i), s1_req_vaddr(i), s0_req_vaddr(i))
175b92f8445Sssszwic    toITLB(i).bits.cmd          := TlbCmd.exec
176b92f8445Sssszwic    toITLB(i).bits.no_translate := false.B
177b92f8445Sssszwic  }
178b92f8445Sssszwic  fromITLB.foreach(_.ready := true.B)
179b92f8445Sssszwic  io.itlb.foreach(_.req_kill := false.B)
1807052722fSJay
181b92f8445Sssszwic  /**
182b92f8445Sssszwic    ******************************************************************************
183b92f8445Sssszwic    * Receive resp from ITLB
184b92f8445Sssszwic    ******************************************************************************
185b92f8445Sssszwic    */
186b92f8445Sssszwic  val s1_req_paddr_wire     = VecInit(fromITLB.map(_.bits.paddr(0)))
187b92f8445Sssszwic  val s1_req_paddr_reg      = VecInit((0 until PortNumber).map( i =>
18888895b11Sxu_zh    RegEnable(s1_req_paddr_wire(i), 0.U(PAddrBits.W), tlb_valid_pulse(i))
18988895b11Sxu_zh  ))
190b92f8445Sssszwic  val s1_req_paddr          = VecInit((0 until PortNumber).map( i =>
19188895b11Sxu_zh    Mux(tlb_valid_pulse(i), s1_req_paddr_wire(i), s1_req_paddr_reg(i))
19288895b11Sxu_zh  ))
19391946104Sxu_zh  val s1_req_gpaddr_tmp     = VecInit((0 until PortNumber).map( i =>
19488895b11Sxu_zh    ResultHoldBypass(valid = tlb_valid_pulse(i), init = 0.U.asTypeOf(fromITLB(i).bits.gpaddr(0)), data = fromITLB(i).bits.gpaddr(0))
19588895b11Sxu_zh  ))
19688895b11Sxu_zh  val s1_itlb_exception     = VecInit((0 until PortNumber).map( i =>
19788895b11Sxu_zh    ResultHoldBypass(valid = tlb_valid_pulse(i), init = 0.U(ExceptionType.width.W), data = ExceptionType.fromTlbResp(fromITLB(i).bits))
19888895b11Sxu_zh  ))
199002c10a4SYanqin Li  val s1_itlb_pbmt          = VecInit((0 until PortNumber).map( i =>
200002c10a4SYanqin Li    ResultHoldBypass(valid = tlb_valid_pulse(i), init = 0.U.asTypeOf(fromITLB(i).bits.pbmt(0)), data = fromITLB(i).bits.pbmt(0))
201002c10a4SYanqin Li  ))
20288895b11Sxu_zh  val s1_itlb_exception_gpf = VecInit(s1_itlb_exception.map(_ === ExceptionType.gpf))
203b92f8445Sssszwic
20491946104Sxu_zh  /* Select gpaddr with the first gpf
20591946104Sxu_zh   * Note: the backend wants the base guest physical address of a fetch block
20691946104Sxu_zh   *       for port(i), its base gpaddr is actually (gpaddr - i * blocksize)
20791946104Sxu_zh   *       see GPAMem: https://github.com/OpenXiangShan/XiangShan/blob/344cf5d55568dd40cd658a9ee66047a505eeb504/src/main/scala/xiangshan/backend/GPAMem.scala#L33-L34
20891946104Sxu_zh   *       see also: https://github.com/OpenXiangShan/XiangShan/blob/344cf5d55568dd40cd658a9ee66047a505eeb504/src/main/scala/xiangshan/frontend/IFU.scala#L374-L375
20991946104Sxu_zh   */
21091946104Sxu_zh  val s1_req_gpaddr = PriorityMuxDefault(
21188895b11Sxu_zh    s1_itlb_exception_gpf zip (0 until PortNumber).map(i => s1_req_gpaddr_tmp(i) - (i << blockOffBits).U),
21291946104Sxu_zh    0.U.asTypeOf(s1_req_gpaddr_tmp(0))
21391946104Sxu_zh  )
21491946104Sxu_zh
215b92f8445Sssszwic  /**
216b92f8445Sssszwic    ******************************************************************************
217b92f8445Sssszwic    * resend metaArray read req when itlb miss finish
218b92f8445Sssszwic    ******************************************************************************
219b92f8445Sssszwic    */
220b92f8445Sssszwic  val s1_need_meta = ((state === m_itlbResend) && itlb_finish) || (state === m_metaResend)
221b92f8445Sssszwic  toMeta.valid              := s1_need_meta || s0_valid
222b92f8445Sssszwic  toMeta.bits               := DontCare
223b92f8445Sssszwic  toMeta.bits.isDoubleLine  := Mux(s1_need_meta, s1_doubleline, s0_doubleline)
224b92f8445Sssszwic
225b92f8445Sssszwic  for (i <- 0 until PortNumber) {
226b92f8445Sssszwic    toMeta.bits.vSetIdx(i)  := Mux(s1_need_meta, s1_req_vSetIdx(i), s0_req_vSetIdx(i))
227cb6e5d3cSssszwic  }
228cb6e5d3cSssszwic
229cb6e5d3cSssszwic  /**
230cb6e5d3cSssszwic    ******************************************************************************
231b92f8445Sssszwic    * Receive resp from IMeta and check
232cb6e5d3cSssszwic    ******************************************************************************
233cb6e5d3cSssszwic    */
23488895b11Sxu_zh  val s1_req_ptags    = VecInit(s1_req_paddr.map(get_phy_tag))
235cb6e5d3cSssszwic
236b92f8445Sssszwic  val s1_meta_ptags   = fromMeta.tags
237b92f8445Sssszwic  val s1_meta_valids  = fromMeta.entryValid
238b92f8445Sssszwic  // If error is found in either way, the tag_eq_vec is unreliable, so we do not use waymask, but directly .orR
23988895b11Sxu_zh  val s1_meta_corrupt = VecInit(fromMeta.errors.map(_.asUInt.orR))
2409bba777eSssszwic
241b92f8445Sssszwic  def get_waymask(paddrs: Vec[UInt]): Vec[UInt] = {
24288895b11Sxu_zh    val ptags         = paddrs.map(get_phy_tag)
243b92f8445Sssszwic    val tag_eq_vec    = VecInit((0 until PortNumber).map( p => VecInit((0 until nWays).map( w => s1_meta_ptags(p)(w) === ptags(p)))))
244b92f8445Sssszwic    val tag_match_vec = VecInit((0 until PortNumber).map( k => VecInit(tag_eq_vec(k).zipWithIndex.map{ case(way_tag_eq, w) => way_tag_eq && s1_meta_valids(k)(w)})))
245b92f8445Sssszwic    val waymasks      = VecInit(tag_match_vec.map(_.asUInt))
246b92f8445Sssszwic    waymasks
247cb6e5d3cSssszwic  }
2489bba777eSssszwic
249b92f8445Sssszwic  val s1_SRAM_waymasks = VecInit((0 until PortNumber).map(i =>
250b92f8445Sssszwic                            Mux(tlb_valid_pulse(i), get_waymask(s1_req_paddr_wire)(i), get_waymask(s1_req_paddr_reg)(i))))
251b92f8445Sssszwic
252b92f8445Sssszwic  /**
253b92f8445Sssszwic    ******************************************************************************
254b4f1e5b2Sxu_zh    * update waymask according to MSHR update data
255b92f8445Sssszwic    ******************************************************************************
256b92f8445Sssszwic    */
257b92f8445Sssszwic  def update_waymask(mask: UInt, vSetIdx: UInt, ptag: UInt): UInt = {
258b92f8445Sssszwic    require(mask.getWidth == nWays)
259b92f8445Sssszwic    val new_mask  = WireInit(mask)
260b4f1e5b2Sxu_zh    val valid = fromMSHR.valid && !fromMSHR.bits.corrupt
261b4f1e5b2Sxu_zh    val vset_same = fromMSHR.bits.vSetIdx === vSetIdx
262b92f8445Sssszwic    val ptag_same = getPhyTagFromBlk(fromMSHR.bits.blkPaddr) === ptag
263b92f8445Sssszwic    val way_same  = fromMSHR.bits.waymask === mask
264b4f1e5b2Sxu_zh    when(valid && vset_same) {
265b92f8445Sssszwic      when(ptag_same) {
266b92f8445Sssszwic        new_mask := fromMSHR.bits.waymask
267b92f8445Sssszwic      }.elsewhen(way_same) {
268b92f8445Sssszwic        new_mask := 0.U
269cb6e5d3cSssszwic      }
270b92f8445Sssszwic    }
271b92f8445Sssszwic    new_mask
272b92f8445Sssszwic  }
273b92f8445Sssszwic
274b92f8445Sssszwic  val s1_SRAM_valid = s0_fire_r || RegNext(s1_need_meta && toMeta.ready)
275b4f1e5b2Sxu_zh  val s1_MSHR_valid = fromMSHR.valid && !fromMSHR.bits.corrupt
276b4f1e5b2Sxu_zh  val s1_waymasks   = WireInit(VecInit(Seq.fill(PortNumber)(0.U(nWays.W))))
277b4f1e5b2Sxu_zh  val s1_waymasks_r = RegEnable(s1_waymasks, 0.U.asTypeOf(s1_waymasks), s1_SRAM_valid || s1_MSHR_valid)
278b92f8445Sssszwic  (0 until PortNumber).foreach{i =>
279b4f1e5b2Sxu_zh    val old_waymask = Mux(s1_SRAM_valid, s1_SRAM_waymasks(i), s1_waymasks_r(i))
280b4f1e5b2Sxu_zh    s1_waymasks(i) := update_waymask(old_waymask, s1_req_vSetIdx(i), s1_req_ptags(i))
281b92f8445Sssszwic  }
282b92f8445Sssszwic
283b92f8445Sssszwic  /**
284b92f8445Sssszwic    ******************************************************************************
285b92f8445Sssszwic    * send enqueu req to WayLookup
286b92f8445Sssszwic    ******** **********************************************************************
287b92f8445Sssszwic    */
288b92f8445Sssszwic  // Disallow enqueuing wayLookup when SRAM write occurs.
2892c9f4a9fSxu_zh  toWayLookup.valid             := ((state === m_enqWay) || ((state === m_idle) && itlb_finish)) &&
2902c9f4a9fSxu_zh    !s1_flush && !fromMSHR.valid && !s1_isSoftPrefetch  // do not enqueue soft prefetch
291b92f8445Sssszwic  toWayLookup.bits.vSetIdx      := s1_req_vSetIdx
292b92f8445Sssszwic  toWayLookup.bits.waymask      := s1_waymasks
293b92f8445Sssszwic  toWayLookup.bits.ptag         := s1_req_ptags
294b92f8445Sssszwic  toWayLookup.bits.gpaddr       := s1_req_gpaddr
2951a5af821Sxu_zh  (0 until PortNumber).foreach { i =>
2961a5af821Sxu_zh    val excpValid = (if (i == 0) true.B else s1_doubleline)  // exception in first line is always valid, in second line is valid iff is doubleline request
29788895b11Sxu_zh    // Send s1_itlb_exception to WayLookup (instead of s1_exception_out) for better timing. Will check pmp again in mainPipe
29888895b11Sxu_zh    toWayLookup.bits.itlb_exception(i) := Mux(excpValid, s1_itlb_exception(i), ExceptionType.none)
299002c10a4SYanqin Li    toWayLookup.bits.itlb_pbmt(i)      := Mux(excpValid, s1_itlb_pbmt(i), Pbmt.pma)
30088895b11Sxu_zh    toWayLookup.bits.meta_corrupt(i)   := excpValid && s1_meta_corrupt(i)
3011a5af821Sxu_zh  }
302b92f8445Sssszwic
303b92f8445Sssszwic  val s1_waymasks_vec = s1_waymasks.map(_.asTypeOf(Vec(nWays, Bool())))
304b92f8445Sssszwic  when(toWayLookup.fire) {
305b92f8445Sssszwic    assert(PopCount(s1_waymasks_vec(0)) <= 1.U && (PopCount(s1_waymasks_vec(1)) <= 1.U || !s1_doubleline),
306b92f8445Sssszwic      "Multiple hit in main pipe, port0:is=%d,ptag=0x%x,vidx=0x%x,vaddr=0x%x port1:is=%d,ptag=0x%x,vidx=0x%x,vaddr=0x%x ",
307b92f8445Sssszwic      PopCount(s1_waymasks_vec(0)) > 1.U, s1_req_ptags(0), get_idx(s1_req_vaddr(0)), s1_req_vaddr(0),
308b92f8445Sssszwic      PopCount(s1_waymasks_vec(1)) > 1.U && s1_doubleline, s1_req_ptags(1), get_idx(s1_req_vaddr(1)), s1_req_vaddr(1))
309b92f8445Sssszwic  }
310b92f8445Sssszwic
311b92f8445Sssszwic  /**
312b92f8445Sssszwic    ******************************************************************************
313b92f8445Sssszwic    * PMP check
314b92f8445Sssszwic    ******************************************************************************
315b92f8445Sssszwic    */
31688895b11Sxu_zh  toPMP.zipWithIndex.foreach { case (p, i) =>
31788895b11Sxu_zh    // if itlb has exception, paddr can be invalid, therefore pmp check can be skipped
31888895b11Sxu_zh    p.valid     := s1_valid // && s1_itlb_exception === ExceptionType.none
319b92f8445Sssszwic    p.bits.addr := s1_req_paddr(i)
320b92f8445Sssszwic    p.bits.size := 3.U // TODO
321b92f8445Sssszwic    p.bits.cmd  := TlbCmd.exec
322b92f8445Sssszwic  }
32388895b11Sxu_zh  val s1_pmp_exception = VecInit(fromPMP.map(ExceptionType.fromPMPResp))
324002c10a4SYanqin Li  val s1_pmp_mmio      = VecInit(fromPMP.map(_.mmio))
32588895b11Sxu_zh
326f80535c3Sxu_zh  // also raise af when meta array corrupt is detected, to cancel prefetch
327f80535c3Sxu_zh  val s1_meta_exception = VecInit(s1_meta_corrupt.map(ExceptionType.fromECC(io.csr_parity_enable, _)))
328f80535c3Sxu_zh
329f80535c3Sxu_zh  // merge s1 itlb/pmp/meta exceptions, itlb has the highest priority, pmp next, meta lowest
330f80535c3Sxu_zh  val s1_exception_out = ExceptionType.merge(
331f80535c3Sxu_zh    s1_itlb_exception,
332f80535c3Sxu_zh    s1_pmp_exception,
333f80535c3Sxu_zh    s1_meta_exception
334f80535c3Sxu_zh  )
335b92f8445Sssszwic
336002c10a4SYanqin Li  // merge pmp mmio and itlb pbmt
337002c10a4SYanqin Li  val s1_mmio = VecInit((s1_pmp_mmio zip s1_itlb_pbmt).map{ case (mmio, pbmt) =>
338002c10a4SYanqin Li    mmio || Pbmt.isUncache(pbmt)
339002c10a4SYanqin Li  })
340002c10a4SYanqin Li
341b92f8445Sssszwic  /**
342b92f8445Sssszwic    ******************************************************************************
343b92f8445Sssszwic    * state machine
344b92f8445Sssszwic    ******** **********************************************************************
345b92f8445Sssszwic    */
346b92f8445Sssszwic
347b92f8445Sssszwic  switch(state) {
348b92f8445Sssszwic    is(m_idle) {
3492c9f4a9fSxu_zh      when(s1_valid) {
3502c9f4a9fSxu_zh        when(!itlb_finish) {
351b92f8445Sssszwic          next_state := m_itlbResend
352*8c57174eSxu_zh        }.elsewhen(!toWayLookup.fire) {  // itlb_finish
353b92f8445Sssszwic          next_state := m_enqWay
354*8c57174eSxu_zh        }.elsewhen(!s2_ready) {  // itlb_finish && toWayLookup.fire
355b92f8445Sssszwic          next_state := m_enterS2
3562c9f4a9fSxu_zh        } // .otherwise { next_state := m_idle }
3572c9f4a9fSxu_zh      } // .otherwise { next_state := m_idle }  // !s1_valid
358b92f8445Sssszwic    }
359b92f8445Sssszwic    is(m_itlbResend) {
3602c9f4a9fSxu_zh      when(itlb_finish) {
3612c9f4a9fSxu_zh        when(!toMeta.ready) {
362b92f8445Sssszwic          next_state := m_metaResend
363*8c57174eSxu_zh        }.otherwise { // toMeta.ready
364b92f8445Sssszwic          next_state := m_enqWay
365b92f8445Sssszwic        }
3662c9f4a9fSxu_zh      } // .otherwise { next_state := m_itlbResend }  // !itlb_finish
367b92f8445Sssszwic    }
368b92f8445Sssszwic    is(m_metaResend) {
369b92f8445Sssszwic      when(toMeta.ready) {
370b92f8445Sssszwic        next_state := m_enqWay
3712c9f4a9fSxu_zh      } // .otherwise { next_state := m_metaResend }  // !toMeta.ready
372b92f8445Sssszwic    }
373b92f8445Sssszwic    is(m_enqWay) {
374*8c57174eSxu_zh      when(toWayLookup.fire || s1_isSoftPrefetch) {
375*8c57174eSxu_zh        when (!s2_ready) {
376b92f8445Sssszwic          next_state := m_enterS2
377*8c57174eSxu_zh        }.otherwise {  // s2_ready
378b92f8445Sssszwic          next_state := m_idle
379b92f8445Sssszwic        }
380*8c57174eSxu_zh      } // .otherwise { next_state := m_enqWay }
381b92f8445Sssszwic    }
382b92f8445Sssszwic    is(m_enterS2) {
383b92f8445Sssszwic      when(s2_ready) {
384b92f8445Sssszwic        next_state := m_idle
385b92f8445Sssszwic      }
386b92f8445Sssszwic    }
387b92f8445Sssszwic  }
388b92f8445Sssszwic
389b92f8445Sssszwic  when(s1_flush) {
390b92f8445Sssszwic    next_state := m_idle
391b92f8445Sssszwic  }
392b92f8445Sssszwic
393b92f8445Sssszwic  /** Stage 1 control */
3942c9f4a9fSxu_zh  from_bpu_s1_flush := s1_valid && !s1_isSoftPrefetch && io.flushFromBpu.shouldFlushByStage3(s1_req_ftqIdx)
395b92f8445Sssszwic  s1_flush := io.flush || from_bpu_s1_flush
396b92f8445Sssszwic
397b92f8445Sssszwic  s1_ready      := next_state === m_idle
398400391a3Sxu_zh  s1_fire       := (next_state === m_idle) && s1_valid && !s1_flush  // used to clear s1_valid & itlb_valid_latch
399400391a3Sxu_zh  val s1_real_fire = s1_fire && io.csr_pf_enable                     // real "s1 fire" that s1 enters s2
400b92f8445Sssszwic
401b92f8445Sssszwic  /**
402b92f8445Sssszwic    ******************************************************************************
403b92f8445Sssszwic    * IPrefetch Stage 2
404b92f8445Sssszwic    * - 1. Monitor the requests from missUnit to write to SRAM.
405b92f8445Sssszwic    * - 2. send req to missUnit
406b92f8445Sssszwic    ******************************************************************************
407b92f8445Sssszwic    */
408400391a3Sxu_zh  val s2_valid  = generatePipeControl(lastFire = s1_real_fire, thisFire = s2_fire, thisFlush = s2_flush, lastFlush = false.B)
409b92f8445Sssszwic
410400391a3Sxu_zh  val s2_req_vaddr    = RegEnable(s1_req_vaddr,     0.U.asTypeOf(s1_req_vaddr),     s1_real_fire)
4112c9f4a9fSxu_zh  val s2_isSoftPrefetch = RegEnable(s1_isSoftPrefetch, 0.U.asTypeOf(s1_isSoftPrefetch), s1_real_fire)
412400391a3Sxu_zh  val s2_doubleline   = RegEnable(s1_doubleline,    0.U.asTypeOf(s1_doubleline),    s1_real_fire)
413400391a3Sxu_zh  val s2_req_paddr    = RegEnable(s1_req_paddr,     0.U.asTypeOf(s1_req_paddr),     s1_real_fire)
414400391a3Sxu_zh  val s2_exception    = RegEnable(s1_exception_out, 0.U.asTypeOf(s1_exception_out), s1_real_fire)  // includes itlb/pmp/meta exception
415400391a3Sxu_zh  val s2_mmio         = RegEnable(s1_mmio,          0.U.asTypeOf(s1_mmio),          s1_real_fire)
416400391a3Sxu_zh  val s2_waymasks     = RegEnable(s1_waymasks,      0.U.asTypeOf(s1_waymasks),      s1_real_fire)
417b92f8445Sssszwic
41888895b11Sxu_zh  val s2_req_vSetIdx  = s2_req_vaddr.map(get_idx)
41988895b11Sxu_zh  val s2_req_ptags    = s2_req_paddr.map(get_phy_tag)
420b92f8445Sssszwic
421b92f8445Sssszwic  /**
422b92f8445Sssszwic    ******************************************************************************
423b92f8445Sssszwic    * Monitor the requests from missUnit to write to SRAM
424b92f8445Sssszwic    ******************************************************************************
425b92f8445Sssszwic    */
426b808ac73Sxu_zh
427b808ac73Sxu_zh  /* NOTE: If fromMSHR.bits.corrupt, we should set s2_MSHR_hits to false.B, and send prefetch requests again.
428b808ac73Sxu_zh   * This is the opposite of how mainPipe handles fromMSHR.bits.corrupt,
429b808ac73Sxu_zh   *   in which we should set s2_MSHR_hits to true.B, and send error to ifu.
430b808ac73Sxu_zh   */
431b808ac73Sxu_zh  val s2_MSHR_match = VecInit((0 until PortNumber).map(i =>
432b808ac73Sxu_zh    (s2_req_vSetIdx(i) === fromMSHR.bits.vSetIdx) &&
433b92f8445Sssszwic    (s2_req_ptags(i) === getPhyTagFromBlk(fromMSHR.bits.blkPaddr)) &&
434b808ac73Sxu_zh    s2_valid && fromMSHR.valid && !fromMSHR.bits.corrupt
435b808ac73Sxu_zh  ))
436b92f8445Sssszwic  val s2_MSHR_hits = (0 until PortNumber).map(i => ValidHoldBypass(s2_MSHR_match(i), s2_fire || s2_flush))
437b92f8445Sssszwic
438b808ac73Sxu_zh  val s2_SRAM_hits = s2_waymasks.map(_.orR)
439b808ac73Sxu_zh  val s2_hits = VecInit((0 until PortNumber).map(i => s2_MSHR_hits(i) || s2_SRAM_hits(i)))
440b808ac73Sxu_zh
441f80535c3Sxu_zh  /* s2_exception includes itlb pf/gpf/af, pmp af and meta corruption (af), neither of which should be prefetched
44288895b11Sxu_zh   * mmio should not be prefetched
443f80535c3Sxu_zh   * also, if previous has exception, latter port should also not be prefetched
44488895b11Sxu_zh   */
445b808ac73Sxu_zh  val s2_miss = VecInit((0 until PortNumber).map { i =>
446b808ac73Sxu_zh    !s2_hits(i) && (if (i==0) true.B else s2_doubleline) &&
44788895b11Sxu_zh      s2_exception.take(i+1).map(_ === ExceptionType.none).reduce(_&&_) &&
44888895b11Sxu_zh      s2_mmio.take(i+1).map(!_).reduce(_&&_)
449b808ac73Sxu_zh  })
450b92f8445Sssszwic
451b92f8445Sssszwic  /**
452b92f8445Sssszwic    ******************************************************************************
453b92f8445Sssszwic    * send req to missUnit
454b92f8445Sssszwic    ******************************************************************************
455b92f8445Sssszwic    */
456b92f8445Sssszwic  val toMSHRArbiter = Module(new Arbiter(new ICacheMissReq, PortNumber))
457b92f8445Sssszwic
458b92f8445Sssszwic  // To avoid sending duplicate requests.
459b808ac73Sxu_zh  val has_send = RegInit(VecInit(Seq.fill(PortNumber)(false.B)))
460b92f8445Sssszwic  (0 until PortNumber).foreach{ i =>
461400391a3Sxu_zh    when(s1_real_fire) {
462b92f8445Sssszwic      has_send(i) := false.B
463b92f8445Sssszwic    }.elsewhen(toMSHRArbiter.io.in(i).fire) {
464b92f8445Sssszwic      has_send(i) := true.B
465b92f8445Sssszwic    }
466b92f8445Sssszwic  }
467b92f8445Sssszwic
468b92f8445Sssszwic  (0 until PortNumber).map{ i =>
469b92f8445Sssszwic    toMSHRArbiter.io.in(i).valid          := s2_valid && s2_miss(i) && !has_send(i)
470b92f8445Sssszwic    toMSHRArbiter.io.in(i).bits.blkPaddr  := getBlkAddr(s2_req_paddr(i))
471b92f8445Sssszwic    toMSHRArbiter.io.in(i).bits.vSetIdx   := s2_req_vSetIdx(i)
472b92f8445Sssszwic  }
473b92f8445Sssszwic
474b92f8445Sssszwic  toMSHR <> toMSHRArbiter.io.out
475b92f8445Sssszwic
476b92f8445Sssszwic  s2_flush := io.flush
477b92f8445Sssszwic
478b92f8445Sssszwic  val s2_finish  = (0 until PortNumber).map(i => has_send(i) || !s2_miss(i) || toMSHRArbiter.io.in(i).fire).reduce(_&&_)
479b92f8445Sssszwic  s2_ready      := s2_finish || !s2_valid
480b92f8445Sssszwic  s2_fire       := s2_valid && s2_finish && !s2_flush
4819bba777eSssszwic
482cb6e5d3cSssszwic  /** PerfAccumulate */
4832c9f4a9fSxu_zh  // the number of bpu flush
4842c9f4a9fSxu_zh  XSPerfAccumulate("bpu_s0_flush", from_bpu_s0_flush)
4852c9f4a9fSxu_zh  XSPerfAccumulate("bpu_s1_flush", from_bpu_s1_flush)
4862c9f4a9fSxu_zh  // the number of prefetch request received from ftq or backend (software prefetch)
4872c9f4a9fSxu_zh//  XSPerfAccumulate("prefetch_req_receive", io.req.fire)
4882c9f4a9fSxu_zh  XSPerfAccumulate("prefetch_req_receive_hw", io.req.fire && !io.req.bits.isSoftPrefetch)
4892c9f4a9fSxu_zh  XSPerfAccumulate("prefetch_req_receive_sw", io.req.fire && io.req.bits.isSoftPrefetch)
490b92f8445Sssszwic  // the number of prefetch request sent to missUnit
4912c9f4a9fSxu_zh//  XSPerfAccumulate("prefetch_req_send", toMSHR.fire)
4922c9f4a9fSxu_zh  XSPerfAccumulate("prefetch_req_send_hw", toMSHR.fire && !s2_isSoftPrefetch)
4932c9f4a9fSxu_zh  XSPerfAccumulate("prefetch_req_send_sw", toMSHR.fire && s2_isSoftPrefetch)
494b92f8445Sssszwic  XSPerfAccumulate("to_missUnit_stall", toMSHR.valid && !toMSHR.ready)
495cb6e5d3cSssszwic  /**
496cb6e5d3cSssszwic    * Count the number of requests that are filtered for various reasons.
497cb6e5d3cSssszwic    * The number of prefetch discard in Performance Accumulator may be
498cb6e5d3cSssszwic    * a littel larger the number of really discarded. Because there can
499cb6e5d3cSssszwic    * be multiple reasons for a canceled request at the same time.
500cb6e5d3cSssszwic    */
501b92f8445Sssszwic  // discard prefetch request by flush
502b92f8445Sssszwic  // XSPerfAccumulate("fdip_prefetch_discard_by_tlb_except",  p1_discard && p1_tlb_except)
503b92f8445Sssszwic  // // discard prefetch request by hit icache SRAM
504b92f8445Sssszwic  // XSPerfAccumulate("fdip_prefetch_discard_by_hit_cache",   p2_discard && p1_meta_hit)
505b92f8445Sssszwic  // // discard prefetch request by hit wirte SRAM
506b92f8445Sssszwic  // XSPerfAccumulate("fdip_prefetch_discard_by_p1_monoitor", p1_discard && p1_monitor_hit)
507b92f8445Sssszwic  // // discard prefetch request by pmp except or mmio
508b92f8445Sssszwic  // XSPerfAccumulate("fdip_prefetch_discard_by_pmp",         p2_discard && p2_pmp_except)
509b92f8445Sssszwic  // // discard prefetch request by hit mainPipe info
510b92f8445Sssszwic  // // XSPerfAccumulate("fdip_prefetch_discard_by_mainPipe",    p2_discard && p2_mainPipe_hit)
5117052722fSJay}