Support devOverride for diplomatic SRAMs

chipsalliance · Mar 29, 2023 · 12e21a6 · 12e21a6
1 parent f5ebf26
commit 12e21a6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/scala/diplomacy/SRAM.scala b/src/main/scala/diplomacy/SRAM.scala
@@ -11,11 +11,13 @@ abstract class DiplomaticSRAM(
     val address: AddressSet,
     beatBytes: Int,
     devName: Option[String],
-    dtsCompat: Option[Seq[String]] = None)(implicit p: Parameters) extends LazyModule
+    dtsCompat: Option[Seq[String]] = None,
+    devOverride: Option[Device with DeviceRegName] = None)(implicit p: Parameters) extends LazyModule
 {
-  val device = devName
+  val device = devOverride.getOrElse(devName
     .map(new SimpleDevice(_, dtsCompat.getOrElse(Seq("sifive,sram0"))))
     .getOrElse(new MemoryDevice())
+  )
 
   val resources = device.reg("mem")