xref: /XiangShan/src/main/scala/xiangshan/backend/datapath/DataPath.scala (revision d91483a658064c7276ee0181b0c527a3e2a7d2ee)
1730cfbc0SXuan Hupackage xiangshan.backend.datapath
2730cfbc0SXuan Hu
3730cfbc0SXuan Huimport chipsalliance.rocketchip.config.Parameters
4730cfbc0SXuan Huimport chisel3._
5730cfbc0SXuan Huimport chisel3.util._
6730cfbc0SXuan Huimport difftest.{DifftestArchFpRegState, DifftestArchIntRegState, DifftestArchVecRegState}
7730cfbc0SXuan Huimport freechips.rocketchip.diplomacy.{LazyModule, LazyModuleImp}
8730cfbc0SXuan Huimport utility._
9730cfbc0SXuan Huimport xiangshan._
10730cfbc0SXuan Huimport xiangshan.backend.BackendParams
11730cfbc0SXuan Huimport xiangshan.backend.datapath.DataConfig._
12730cfbc0SXuan Huimport xiangshan.backend.datapath.RdConfig._
13730cfbc0SXuan Huimport xiangshan.backend.issue.{ImmExtractor, IntScheduler, MemScheduler, VfScheduler}
14730cfbc0SXuan Huimport xiangshan.backend.Bundles._
15730cfbc0SXuan Huimport xiangshan.backend.regfile._
16730cfbc0SXuan Hu
17730cfbc0SXuan Huclass RFArbiterBundle(addrWidth: Int)(implicit p: Parameters) extends XSBundle {
18730cfbc0SXuan Hu  val addr = UInt(addrWidth.W)
19730cfbc0SXuan Hu}
20730cfbc0SXuan Hu
21730cfbc0SXuan Huclass RFReadArbiterIO(inPortSize: Int, outPortSize: Int, pregWidth: Int)(implicit p: Parameters) extends XSBundle {
22730cfbc0SXuan Hu  val in = Vec(inPortSize, Flipped(DecoupledIO(new RFArbiterBundle(pregWidth))))
23730cfbc0SXuan Hu  val out = Vec(outPortSize, Valid(new RFArbiterBundle(pregWidth)))
24730cfbc0SXuan Hu  val flush = Flipped(ValidIO(new Redirect))
25730cfbc0SXuan Hu}
26730cfbc0SXuan Hu
27730cfbc0SXuan Huclass RFReadArbiter(isInt: Boolean)(implicit p: Parameters) extends XSModule {
28730cfbc0SXuan Hu  val allExuParams = backendParams.allExuParams
29730cfbc0SXuan Hu
30730cfbc0SXuan Hu  val portConfigs = allExuParams.map(_.rfrPortConfigs.flatten).flatten.filter{
31730cfbc0SXuan Hu    rfrPortConfigs =>
32730cfbc0SXuan Hu      if(isInt){
33730cfbc0SXuan Hu        rfrPortConfigs.isInstanceOf[IntRD]
34730cfbc0SXuan Hu      }
35730cfbc0SXuan Hu      else{
36730cfbc0SXuan Hu        rfrPortConfigs.isInstanceOf[VfRD]
37730cfbc0SXuan Hu      }
38730cfbc0SXuan Hu  }
39730cfbc0SXuan Hu
40730cfbc0SXuan Hu  val pregParams = if(isInt) backendParams.intPregParams else backendParams.vfPregParams
41730cfbc0SXuan Hu
42730cfbc0SXuan Hu  val io = IO(new RFReadArbiterIO(portConfigs.size, backendParams.numRfRead, pregParams.addrWidth))
43730cfbc0SXuan Hu  // inGroup[port -> Bundle]
44730cfbc0SXuan Hu  val inGroup: Map[Int, IndexedSeq[(DecoupledIO[RFArbiterBundle], RdConfig)]] = io.in.zip(portConfigs).groupBy{ case(port, config) => config.port}
45730cfbc0SXuan Hu  // sort by priority
46730cfbc0SXuan Hu  val inGroupSorted: Map[Int, IndexedSeq[(DecoupledIO[RFArbiterBundle], RdConfig)]] = inGroup.map{
47730cfbc0SXuan Hu    case(key, value) => (key -> value.sortBy{ case(port, config) => config.priority})
48730cfbc0SXuan Hu  }
49730cfbc0SXuan Hu
50730cfbc0SXuan Hu  private val arbiters: Seq[Option[Arbiter[RFArbiterBundle]]] = Seq.tabulate(backendParams.numRfRead) { x => {
51730cfbc0SXuan Hu    if (inGroupSorted.contains(x)) {
52730cfbc0SXuan Hu      Some(Module(new Arbiter(new RFArbiterBundle(pregParams.addrWidth), inGroupSorted(x).length)))
53730cfbc0SXuan Hu    } else {
54730cfbc0SXuan Hu      None
55730cfbc0SXuan Hu    }
56730cfbc0SXuan Hu  }}
57730cfbc0SXuan Hu
58730cfbc0SXuan Hu  arbiters.zipWithIndex.foreach { case (arb, i) =>
59730cfbc0SXuan Hu    if (arb.nonEmpty) {
60730cfbc0SXuan Hu      arb.get.io.in.zip(inGroupSorted(i).map(_._1)).foreach { case (arbIn, addrIn) =>
61730cfbc0SXuan Hu        arbIn <> addrIn
62730cfbc0SXuan Hu      }
63730cfbc0SXuan Hu    }
64730cfbc0SXuan Hu  }
65730cfbc0SXuan Hu
66730cfbc0SXuan Hu  io.out.zip(arbiters).foreach { case (addrOut, arb) =>
67730cfbc0SXuan Hu    if (arb.nonEmpty) {
68730cfbc0SXuan Hu      val arbOut = arb.get.io.out
69730cfbc0SXuan Hu      arbOut.ready := true.B
70730cfbc0SXuan Hu      addrOut.valid := arbOut.valid
71730cfbc0SXuan Hu      addrOut.bits := arbOut.bits
72730cfbc0SXuan Hu    } else {
73730cfbc0SXuan Hu      addrOut := 0.U.asTypeOf(addrOut)
74730cfbc0SXuan Hu    }
75730cfbc0SXuan Hu  }
76730cfbc0SXuan Hu}
77730cfbc0SXuan Hu
78730cfbc0SXuan Huclass DataPath(params: BackendParams)(implicit p: Parameters) extends LazyModule {
79730cfbc0SXuan Hu  private implicit val dpParams: BackendParams = params
80730cfbc0SXuan Hu  lazy val module = new DataPathImp(this)
81730cfbc0SXuan Hu}
82730cfbc0SXuan Hu
83730cfbc0SXuan Huclass DataPathImp(override val wrapper: DataPath)(implicit p: Parameters, params: BackendParams)
84730cfbc0SXuan Hu  extends LazyModuleImp(wrapper) with HasXSParameter {
85730cfbc0SXuan Hu
86*d91483a6Sfdy  private val VCONFIG_PORT = params.vconfigPort
87*d91483a6Sfdy
88730cfbc0SXuan Hu  val io = IO(new DataPathIO())
89730cfbc0SXuan Hu
90730cfbc0SXuan Hu  private val (fromIntIQ, toIntIQ, toIntExu) = (io.fromIntIQ, io.toIntIQ, io.toIntExu)
91730cfbc0SXuan Hu  private val (fromMemIQ, toMemIQ, toMemExu) = (io.fromMemIQ, io.toMemIQ, io.toMemExu)
92730cfbc0SXuan Hu  private val (fromVfIQ , toVfIQ , toVfExu ) = (io.fromVfIQ , io.toVfIQ , io.toFpExu)
93730cfbc0SXuan Hu
94730cfbc0SXuan Hu  println(s"[DataPath] IntIQ(${fromIntIQ.size}), MemIQ(${fromMemIQ.size})")
95730cfbc0SXuan Hu  println(s"[DataPath] IntExu(${fromIntIQ.map(_.size).sum}), MemExu(${fromMemIQ.map(_.size).sum})")
96730cfbc0SXuan Hu
97730cfbc0SXuan Hu  // just refences for convience
98730cfbc0SXuan Hu  private val fromIQ = fromIntIQ ++ fromVfIQ ++ fromMemIQ
99730cfbc0SXuan Hu
100730cfbc0SXuan Hu  private val toIQs = toIntIQ ++ toVfIQ ++ toMemIQ
101730cfbc0SXuan Hu
102730cfbc0SXuan Hu  private val toExu = toIntExu ++ toVfExu ++ toMemExu
103730cfbc0SXuan Hu
104730cfbc0SXuan Hu  private val intRFReadArbiter = Module(new RFReadArbiter(true))
105730cfbc0SXuan Hu  private val vfRFReadArbiter = Module(new RFReadArbiter(false))
106730cfbc0SXuan Hu
107730cfbc0SXuan Hu  private val issuePortsIn = fromIQ.flatten
108730cfbc0SXuan Hu  private val intBlocks = fromIQ.map{ case iq => Wire(Vec(iq.size, Bool())) }
109730cfbc0SXuan Hu  private val intBlocksSeq = intBlocks.flatten
110730cfbc0SXuan Hu  private val vfBlocks = fromIQ.map { case iq => Wire(Vec(iq.size, Bool())) }
111730cfbc0SXuan Hu  private val vfBlocksSeq = vfBlocks.flatten
112730cfbc0SXuan Hu
113730cfbc0SXuan Hu  val intReadPortInSize = issuePortsIn.map(issuePortIn => issuePortIn.bits.getIntRfReadBundle.size).scan(0)(_ + _)
114730cfbc0SXuan Hu  issuePortsIn.zipWithIndex.foreach{
115730cfbc0SXuan Hu    case (issuePortIn, idx) =>
116730cfbc0SXuan Hu      val readPortIn = issuePortIn.bits.getIntRfReadBundle
117730cfbc0SXuan Hu      val l = intReadPortInSize(idx)
118730cfbc0SXuan Hu      val r = intReadPortInSize(idx + 1)
119730cfbc0SXuan Hu      val arbiterIn = intRFReadArbiter.io.in.slice(l, r)
120730cfbc0SXuan Hu      arbiterIn.zip(readPortIn).foreach{
121730cfbc0SXuan Hu        case(sink, source) =>
122730cfbc0SXuan Hu          sink.bits.addr := source.addr
123730cfbc0SXuan Hu          sink.valid := issuePortIn.valid && SrcType.isXp(source.srcType)
124730cfbc0SXuan Hu      }
125730cfbc0SXuan Hu      if(r > l){
126730cfbc0SXuan Hu        intBlocksSeq(idx) := !arbiterIn.zip(readPortIn).map {
127730cfbc0SXuan Hu          case (sink, source) => Mux(SrcType.isXp(source.srcType), sink.ready, true.B)
128730cfbc0SXuan Hu        }.reduce(_ & _)
129730cfbc0SXuan Hu      }
130730cfbc0SXuan Hu      else{
131730cfbc0SXuan Hu        intBlocksSeq(idx) := false.B
132730cfbc0SXuan Hu      }
133730cfbc0SXuan Hu  }
134730cfbc0SXuan Hu  intRFReadArbiter.io.flush := io.flush
135730cfbc0SXuan Hu
136730cfbc0SXuan Hu  val vfReadPortInSize = issuePortsIn.map(issuePortIn => issuePortIn.bits.getFpRfReadBundle.size).scan(0)(_ + _)
137730cfbc0SXuan Hu  issuePortsIn.zipWithIndex.foreach {
138730cfbc0SXuan Hu    case (issuePortIn, idx) =>
139730cfbc0SXuan Hu      val readPortIn = issuePortIn.bits.getFpRfReadBundle
140730cfbc0SXuan Hu      val l = vfReadPortInSize(idx)
141730cfbc0SXuan Hu      val r = vfReadPortInSize(idx + 1)
142730cfbc0SXuan Hu      val arbiterIn = vfRFReadArbiter.io.in.slice(l, r)
143730cfbc0SXuan Hu      arbiterIn.zip(readPortIn).foreach {
144730cfbc0SXuan Hu        case (sink, source) =>
145730cfbc0SXuan Hu          sink.bits.addr := source.addr
146730cfbc0SXuan Hu          sink.valid := issuePortIn.valid && SrcType.isVfp(source.srcType)
147730cfbc0SXuan Hu      }
148730cfbc0SXuan Hu      if (r > l) {
149730cfbc0SXuan Hu        vfBlocksSeq(idx) := !arbiterIn.zip(readPortIn).map {
150730cfbc0SXuan Hu          case (sink, source) => Mux(SrcType.isVfp(source.srcType), sink.ready, true.B)
151730cfbc0SXuan Hu        }.reduce(_ & _)
152730cfbc0SXuan Hu      }
153730cfbc0SXuan Hu      else {
154730cfbc0SXuan Hu        vfBlocksSeq(idx) := false.B
155730cfbc0SXuan Hu      }
156730cfbc0SXuan Hu  }
157730cfbc0SXuan Hu  vfRFReadArbiter.io.flush := io.flush
158730cfbc0SXuan Hu
159730cfbc0SXuan Hu  private val intSchdParams = params.schdParams(IntScheduler())
160730cfbc0SXuan Hu  private val vfSchdParams = params.schdParams(VfScheduler())
161730cfbc0SXuan Hu  private val memSchdParams = params.schdParams(MemScheduler())
162730cfbc0SXuan Hu
163730cfbc0SXuan Hu  private val numIntRfReadByExu = intSchdParams.numIntRfReadByExu + memSchdParams.numIntRfReadByExu
164730cfbc0SXuan Hu  private val numVfRfReadByExu = vfSchdParams.numVfRfReadByExu + memSchdParams.numVfRfReadByExu
165730cfbc0SXuan Hu  // Todo: limit read port
166730cfbc0SXuan Hu  private val numIntR = numIntRfReadByExu
167730cfbc0SXuan Hu  private val numVfR = numVfRfReadByExu
168730cfbc0SXuan Hu  println(s"[DataPath] RegFile read req needed by Exu: Int(${numIntRfReadByExu}), Vf(${numVfRfReadByExu})")
169730cfbc0SXuan Hu  println(s"[DataPath] RegFile read port: Int(${numIntR}), Vf(${numVfR})")
170730cfbc0SXuan Hu
171730cfbc0SXuan Hu  private val schdParams = params.allSchdParams
172730cfbc0SXuan Hu
173730cfbc0SXuan Hu  private val intRfRaddr = Wire(Vec(params.numRfRead, UInt(intSchdParams.pregIdxWidth.W)))
174730cfbc0SXuan Hu  private val intRfRdata = Wire(Vec(params.numRfRead, UInt(intSchdParams.rfDataWidth.W)))
175730cfbc0SXuan Hu  private val intRfWen = Wire(Vec(io.fromIntWb.length, Bool()))
176730cfbc0SXuan Hu  private val intRfWaddr = Wire(Vec(io.fromIntWb.length, UInt(intSchdParams.pregIdxWidth.W)))
177730cfbc0SXuan Hu  private val intRfWdata = Wire(Vec(io.fromIntWb.length, UInt(intSchdParams.rfDataWidth.W)))
178730cfbc0SXuan Hu
179730cfbc0SXuan Hu  private val vfRfSplitNum = VLEN / XLEN
180730cfbc0SXuan Hu  private val vfRfRaddr = Wire(Vec(params.numRfRead, UInt(vfSchdParams.pregIdxWidth.W)))
181730cfbc0SXuan Hu  private val vfRfRdata = Wire(Vec(params.numRfRead, UInt(vfSchdParams.rfDataWidth.W)))
182730cfbc0SXuan Hu  private val vfRfWen = Wire(Vec(vfRfSplitNum, Vec(io.fromVfWb.length, Bool())))
183730cfbc0SXuan Hu  private val vfRfWaddr = Wire(Vec(io.fromVfWb.length, UInt(vfSchdParams.pregIdxWidth.W)))
184730cfbc0SXuan Hu  private val vfRfWdata = Wire(Vec(io.fromVfWb.length, UInt(vfSchdParams.rfDataWidth.W)))
185730cfbc0SXuan Hu
186730cfbc0SXuan Hu  private val intDebugRead: Option[(Vec[UInt], Vec[UInt])] =
187730cfbc0SXuan Hu    if (env.AlwaysBasicDiff || env.EnableDifftest) {
188730cfbc0SXuan Hu      Some(Wire(Vec(32, UInt(intSchdParams.pregIdxWidth.W))), Wire(Vec(32, UInt(XLEN.W))))
189730cfbc0SXuan Hu    } else { None }
190730cfbc0SXuan Hu  private val vfDebugRead: Option[(Vec[UInt], Vec[UInt])] =
191730cfbc0SXuan Hu    if (env.AlwaysBasicDiff || env.EnableDifftest) {
192730cfbc0SXuan Hu      Some(Wire(Vec(32 + 32, UInt(vfSchdParams.pregIdxWidth.W))), Wire(Vec(32 + 32, UInt(VLEN.W))))
193730cfbc0SXuan Hu    } else { None }
194730cfbc0SXuan Hu
195730cfbc0SXuan Hu  private val fpDebugReadData: Option[Vec[UInt]] =
196730cfbc0SXuan Hu    if (env.AlwaysBasicDiff || env.EnableDifftest) {
197730cfbc0SXuan Hu      Some(Wire(Vec(32, UInt(XLEN.W))))
198730cfbc0SXuan Hu    } else { None }
199730cfbc0SXuan Hu  private val vecDebugReadData: Option[Vec[UInt]] =
200730cfbc0SXuan Hu    if (env.AlwaysBasicDiff || env.EnableDifftest) {
201730cfbc0SXuan Hu      Some(Wire(Vec(64, UInt(64.W)))) // v0 = Cat(Vec(1), Vec(0))
202730cfbc0SXuan Hu    } else { None }
203730cfbc0SXuan Hu
204730cfbc0SXuan Hu  fpDebugReadData.foreach(_ := vfDebugRead
205730cfbc0SXuan Hu    .get._2
206730cfbc0SXuan Hu    .slice(0, 32)
207730cfbc0SXuan Hu    .map(_(63, 0))
208730cfbc0SXuan Hu  ) // fp only used [63, 0]
209730cfbc0SXuan Hu  vecDebugReadData.foreach(_ := vfDebugRead
210730cfbc0SXuan Hu    .get._2
211730cfbc0SXuan Hu    .slice(32, 64)
212730cfbc0SXuan Hu    .map(x => Seq(x(63, 0), x(127, 64))).flatten
213730cfbc0SXuan Hu  )
214730cfbc0SXuan Hu
215730cfbc0SXuan Hu  IntRegFile("IntRegFile", intSchdParams.numPregs, intRfRaddr, intRfRdata, intRfWen, intRfWaddr, intRfWdata,
216730cfbc0SXuan Hu    debugReadAddr = intDebugRead.map(_._1),
217730cfbc0SXuan Hu    debugReadData = intDebugRead.map(_._2))
218730cfbc0SXuan Hu  VfRegFile("VfRegFile", vfSchdParams.numPregs, vfRfSplitNum, vfRfRaddr, vfRfRdata, vfRfWen, vfRfWaddr, vfRfWdata,
219730cfbc0SXuan Hu    debugReadAddr = vfDebugRead.map(_._1),
220730cfbc0SXuan Hu    debugReadData = vfDebugRead.map(_._2))
221730cfbc0SXuan Hu
222730cfbc0SXuan Hu  intRfWaddr := io.fromIntWb.map(_.addr)
223730cfbc0SXuan Hu  intRfWdata := io.fromIntWb.map(_.data)
224730cfbc0SXuan Hu  intRfWen := io.fromIntWb.map(_.wen)
225730cfbc0SXuan Hu
226730cfbc0SXuan Hu  intRFReadArbiter.io.out.map(_.bits.addr).zip(intRfRaddr).foreach{ case(source, sink) => sink := source }
227730cfbc0SXuan Hu
228730cfbc0SXuan Hu  vfRfWaddr := io.fromVfWb.map(_.addr)
229730cfbc0SXuan Hu  vfRfWdata := io.fromVfWb.map(_.data)
230730cfbc0SXuan Hu  vfRfWen.foreach(_.zip(io.fromVfWb.map(_.wen)).foreach { case (wenSink, wenSource) => wenSink := wenSource } )// Todo: support fp multi-write
231730cfbc0SXuan Hu
232730cfbc0SXuan Hu  vfRFReadArbiter.io.out.map(_.bits.addr).zip(vfRfRaddr).foreach{ case(source, sink) => sink := source }
233*d91483a6Sfdy  vfRfRaddr(VCONFIG_PORT) := io.vconfigReadPort.addr
234*d91483a6Sfdy  io.vconfigReadPort.data := vfRfRdata(VCONFIG_PORT)
235730cfbc0SXuan Hu
236730cfbc0SXuan Hu  intDebugRead.foreach { case (addr, _) =>
237730cfbc0SXuan Hu    addr := io.debugIntRat
238730cfbc0SXuan Hu  }
239730cfbc0SXuan Hu
240730cfbc0SXuan Hu  vfDebugRead.foreach { case (addr, _) =>
241730cfbc0SXuan Hu    addr := io.debugFpRat ++ io.debugVecRat
242730cfbc0SXuan Hu  }
243730cfbc0SXuan Hu  println(s"[DataPath] " +
244730cfbc0SXuan Hu    s"has intDebugRead: ${intDebugRead.nonEmpty}, " +
245730cfbc0SXuan Hu    s"has vfDebugRead: ${vfDebugRead.nonEmpty}")
246730cfbc0SXuan Hu
247730cfbc0SXuan Hu  val s1_addrOHs = Reg(MixedVec(
248730cfbc0SXuan Hu    fromIQ.map(x => MixedVec(x.map(_.bits.addrOH.cloneType)))
249730cfbc0SXuan Hu  ))
250730cfbc0SXuan Hu  val s1_toExuValid: MixedVec[MixedVec[Bool]] = Reg(MixedVec(
251730cfbc0SXuan Hu    toExu.map(x => MixedVec(x.map(_.valid.cloneType)))
252730cfbc0SXuan Hu  ))
253730cfbc0SXuan Hu  val s1_toExuData: MixedVec[MixedVec[ExuInput]] = Reg(MixedVec(toExu.map(x => MixedVec(x.map(_.bits.cloneType)))))
254730cfbc0SXuan Hu  val s1_toExuReady = Wire(MixedVec(toExu.map(x => MixedVec(x.map(_.ready.cloneType))))) // Todo
255730cfbc0SXuan Hu  val s1_srcType: MixedVec[MixedVec[Vec[UInt]]] = MixedVecInit(fromIQ.map(x => MixedVecInit(x.map(xx => RegEnable(xx.bits.srcType, xx.fire)))))
256730cfbc0SXuan Hu
257730cfbc0SXuan Hu  val s1_intPregRData: MixedVec[MixedVec[Vec[UInt]]] = Wire(MixedVec(toExu.map(x => MixedVec(x.map(_.bits.src.cloneType)))))
258730cfbc0SXuan Hu  val s1_vfPregRData: MixedVec[MixedVec[Vec[UInt]]] = Wire(MixedVec(toExu.map(x => MixedVec(x.map(_.bits.src.cloneType)))))
259730cfbc0SXuan Hu
260730cfbc0SXuan Hu  val rfrPortConfigs = schdParams.map(_.issueBlockParams).flatten.map(_.exuBlockParams.map(_.rfrPortConfigs))
261730cfbc0SXuan Hu
262730cfbc0SXuan Hu  println(s"[DataPath] s1_intPregRData.flatten.flatten.size: ${s1_intPregRData.flatten.flatten.size}, intRfRdata.size: ${intRfRdata.size}")
263730cfbc0SXuan Hu  s1_intPregRData.foreach(_.foreach(_.foreach(_ := 0.U)))
264730cfbc0SXuan Hu  s1_intPregRData.zip(rfrPortConfigs).foreach { case (iqRdata, iqCfg) =>
265730cfbc0SXuan Hu      iqRdata.zip(iqCfg).foreach { case (iuRdata, iuCfg) =>
266730cfbc0SXuan Hu        val realIuCfg = iuCfg.map(x => if(x.size > 1) x.filter(_.isInstanceOf[IntRD]) else x).flatten
267730cfbc0SXuan Hu        assert(iuRdata.size == realIuCfg.size, "iuRdata.size != realIuCfg.size")
268730cfbc0SXuan Hu        iuRdata.zip(realIuCfg)
269730cfbc0SXuan Hu          .filter { case (_, rfrPortConfig) => rfrPortConfig.isInstanceOf[IntRD] }
270730cfbc0SXuan Hu          .foreach { case (sink, cfg) => sink := intRfRdata(cfg.port) }
271730cfbc0SXuan Hu      }
272730cfbc0SXuan Hu  }
273730cfbc0SXuan Hu
274730cfbc0SXuan Hu  println(s"[DataPath] s1_vfPregRData.flatten.flatten.size: ${s1_vfPregRData.flatten.flatten.size}, vfRfRdata.size: ${vfRfRdata.size}")
275730cfbc0SXuan Hu  s1_vfPregRData.foreach(_.foreach(_.foreach(_ := 0.U)))
276730cfbc0SXuan Hu  s1_vfPregRData.zip(rfrPortConfigs).foreach{ case(iqRdata, iqCfg) =>
277730cfbc0SXuan Hu      iqRdata.zip(iqCfg).foreach{ case(iuRdata, iuCfg) =>
278730cfbc0SXuan Hu        val realIuCfg = iuCfg.map(x => if(x.size > 1) x.filter(_.isInstanceOf[VfRD]) else x).flatten
279730cfbc0SXuan Hu        assert(iuRdata.size == realIuCfg.size, "iuRdata.size != realIuCfg.size")
280730cfbc0SXuan Hu        iuRdata.zip(realIuCfg)
281730cfbc0SXuan Hu          .filter { case (_, rfrPortConfig) => rfrPortConfig.isInstanceOf[VfRD] }
282730cfbc0SXuan Hu          .foreach { case (sink, cfg) => sink := vfRfRdata(cfg.port) }
283730cfbc0SXuan Hu      }
284730cfbc0SXuan Hu  }
285730cfbc0SXuan Hu
286730cfbc0SXuan Hu  //  var intRfRdataIdx = 0
287730cfbc0SXuan Hu//  var vfRfRdataIdx = 0
288730cfbc0SXuan Hu//  for (iqIdx <- toExu.indices) {
289730cfbc0SXuan Hu//    for (exuIdx <- toExu(iqIdx).indices) {
290730cfbc0SXuan Hu//      for (srcIdx <- toExu(iqIdx)(exuIdx).bits.src.indices) {
291730cfbc0SXuan Hu//        val readDataCfgSet = toExu(iqIdx)(exuIdx).bits.params.getSrcDataType(srcIdx)
292730cfbc0SXuan Hu//        // need read int reg
293730cfbc0SXuan Hu//        if (readDataCfgSet.intersect(IntRegSrcDataSet).nonEmpty) {
294730cfbc0SXuan Hu//          println(s"[DataPath] (iqIdx, exuIdx, srcIdx): ($iqIdx, $exuIdx, $srcIdx)")
295730cfbc0SXuan Hu//          s1_intPregRData(iqIdx)(exuIdx)(srcIdx) := intRfRdata(intRfRdataIdx)
296730cfbc0SXuan Hu//        } else {
297730cfbc0SXuan Hu//          // better for debug, should never assigned to other bundles
298730cfbc0SXuan Hu//          s1_intPregRData(iqIdx)(exuIdx)(srcIdx) := "hdead_beef_dead_beef".U
299730cfbc0SXuan Hu//        }
300730cfbc0SXuan Hu//        // need read vf reg
301730cfbc0SXuan Hu//        if (readDataCfgSet.intersect(VfRegSrcDataSet).nonEmpty) {
302730cfbc0SXuan Hu//          s1_vfPregRData(iqIdx)(exuIdx)(srcIdx) := vfRfRdata(vfRfRdataIdx)
303730cfbc0SXuan Hu//          vfRfRdataIdx += 1
304730cfbc0SXuan Hu//        } else {
305730cfbc0SXuan Hu//          // better for debug, should never assigned to other bundles
306730cfbc0SXuan Hu//          s1_vfPregRData(iqIdx)(exuIdx)(srcIdx) := "hdead_beef_dead_beef_dead_beef_dead_beef".U
307730cfbc0SXuan Hu//        }
308730cfbc0SXuan Hu//      }
309730cfbc0SXuan Hu//    }
310730cfbc0SXuan Hu//  }
311730cfbc0SXuan Hu//
312730cfbc0SXuan Hu//  println(s"[DataPath] assigned RegFile Rdata: int(${intRfRdataIdx}), vf(${vfRfRdataIdx})")
313730cfbc0SXuan Hu
314730cfbc0SXuan Hu  for (i <- fromIQ.indices) {
315730cfbc0SXuan Hu    for (j <- fromIQ(i).indices) {
316730cfbc0SXuan Hu      // IQ(s0) --[Ctrl]--> s1Reg ---------- begin
317730cfbc0SXuan Hu      // refs
318730cfbc0SXuan Hu      val s1_valid = s1_toExuValid(i)(j)
319730cfbc0SXuan Hu      val s1_ready = s1_toExuReady(i)(j)
320730cfbc0SXuan Hu      val s1_data = s1_toExuData(i)(j)
321730cfbc0SXuan Hu      val s1_addrOH = s1_addrOHs(i)(j)
322730cfbc0SXuan Hu      val s0 = fromIQ(i)(j) // s0
323730cfbc0SXuan Hu      val block = intBlocks(i)(j) || vfBlocks(i)(j)
324730cfbc0SXuan Hu      val s1_flush = s0.bits.common.robIdx.needFlush(Seq(io.flush, RegNextWithEnable(io.flush)))
325730cfbc0SXuan Hu      when (s0.fire && !s1_flush && !block) {
326730cfbc0SXuan Hu        s1_valid := s0.valid
327730cfbc0SXuan Hu        s1_data.fromIssueBundle(s0.bits) // no src data here
328730cfbc0SXuan Hu        s1_addrOH := s0.bits.addrOH
329730cfbc0SXuan Hu      }.otherwise {
330730cfbc0SXuan Hu        s1_valid := false.B
331730cfbc0SXuan Hu      }
332730cfbc0SXuan Hu
333730cfbc0SXuan Hu      s0.ready := (s1_ready || !s1_valid) && !block
334730cfbc0SXuan Hu      // IQ(s0) --[Ctrl]--> s1Reg ---------- end
335730cfbc0SXuan Hu
336730cfbc0SXuan Hu      // IQ(s0) --[Data]--> s1Reg ---------- begin
337730cfbc0SXuan Hu      // imm extract
338730cfbc0SXuan Hu      when (s0.fire && !s1_flush && !block) {
339730cfbc0SXuan Hu        if (s1_data.params.immType.nonEmpty && s1_data.src.size > 1) {
340730cfbc0SXuan Hu          // rs1 is always int reg, rs2 may be imm
341730cfbc0SXuan Hu          when(SrcType.isImm(s0.bits.srcType(1))) {
342730cfbc0SXuan Hu            s1_data.src(1) := ImmExtractor(
343730cfbc0SXuan Hu              s0.bits.common.imm,
344730cfbc0SXuan Hu              s0.bits.immType,
345730cfbc0SXuan Hu              s1_data.DataBits,
346730cfbc0SXuan Hu              s1_data.params.immType.map(_.litValue)
347730cfbc0SXuan Hu            )
348730cfbc0SXuan Hu          }
349730cfbc0SXuan Hu        }
350730cfbc0SXuan Hu        if (s1_data.params.hasJmpFu) {
351730cfbc0SXuan Hu          when(SrcType.isPc(s0.bits.srcType(0))) {
352730cfbc0SXuan Hu            s1_data.src(0) := SignExt(s0.bits.jmp.get.pc, XLEN)
353730cfbc0SXuan Hu          }
354730cfbc0SXuan Hu        }
355730cfbc0SXuan Hu      }
356730cfbc0SXuan Hu      // IQ(s0) --[Data]--> s1Reg ---------- end
357730cfbc0SXuan Hu    }
358730cfbc0SXuan Hu  }
359730cfbc0SXuan Hu
360ea0f92d8Sczw  private val fromIQFire = fromIQ.map(_.map(_.fire))
361ea0f92d8Sczw  private val toExuFire = toExu.map(_.map(_.fire))
362730cfbc0SXuan Hu  toIQs.zipWithIndex.foreach {
363730cfbc0SXuan Hu    case(toIQ, iqIdx) =>
364730cfbc0SXuan Hu      toIQ.zipWithIndex.foreach {
365730cfbc0SXuan Hu        case (toIU, iuIdx) =>
366730cfbc0SXuan Hu          // IU: issue unit
367730cfbc0SXuan Hu          val og0resp = toIU.og0resp
368ea0f92d8Sczw          og0resp.valid := fromIQ(iqIdx)(iuIdx).valid && (!fromIQFire(iqIdx)(iuIdx))
369ea0f92d8Sczw          og0resp.bits.respType := RSFeedbackType.rfArbitFail
370730cfbc0SXuan Hu          og0resp.bits.success := false.B
371730cfbc0SXuan Hu          og0resp.bits.addrOH := fromIQ(iqIdx)(iuIdx).bits.addrOH
372730cfbc0SXuan Hu
373730cfbc0SXuan Hu          val og1resp = toIU.og1resp
374730cfbc0SXuan Hu          og1resp.valid := s1_toExuValid(iqIdx)(iuIdx)
375ea0f92d8Sczw          og1resp.bits.respType := Mux(toExuFire(iqIdx)(iuIdx), RSFeedbackType.fuIdle, RSFeedbackType.fuBusy)
376730cfbc0SXuan Hu          og1resp.bits.success := false.B
377730cfbc0SXuan Hu          og1resp.bits.addrOH := s1_addrOHs(iqIdx)(iuIdx)
378730cfbc0SXuan Hu      }
379730cfbc0SXuan Hu  }
3808a00ff56SXuan Hu
381730cfbc0SXuan Hu  for (i <- toExu.indices) {
382730cfbc0SXuan Hu    for (j <- toExu(i).indices) {
383730cfbc0SXuan Hu      // s1Reg --[Ctrl]--> exu(s1) ---------- begin
384730cfbc0SXuan Hu      // refs
385730cfbc0SXuan Hu      val sinkData = toExu(i)(j).bits
386730cfbc0SXuan Hu      // assign
387730cfbc0SXuan Hu      toExu(i)(j).valid := s1_toExuValid(i)(j)
388730cfbc0SXuan Hu      s1_toExuReady(i)(j) := toExu(i)(j).ready
389730cfbc0SXuan Hu      sinkData := s1_toExuData(i)(j)
390730cfbc0SXuan Hu      // s1Reg --[Ctrl]--> exu(s1) ---------- end
391730cfbc0SXuan Hu
392730cfbc0SXuan Hu      // s1Reg --[Data]--> exu(s1) ---------- begin
393730cfbc0SXuan Hu      // data source1: preg read data
394730cfbc0SXuan Hu      for (k <- sinkData.src.indices) {
395730cfbc0SXuan Hu        val srcDataTypeSet: Set[DataConfig] = sinkData.params.getSrcDataType(k)
396730cfbc0SXuan Hu
397730cfbc0SXuan Hu        val readRfMap: Seq[(Bool, UInt)] = (Seq(None) :+
398730cfbc0SXuan Hu          (if (s1_intPregRData(i)(j).isDefinedAt(k) && srcDataTypeSet.intersect(IntRegSrcDataSet).nonEmpty)
399730cfbc0SXuan Hu            Some(SrcType.isXp(s1_srcType(i)(j)(k)) -> s1_intPregRData(i)(j)(k))
400730cfbc0SXuan Hu          else None) :+
401730cfbc0SXuan Hu          (if (s1_vfPregRData(i)(j).isDefinedAt(k) && srcDataTypeSet.intersect(VfRegSrcDataSet).nonEmpty)
402730cfbc0SXuan Hu            Some(SrcType.isVfp(s1_srcType(i)(j)(k))-> s1_vfPregRData(i)(j)(k))
403730cfbc0SXuan Hu          else None)
404730cfbc0SXuan Hu        ).filter(_.nonEmpty).map(_.get)
405730cfbc0SXuan Hu        if (readRfMap.nonEmpty)
406730cfbc0SXuan Hu          sinkData.src(k) := Mux1H(readRfMap)
407730cfbc0SXuan Hu      }
408730cfbc0SXuan Hu
409730cfbc0SXuan Hu      // data source2: extracted imm and pc saved in s1Reg
410730cfbc0SXuan Hu      if (sinkData.params.immType.nonEmpty && sinkData.src.size > 1) {
411730cfbc0SXuan Hu        when(SrcType.isImm(s1_srcType(i)(j)(1))) {
412730cfbc0SXuan Hu          sinkData.src(1) := s1_toExuData(i)(j).src(1)
413730cfbc0SXuan Hu        }
414730cfbc0SXuan Hu      }
415730cfbc0SXuan Hu      if (sinkData.params.hasJmpFu) {
416730cfbc0SXuan Hu        when(SrcType.isPc(s1_srcType(i)(j)(0))) {
417730cfbc0SXuan Hu          sinkData.src(0) := s1_toExuData(i)(j).src(0)
418730cfbc0SXuan Hu        }
419730cfbc0SXuan Hu      }
420730cfbc0SXuan Hu      // s1Reg --[Data]--> exu(s1) ---------- end
421730cfbc0SXuan Hu    }
422730cfbc0SXuan Hu  }
423730cfbc0SXuan Hu
424730cfbc0SXuan Hu  if (env.AlwaysBasicDiff || env.EnableDifftest) {
425730cfbc0SXuan Hu    val delayedCnt = 2
426730cfbc0SXuan Hu    val difftestArchIntRegState = Module(new DifftestArchIntRegState)
427730cfbc0SXuan Hu    difftestArchIntRegState.io.clock := clock
428730cfbc0SXuan Hu    difftestArchIntRegState.io.coreid := io.hartId
429730cfbc0SXuan Hu    difftestArchIntRegState.io.gpr := DelayN(intDebugRead.get._2, delayedCnt)
430730cfbc0SXuan Hu
431730cfbc0SXuan Hu    val difftestArchFpRegState = Module(new DifftestArchFpRegState)
432730cfbc0SXuan Hu    difftestArchFpRegState.io.clock := clock
433730cfbc0SXuan Hu    difftestArchFpRegState.io.coreid := io.hartId
434730cfbc0SXuan Hu    difftestArchFpRegState.io.fpr := DelayN(fpDebugReadData.get, delayedCnt)
435730cfbc0SXuan Hu
436730cfbc0SXuan Hu    val difftestArchVecRegState = Module(new DifftestArchVecRegState)
437730cfbc0SXuan Hu    difftestArchVecRegState.io.clock := clock
438730cfbc0SXuan Hu    difftestArchVecRegState.io.coreid := io.hartId
439730cfbc0SXuan Hu    difftestArchVecRegState.io.vpr := DelayN(vecDebugReadData.get, delayedCnt)
440730cfbc0SXuan Hu  }
441730cfbc0SXuan Hu}
442730cfbc0SXuan Hu
443730cfbc0SXuan Huclass DataPathIO()(implicit p: Parameters, params: BackendParams) extends XSBundle {
444730cfbc0SXuan Hu  // params
445730cfbc0SXuan Hu  private val intSchdParams = params.schdParams(IntScheduler())
446730cfbc0SXuan Hu  private val vfSchdParams = params.schdParams(VfScheduler())
447730cfbc0SXuan Hu  private val memSchdParams = params.schdParams(MemScheduler())
448730cfbc0SXuan Hu  // bundles
449730cfbc0SXuan Hu  val hartId = Input(UInt(8.W))
450730cfbc0SXuan Hu
451730cfbc0SXuan Hu  val flush: ValidIO[Redirect] = Flipped(ValidIO(new Redirect))
452730cfbc0SXuan Hu
453*d91483a6Sfdy  val vconfigReadPort = new RfReadPort(XLEN, PhyRegIdxWidth)
454*d91483a6Sfdy
455730cfbc0SXuan Hu  val fromIntIQ: MixedVec[MixedVec[DecoupledIO[IssueQueueIssueBundle]]] =
456730cfbc0SXuan Hu    Flipped(MixedVec(intSchdParams.issueBlockParams.map(_.genIssueDecoupledBundle)))
457730cfbc0SXuan Hu
458730cfbc0SXuan Hu  val fromMemIQ: MixedVec[MixedVec[DecoupledIO[IssueQueueIssueBundle]]] =
459730cfbc0SXuan Hu    Flipped(MixedVec(memSchdParams.issueBlockParams.map(_.genIssueDecoupledBundle)))
460730cfbc0SXuan Hu
461730cfbc0SXuan Hu  val fromVfIQ = Flipped(MixedVec(vfSchdParams.issueBlockParams.map(_.genIssueDecoupledBundle)))
462730cfbc0SXuan Hu
463730cfbc0SXuan Hu  val toIntIQ = MixedVec(intSchdParams.issueBlockParams.map(_.genOGRespBundle))
464730cfbc0SXuan Hu
465730cfbc0SXuan Hu  val toMemIQ = MixedVec(memSchdParams.issueBlockParams.map(_.genOGRespBundle))
466730cfbc0SXuan Hu
467730cfbc0SXuan Hu  val toVfIQ = MixedVec(vfSchdParams.issueBlockParams.map(_.genOGRespBundle))
468730cfbc0SXuan Hu
469730cfbc0SXuan Hu  val toIntExu: MixedVec[MixedVec[DecoupledIO[ExuInput]]] = intSchdParams.genExuInputBundle
470730cfbc0SXuan Hu
471730cfbc0SXuan Hu  val toFpExu: MixedVec[MixedVec[DecoupledIO[ExuInput]]] = MixedVec(vfSchdParams.genExuInputBundle)
472730cfbc0SXuan Hu
473730cfbc0SXuan Hu  val toMemExu: MixedVec[MixedVec[DecoupledIO[ExuInput]]] = memSchdParams.genExuInputBundle
474730cfbc0SXuan Hu
475730cfbc0SXuan Hu  val fromIntWb: MixedVec[RfWritePortWithConfig] = MixedVec(params.genIntWriteBackBundle)
476730cfbc0SXuan Hu
477730cfbc0SXuan Hu  val fromVfWb: MixedVec[RfWritePortWithConfig] = MixedVec(params.genVfWriteBackBundle)
478730cfbc0SXuan Hu
479730cfbc0SXuan Hu  val debugIntRat = Input(Vec(32, UInt(intSchdParams.pregIdxWidth.W)))
480730cfbc0SXuan Hu  val debugFpRat = Input(Vec(32, UInt(vfSchdParams.pregIdxWidth.W)))
481730cfbc0SXuan Hu  val debugVecRat = Input(Vec(32, UInt(vfSchdParams.pregIdxWidth.W)))
482730cfbc0SXuan Hu}
483