diff --git a/src/main/scala/diplomacy/SRAM.scala b/src/main/scala/diplomacy/SRAM.scala
index 73b6d269d1c..5bd79302454 100644
--- 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")