Support needUpdate and defaultUpdate for DifftestBaseBundle (OpenXian…

…gShan#262)

The valid bits are actually defined in the base bundle.
Thus, we move them from DifftestBundle to DifftestBaseBundle.

src/main/scala/Bundles.scala

@@ -38,6 +38,14 @@ sealed trait DifftestBaseBundle extends Bundle {
     }
   }
 
+  def needUpdate: Option[Bool] = if (hasValid) Some(getValid) else None
+  def defaultUpdate(): DifftestBaseBundle = {
+    if (hasValid) {
+      getValid := false.B
+    }
+    this
+  }
+
   def hasAddress: Boolean = this.isInstanceOf[HasAddress]
   def getNumElements: Int = {
     this match {
@@ -89,6 +97,13 @@ class TrapEvent extends DifftestBaseBundle {
 
   val code = UInt(3.W)
   val pc = UInt(64.W)
+
+  override def needUpdate: Option[Bool] = Some(hasTrap || hasWFI)
+  override def defaultUpdate(): TrapEvent = {
+    hasTrap := false.B
+    hasWFI := false.B
+    this
+  }
 }
 
 class CSRState extends DifftestBaseBundle {

src/main/scala/Difftest.scala

@@ -56,8 +56,6 @@ sealed trait DifftestBundle extends Bundle with DifftestWithCoreid { this: Difft
     }
   }
 
-  def needUpdate: Option[Bool] = if (hasValid) Some(getValid) else None
-
   protected val needFlatten: Boolean = false
   def isFlatten: Boolean = hasAddress && this.needFlatten
 
@@ -142,7 +140,6 @@ class DiffInstrCommit(nPhyRegs: Int = 32) extends InstrCommit(nPhyRegs)
 
 class DiffTrapEvent extends TrapEvent with DifftestBundle {
   override val desiredCppName: String = "trap"
-  override def needUpdate: Option[Bool] = Some(hasTrap || hasWFI)
   override def supportsSquashBase: Bool = !hasTrap && !hasWFI
 }
 

src/main/scala/Gateway.scala

@@ -183,7 +183,7 @@ class GatewayEndpoint(signals: Seq[DifftestBundle], config: GatewayConfig) exten
 
   val global_enable = WireInit(true.B)
   if(config.hasGlobalEnable) {
-    global_enable := VecInit(out.filter(_.needUpdate.isDefined).map(_.needUpdate.get).toSeq).asUInt.orR
+    global_enable := VecInit(out.flatMap(_.bits.needUpdate).toSeq).asUInt.orR
   }
 
   val batch_data = Option.when(config.isBatch)(Mem(config.batchSize, out_pack.cloneType))

src/main/scala/Squash.scala

@@ -96,7 +96,7 @@ class SquashEndpoint(bundles: Seq[DifftestBundle], config: GatewayConfig) extend
     // ignore useless unsquashed data when hasGlobalEnable
     val needStore = WireInit(true.B)
     if (config.hasGlobalEnable) {
-      needStore := VecInit(in.filter(_.needUpdate.isDefined).map(_.needUpdate.get).toSeq).asUInt.orR
+      needStore := VecInit(in.flatMap(_.bits.needUpdate).toSeq).asUInt.orR
     }
     when ((should_tick || do_squash.asUInt.orR) && needStore && !control.replay.get) {
       replay_data(replay_ptr) := in