1package xiangshan.backend.issue 2 3import org.chipsalliance.cde.config.Parameters 4import chisel3._ 5import chisel3.util._ 6import freechips.rocketchip.diplomacy.{LazyModule, LazyModuleImp} 7import utils.OptionWrapper 8import xiangshan._ 9import xiangshan.backend.Bundles._ 10import xiangshan.backend.datapath.DataConfig.{IntData, VAddrData, VecData} 11import xiangshan.backend.datapath.WbConfig.{IntWB, VfWB} 12import xiangshan.backend.fu.FuType 13import xiangshan.backend.regfile.RfWritePortWithConfig 14import xiangshan.backend.rename.BusyTable 15import xiangshan.mem.{LsqEnqCtrl, LsqEnqIO, MemWaitUpdateReq, SqPtr, LqPtr} 16 17sealed trait SchedulerType 18 19case class IntScheduler() extends SchedulerType 20case class MemScheduler() extends SchedulerType 21case class VfScheduler() extends SchedulerType 22case class NoScheduler() extends SchedulerType 23 24class Scheduler(val params: SchdBlockParams)(implicit p: Parameters) extends LazyModule with HasXSParameter { 25 override def shouldBeInlined: Boolean = false 26 27 val numIntStateWrite = backendParams.numPregWb(IntData()) 28 val numVfStateWrite = backendParams.numPregWb(VecData()) 29 30 val dispatch2Iq = LazyModule(new Dispatch2Iq(params)) 31 val issueQueue = params.issueBlockParams.map(x => LazyModule(new IssueQueue(x).suggestName(x.getIQName))) 32 33 lazy val module: SchedulerImpBase = params.schdType match { 34 case IntScheduler() => new SchedulerArithImp(this)(params, p) 35 case MemScheduler() => new SchedulerMemImp(this)(params, p) 36 case VfScheduler() => new SchedulerArithImp(this)(params, p) 37 case _ => null 38 } 39} 40 41class SchedulerIO()(implicit params: SchdBlockParams, p: Parameters) extends XSBundle { 42 // params alias 43 private val LoadQueueSize = VirtualLoadQueueSize 44 45 val fromTop = new Bundle { 46 val hartId = Input(UInt(8.W)) 47 } 48 val fromWbFuBusyTable = new Bundle{ 49 val fuBusyTableRead = MixedVec(params.issueBlockParams.map(x => Input(x.genWbFuBusyTableReadBundle))) 50 } 51 val wbFuBusyTable = MixedVec(params.issueBlockParams.map(x => Output(x.genWbFuBusyTableWriteBundle))) 52 53 val fromCtrlBlock = new Bundle { 54 val pcVec = Input(Vec(params.numPcReadPort, UInt(VAddrData().dataWidth.W))) 55 val flush = Flipped(ValidIO(new Redirect)) 56 } 57 val fromDispatch = new Bundle { 58 val allocPregs = Vec(RenameWidth, Input(new ResetPregStateReq)) 59 val uops = Vec(params.numUopIn, Flipped(DecoupledIO(new DynInst))) 60 } 61 val intWriteBack = MixedVec(Vec(backendParams.numPregWb(IntData()), 62 new RfWritePortWithConfig(backendParams.intPregParams.dataCfg, backendParams.intPregParams.addrWidth))) 63 val vfWriteBack = MixedVec(Vec(backendParams.numPregWb(VecData()), 64 new RfWritePortWithConfig(backendParams.vfPregParams.dataCfg, backendParams.vfPregParams.addrWidth))) 65 val toDataPathAfterDelay: MixedVec[MixedVec[DecoupledIO[IssueQueueIssueBundle]]] = MixedVec(params.issueBlockParams.map(_.genIssueDecoupledBundle)) 66 67 val fromSchedulers = new Bundle { 68 val wakeupVec: MixedVec[ValidIO[IssueQueueIQWakeUpBundle]] = Flipped(params.genIQWakeUpInValidBundle) 69 } 70 71 val toSchedulers = new Bundle { 72 val wakeupVec: MixedVec[ValidIO[IssueQueueIQWakeUpBundle]] = params.genIQWakeUpOutValidBundle 73 } 74 75 val fromDataPath = new Bundle { 76 val resp: MixedVec[MixedVec[OGRespBundle]] = MixedVec(params.issueBlockParams.map(x => Flipped(x.genOGRespBundle))) 77 val og0Cancel = Input(ExuOH(backendParams.numExu)) 78 // Todo: remove this after no cancel signal from og1 79 val og1Cancel = Input(ExuOH(backendParams.numExu)) 80 val cancelToBusyTable = Vec(backendParams.numExu, Flipped(ValidIO(new CancelSignal))) 81 // just be compatible to old code 82 def apply(i: Int)(j: Int) = resp(i)(j) 83 } 84 85 val loadFinalIssueResp = MixedVec(params.issueBlockParams.map(x => MixedVec(Vec(x.LdExuCnt, Flipped(ValidIO(new IssueQueueDeqRespBundle()(p, x))))))) 86 val memAddrIssueResp = MixedVec(params.issueBlockParams.map(x => MixedVec(Vec(x.LdExuCnt, Flipped(ValidIO(new IssueQueueDeqRespBundle()(p, x))))))) 87 88 val ldCancel = Vec(backendParams.LduCnt + backendParams.HyuCnt, Flipped(new LoadCancelIO)) 89 90 val finalBlockMem = OptionWrapper(params.isMemSchd, MixedVec(params.issueBlockParams.map(x => MixedVec(Vec(x.numExu, Input(Bool())))))) 91 92 val memIO = if (params.isMemSchd) Some(new Bundle { 93 val lsqEnqIO = Flipped(new LsqEnqIO) 94 }) else None 95 val fromMem = if (params.isMemSchd) Some(new Bundle { 96 val ldaFeedback = Flipped(Vec(params.LduCnt, new MemRSFeedbackIO)) 97 val staFeedback = Flipped(Vec(params.StaCnt, new MemRSFeedbackIO)) 98 val hyuFeedback = Flipped(Vec(params.HyuCnt, new MemRSFeedbackIO)) 99 val stIssuePtr = Input(new SqPtr()) 100 val lcommit = Input(UInt(log2Up(CommitWidth + 1).W)) 101 val scommit = Input(UInt(log2Ceil(EnsbufferWidth + 1).W)) // connected to `memBlock.io.sqDeq` instead of ROB 102 val lqDeqPtr = Input(new LqPtr) 103 val sqDeqPtr = Input(new SqPtr) 104 // from lsq 105 val lqCancelCnt = Input(UInt(log2Up(LoadQueueSize + 1).W)) 106 val sqCancelCnt = Input(UInt(log2Up(StoreQueueSize + 1).W)) 107 val memWaitUpdateReq = Flipped(new MemWaitUpdateReq) 108 }) else None 109 val toMem = if (params.isMemSchd) Some(new Bundle { 110 val loadFastMatch = Output(Vec(params.LduCnt, new IssueQueueLoadBundle)) 111 }) else None 112} 113 114abstract class SchedulerImpBase(wrapper: Scheduler)(implicit params: SchdBlockParams, p: Parameters) 115 extends LazyModuleImp(wrapper) 116 with HasXSParameter 117{ 118 val io = IO(new SchedulerIO()) 119 120 // alias 121 private val iqWakeUpInMap: Map[Int, ValidIO[IssueQueueIQWakeUpBundle]] = 122 io.fromSchedulers.wakeupVec.map(x => (x.bits.exuIdx, x)).toMap 123 private val schdType = params.schdType 124 125 // Modules 126 val dispatch2Iq: Dispatch2IqImp = wrapper.dispatch2Iq.module 127 val issueQueues: Seq[IssueQueueImp] = wrapper.issueQueue.map(_.module) 128 129 // valid count 130 dispatch2Iq.io.iqValidCnt := issueQueues.filter(_.params.StdCnt == 0).map(_.io.status.validCnt) 131 132 // BusyTable Modules 133 val intBusyTable = schdType match { 134 case IntScheduler() | MemScheduler() => Some(Module(new BusyTable(dispatch2Iq.numIntStateRead, wrapper.numIntStateWrite, IntPhyRegs, IntWB()))) 135 case _ => None 136 } 137 138 val vfBusyTable = schdType match { 139 case VfScheduler() | MemScheduler() => Some(Module(new BusyTable(dispatch2Iq.numVfStateRead, wrapper.numVfStateWrite, VfPhyRegs, VfWB()))) 140 case _ => None 141 } 142 143 dispatch2Iq.io match { case dp2iq => 144 dp2iq.redirect <> io.fromCtrlBlock.flush 145 dp2iq.in <> io.fromDispatch.uops 146 dp2iq.readIntState.foreach(_ <> intBusyTable.get.io.read) 147 dp2iq.readVfState.foreach(_ <> vfBusyTable.get.io.read) 148 } 149 150 intBusyTable match { 151 case Some(bt) => 152 bt.io.allocPregs.zip(io.fromDispatch.allocPregs).foreach { case (btAllocPregs, dpAllocPregs) => 153 btAllocPregs.valid := dpAllocPregs.isInt 154 btAllocPregs.bits := dpAllocPregs.preg 155 } 156 bt.io.wbPregs.zipWithIndex.foreach { case (wb, i) => 157 wb.valid := io.intWriteBack(i).wen && io.intWriteBack(i).intWen 158 wb.bits := io.intWriteBack(i).addr 159 } 160 bt.io.wakeUp := io.fromSchedulers.wakeupVec 161 bt.io.cancel := io.fromDataPath.cancelToBusyTable 162 case None => 163 } 164 165 vfBusyTable match { 166 case Some(bt) => 167 bt.io.allocPregs.zip(io.fromDispatch.allocPregs).foreach { case (btAllocPregs, dpAllocPregs) => 168 btAllocPregs.valid := dpAllocPregs.isFp 169 btAllocPregs.bits := dpAllocPregs.preg 170 } 171 bt.io.wbPregs.zipWithIndex.foreach { case (wb, i) => 172 wb.valid := io.vfWriteBack(i).wen && (io.vfWriteBack(i).fpWen || io.vfWriteBack(i).vecWen) 173 wb.bits := io.vfWriteBack(i).addr 174 } 175 bt.io.wakeUp := io.fromSchedulers.wakeupVec 176 bt.io.cancel := io.fromDataPath.cancelToBusyTable 177 case None => 178 } 179 180 val wakeupFromWBVec = Wire(params.genWBWakeUpSinkValidBundle) 181 val writeback = params.schdType match { 182 case IntScheduler() => io.intWriteBack 183 case MemScheduler() => io.intWriteBack ++ io.vfWriteBack 184 case VfScheduler() => io.vfWriteBack 185 case _ => Seq() 186 } 187 wakeupFromWBVec.zip(writeback).foreach { case (sink, source) => 188 sink.valid := source.wen 189 sink.bits.rfWen := source.intWen 190 sink.bits.fpWen := source.fpWen 191 sink.bits.vecWen := source.vecWen 192 sink.bits.pdest := source.addr 193 } 194 195 // Connect bundles having the same wakeup source 196 issueQueues.zipWithIndex.foreach { case(iq, i) => 197 iq.io.wakeupFromIQ.foreach { wakeUp => 198 wakeUp := iqWakeUpInMap(wakeUp.bits.exuIdx) 199 } 200 iq.io.og0Cancel := io.fromDataPath.og0Cancel 201 iq.io.og1Cancel := io.fromDataPath.og1Cancel 202 iq.io.ldCancel := io.ldCancel 203 if(params.isMemSchd) { 204 iq.io.finalBlock.zip(io.finalBlockMem.get(i)).foreach(x => x._1 := x._2) 205 } else { 206 iq.io.finalBlock.foreach(_ := false.B) 207 } 208 } 209 210 private val iqWakeUpOutMap: Map[Int, ValidIO[IssueQueueIQWakeUpBundle]] = 211 issueQueues.flatMap(_.io.wakeupToIQ) 212 .map(x => (x.bits.exuIdx, x)) 213 .toMap 214 215 // Connect bundles having the same wakeup source 216 io.toSchedulers.wakeupVec.foreach { wakeUp => 217 wakeUp := iqWakeUpOutMap(wakeUp.bits.exuIdx) 218 } 219 220 io.toDataPathAfterDelay.zipWithIndex.foreach { case (toDpDy, i) => 221 toDpDy <> issueQueues(i).io.deqDelay 222 } 223 224 // Response 225 issueQueues.zipWithIndex.foreach { case (iq, i) => 226 iq.io.og0Resp.zipWithIndex.foreach { case (og0Resp, j) => 227 og0Resp := io.fromDataPath(i)(j).og0resp 228 } 229 iq.io.og1Resp.zipWithIndex.foreach { case (og1Resp, j) => 230 og1Resp := io.fromDataPath(i)(j).og1resp 231 } 232 iq.io.finalIssueResp.foreach(_.zipWithIndex.foreach { case (finalIssueResp, j) => 233 if (io.loadFinalIssueResp(i).isDefinedAt(j)) { 234 finalIssueResp := io.loadFinalIssueResp(i)(j) 235 } else { 236 finalIssueResp := 0.U.asTypeOf(finalIssueResp) 237 } 238 }) 239 iq.io.memAddrIssueResp.foreach(_.zipWithIndex.foreach { case (memAddrIssueResp, j) => 240 memAddrIssueResp := io.memAddrIssueResp(i)(j) 241 }) 242 iq.io.wbBusyTableRead := io.fromWbFuBusyTable.fuBusyTableRead(i) 243 io.wbFuBusyTable(i) := iq.io.wbBusyTableWrite 244 } 245 246 println(s"[Scheduler] io.fromSchedulers.wakeupVec: ${io.fromSchedulers.wakeupVec.map(x => backendParams.getExuName(x.bits.exuIdx))}") 247 println(s"[Scheduler] iqWakeUpInKeys: ${iqWakeUpInMap.keys}") 248 249 println(s"[Scheduler] iqWakeUpOutKeys: ${iqWakeUpOutMap.keys}") 250 println(s"[Scheduler] io.toSchedulers.wakeupVec: ${io.toSchedulers.wakeupVec.map(x => backendParams.getExuName(x.bits.exuIdx))}") 251} 252 253class SchedulerArithImp(override val wrapper: Scheduler)(implicit params: SchdBlockParams, p: Parameters) 254 extends SchedulerImpBase(wrapper) 255 with HasXSParameter 256{ 257// dontTouch(io.vfWbFuBusyTable) 258 println(s"[SchedulerArithImp] " + 259 s"has intBusyTable: ${intBusyTable.nonEmpty}, " + 260 s"has vfBusyTable: ${vfBusyTable.nonEmpty}") 261 262 issueQueues.zipWithIndex.foreach { case (iq, i) => 263 iq.io.flush <> io.fromCtrlBlock.flush 264 iq.io.enq <> dispatch2Iq.io.out(i) 265 iq.io.wakeupFromWB := wakeupFromWBVec 266 } 267} 268 269// FIXME: Vector mem instructions may not be handled properly! 270class SchedulerMemImp(override val wrapper: Scheduler)(implicit params: SchdBlockParams, p: Parameters) 271 extends SchedulerImpBase(wrapper) 272 with HasXSParameter 273{ 274 println(s"[SchedulerMemImp] " + 275 s"has intBusyTable: ${intBusyTable.nonEmpty}, " + 276 s"has vfBusyTable: ${vfBusyTable.nonEmpty}") 277 278 val memAddrIQs = issueQueues.filter(_.params.isMemAddrIQ) 279 val stAddrIQs = issueQueues.filter(iq => iq.params.StaCnt > 0 || iq.params.VstaCnt > 0) // included in memAddrIQs 280 val ldAddrIQs = issueQueues.filter(iq => iq.params.LduCnt > 0 || iq.params.VlduCnt > 0) 281 val stDataIQs = issueQueues.filter(iq => iq.params.StdCnt > 0 || iq.params.VstdCnt > 0) 282 val vecMemIQs = issueQueues.filter(_.params.isVecMemIQ) 283 val (hyuIQs, hyuIQIdxs) = issueQueues.zipWithIndex.filter(_._1.params.HyuCnt > 0).unzip 284 285 println(s"[SchedulerMemImp] memAddrIQs.size: ${memAddrIQs.size}, enq.size: ${memAddrIQs.map(_.io.enq.size).sum}") 286 println(s"[SchedulerMemImp] stAddrIQs.size: ${stAddrIQs.size }, enq.size: ${stAddrIQs.map(_.io.enq.size).sum}") 287 println(s"[SchedulerMemImp] ldAddrIQs.size: ${ldAddrIQs.size }, enq.size: ${ldAddrIQs.map(_.io.enq.size).sum}") 288 println(s"[SchedulerMemImp] stDataIQs.size: ${stDataIQs.size }, enq.size: ${stDataIQs.map(_.io.enq.size).sum}") 289 println(s"[SchedulerMemImp] hyuIQs.size: ${hyuIQs.size }, enq.size: ${hyuIQs.map(_.io.enq.size).sum}") 290 require(memAddrIQs.nonEmpty && stDataIQs.nonEmpty) 291 292 io.toMem.get.loadFastMatch := 0.U.asTypeOf(io.toMem.get.loadFastMatch) // TODO: is still needed? 293 294 memAddrIQs.zipWithIndex.foreach { case (iq, i) => 295 iq.io.flush <> io.fromCtrlBlock.flush 296 iq.io.enq <> dispatch2Iq.io.out(i) 297 iq.io.wakeupFromWB := wakeupFromWBVec 298 } 299 300 ldAddrIQs.zipWithIndex.foreach { 301 case (imp: IssueQueueMemAddrImp, i) => 302 imp.io.memIO.get.feedbackIO.head := 0.U.asTypeOf(imp.io.memIO.get.feedbackIO.head) 303 imp.io.memIO.get.checkWait.stIssuePtr := io.fromMem.get.stIssuePtr 304 imp.io.memIO.get.checkWait.memWaitUpdateReq := io.fromMem.get.memWaitUpdateReq 305 case _ => 306 } 307 308 stAddrIQs.zipWithIndex.foreach { 309 case (imp: IssueQueueMemAddrImp, i) => 310 imp.io.memIO.get.feedbackIO.head := io.fromMem.get.staFeedback(i) 311 imp.io.memIO.get.checkWait.stIssuePtr := io.fromMem.get.stIssuePtr 312 imp.io.memIO.get.checkWait.memWaitUpdateReq := io.fromMem.get.memWaitUpdateReq 313 case _ => 314 } 315 316 hyuIQs.zip(hyuIQIdxs).foreach { 317 case (imp: IssueQueueMemAddrImp, idx) => 318 imp.io.memIO.get.feedbackIO.head := io.fromMem.get.hyuFeedback.head 319 imp.io.memIO.get.feedbackIO(1) := 0.U.asTypeOf(imp.io.memIO.get.feedbackIO(1)) 320 imp.io.memIO.get.checkWait.stIssuePtr := io.fromMem.get.stIssuePtr 321 imp.io.memIO.get.checkWait.memWaitUpdateReq := io.fromMem.get.memWaitUpdateReq 322 // TODO: refactor ditry code 323 imp.io.deqDelay(1).ready := false.B 324 io.toDataPathAfterDelay(idx)(1).valid := false.B 325 io.toDataPathAfterDelay(idx)(1).bits := 0.U.asTypeOf(io.toDataPathAfterDelay(idx)(1).bits) 326 case _ => 327 } 328 329 private val staIdxSeq = (stAddrIQs).map(iq => iq.params.idxInSchBlk) 330 private val hyaIdxSeq = (hyuIQs).map(iq => iq.params.idxInSchBlk) 331 332 println(s"[SchedulerMemImp] sta iq idx in memSchdBlock: $staIdxSeq") 333 println(s"[SchedulerMemImp] hya iq idx in memSchdBlock: $hyaIdxSeq") 334 335 private val staEnqs = stAddrIQs.map(_.io.enq).flatten 336 private val stdEnqs = stDataIQs.map(_.io.enq).flatten.take(staEnqs.size) 337 private val hyaEnqs = hyuIQs.map(_.io.enq).flatten 338 private val hydEnqs = stDataIQs.map(_.io.enq).flatten.drop(staEnqs.size) 339 340 require(staEnqs.size == stdEnqs.size, s"number of enq ports of store address IQs(${staEnqs.size}) " + 341 s"should be equal to number of enq ports of store data IQs(${stdEnqs.size})") 342 343 require(hyaEnqs.size == hydEnqs.size, s"number of enq ports of hybrid address IQs(${hyaEnqs.size}) " + 344 s"should be equal to number of enq ports of hybrid data IQs(${hydEnqs.size})") 345 346 for ((idxInSchBlk, i) <- staIdxSeq.zipWithIndex) { 347 dispatch2Iq.io.out(idxInSchBlk).zip(staEnqs).zip(stdEnqs).foreach{ case((dp, staIQ), stdIQ) => 348 val isAllReady = staIQ.ready && stdIQ.ready 349 dp.ready := isAllReady 350 staIQ.valid := dp.valid && isAllReady 351 stdIQ.valid := dp.valid && isAllReady && FuType.isStore(dp.bits.fuType) 352 } 353 } 354 355 for ((idxInSchBlk, i) <- hyaIdxSeq.zipWithIndex) { 356 dispatch2Iq.io.out(idxInSchBlk).zip(hyaEnqs).zip(hydEnqs).foreach{ case((dp, hyaIQ), hydIQ) => 357 val isAllReady = hyaIQ.ready && hydIQ.ready 358 dp.ready := isAllReady 359 hyaIQ.valid := dp.valid && isAllReady 360 hydIQ.valid := dp.valid && isAllReady && FuType.FuTypeOrR(dp.bits.fuType, FuType.stu, FuType.mou) 361 } 362 } 363 364 stDataIQs.zipWithIndex.foreach { case (iq, i) => 365 iq.io.flush <> io.fromCtrlBlock.flush 366 iq.io.wakeupFromWB := wakeupFromWBVec 367 } 368 369 (stdEnqs ++ hydEnqs).zip(staEnqs ++ hyaEnqs).zipWithIndex.foreach { case ((stdIQEnq, staIQEnq), i) => 370 stdIQEnq.bits := staIQEnq.bits 371 // Store data reuses store addr src(1) in dispatch2iq 372 // [dispatch2iq] --src*------src*(0)--> [staIQ|hyaIQ] 373 // \ 374 // ---src*(1)--> [stdIQ] 375 // Since the src(1) of sta is easier to get, stdIQEnq.bits.src*(0) is assigned to staIQEnq.bits.src*(1) 376 // instead of dispatch2Iq.io.out(x).bits.src*(1) 377 val stdIdx = 1 378 stdIQEnq.bits.srcState(0) := staIQEnq.bits.srcState(stdIdx) 379 stdIQEnq.bits.srcType(0) := staIQEnq.bits.srcType(stdIdx) 380 stdIQEnq.bits.psrc(0) := staIQEnq.bits.psrc(stdIdx) 381 stdIQEnq.bits.sqIdx := staIQEnq.bits.sqIdx 382 } 383 384 vecMemIQs.foreach { 385 case imp: IssueQueueVecMemImp => 386 imp.io.memIO.get.sqDeqPtr.foreach(_ := io.fromMem.get.sqDeqPtr) 387 imp.io.memIO.get.lqDeqPtr.foreach(_ := io.fromMem.get.lqDeqPtr) 388 // not used 389 imp.io.memIO.get.feedbackIO := 0.U.asTypeOf(imp.io.memIO.get.feedbackIO) 390 // maybe not used 391 imp.io.memIO.get.checkWait.stIssuePtr := io.fromMem.get.stIssuePtr 392 imp.io.memIO.get.checkWait.memWaitUpdateReq := io.fromMem.get.memWaitUpdateReq 393 case _ => 394 } 395 396 val lsqEnqCtrl = Module(new LsqEnqCtrl) 397 398 lsqEnqCtrl.io.redirect <> io.fromCtrlBlock.flush 399 lsqEnqCtrl.io.enq <> dispatch2Iq.io.enqLsqIO.get 400 lsqEnqCtrl.io.lcommit := io.fromMem.get.lcommit 401 lsqEnqCtrl.io.scommit := io.fromMem.get.scommit 402 lsqEnqCtrl.io.lqCancelCnt := io.fromMem.get.lqCancelCnt 403 lsqEnqCtrl.io.sqCancelCnt := io.fromMem.get.sqCancelCnt 404 io.memIO.get.lsqEnqIO <> lsqEnqCtrl.io.enqLsq 405} 406