From 67e492426bccfd3e22a4101fdd2091d29bb3207b Mon Sep 17 00:00:00 2001
From: unlsycn <unlsycn@unlsycn.com>
Date: Mon, 2 Dec 2024 05:58:36 +0000
Subject: [PATCH] fix sram module name

---
 rocketv/src/SRAM.scala | 25 ++++++++++---------------
 t1/src/SRAM.scala      | 25 ++++++++++---------------
 2 files changed, 20 insertions(+), 30 deletions(-)

diff --git a/rocketv/src/SRAM.scala b/rocketv/src/SRAM.scala
index 29836be71..30e2fa076 100644
--- a/rocketv/src/SRAM.scala
+++ b/rocketv/src/SRAM.scala
@@ -322,8 +322,12 @@ object SRAM {
     val numReadPorts      = readPortClocks.size
     val numWritePorts     = writePortClocks.size
     val numReadwritePorts = readwritePortClocks.size
-    val isVecMem          = evidenceOpt.isDefined
+    val enableMask        = evidenceOpt.isDefined
     val isValidSRAM       = ((numReadPorts + numReadwritePorts) > 0) && ((numWritePorts + numReadwritePorts) > 0)
+    val maskGranularity   = tpe match {
+      case vec: Vec[_] if enableMask => vec.sample_element.getWidth
+      case _ => 0
+    }
 
     if (!isValidSRAM) {
       val badMemory =
@@ -339,16 +343,13 @@ object SRAM {
     val mem = Instantiate(
       new SRAMBlackbox(
         new CIRCTSRAMParameter(
-          s"sram_${size}x${tpe.getWidth}",
+          s"sram_${numReadPorts}R_${numWritePorts}W_${numReadwritePorts}RW_${maskGranularity}M_${size}x${tpe.getWidth}",
           numReadPorts,
           numWritePorts,
           numReadwritePorts,
           size.intValue,
           tpe.getWidth,
-          tpe match {
-            case vec: Vec[_] if isVecMem => vec.sample_element.getWidth
-            case _ => 0
-          }
+          maskGranularity
         )
       )
     )
@@ -360,7 +361,7 @@ object SRAM {
     val includeMetadata = Builder.includeUtilMetadata
 
     val out = Wire(
-      new SRAMInterface(size, tpe, numReadPorts, numWritePorts, numReadwritePorts, isVecMem, includeMetadata)
+      new SRAMInterface(size, tpe, numReadPorts, numWritePorts, numReadwritePorts, enableMask, includeMetadata)
     )
 
     out.readPorts.zip(sramReadPorts).zip(readPortClocks).map { case ((intfReadPort, sramReadPort), readClock) =>
@@ -397,17 +398,11 @@ object SRAM {
       val descriptionInstance: Instance[SRAMDescription] = Instantiate(new SRAMDescription)
       descriptionInstance.depthIn           := Property(size)
       descriptionInstance.widthIn           := Property(tpe.getWidth)
-      descriptionInstance.maskedIn          := Property(isVecMem)
+      descriptionInstance.maskedIn          := Property(enableMask)
       descriptionInstance.readIn            := Property(numReadPorts)
       descriptionInstance.writeIn           := Property(numWritePorts)
       descriptionInstance.readwriteIn       := Property(numReadwritePorts)
-      descriptionInstance.maskGranularityIn := Property(
-        Option
-          .when(isVecMem)(tpe match {
-            case t: Vec[_] => t.sample_element.getWidth
-          })
-          .getOrElse(0)
-      )
+      descriptionInstance.maskGranularityIn := Property(maskGranularity)
       descriptionInstance.hierarchyIn       := Property(Path(mem.toTarget))
       description                           := descriptionInstance.getPropertyReference
     }
diff --git a/t1/src/SRAM.scala b/t1/src/SRAM.scala
index 29836be71..30e2fa076 100644
--- a/t1/src/SRAM.scala
+++ b/t1/src/SRAM.scala
@@ -322,8 +322,12 @@ object SRAM {
     val numReadPorts      = readPortClocks.size
     val numWritePorts     = writePortClocks.size
     val numReadwritePorts = readwritePortClocks.size
-    val isVecMem          = evidenceOpt.isDefined
+    val enableMask        = evidenceOpt.isDefined
     val isValidSRAM       = ((numReadPorts + numReadwritePorts) > 0) && ((numWritePorts + numReadwritePorts) > 0)
+    val maskGranularity   = tpe match {
+      case vec: Vec[_] if enableMask => vec.sample_element.getWidth
+      case _ => 0
+    }
 
     if (!isValidSRAM) {
       val badMemory =
@@ -339,16 +343,13 @@ object SRAM {
     val mem = Instantiate(
       new SRAMBlackbox(
         new CIRCTSRAMParameter(
-          s"sram_${size}x${tpe.getWidth}",
+          s"sram_${numReadPorts}R_${numWritePorts}W_${numReadwritePorts}RW_${maskGranularity}M_${size}x${tpe.getWidth}",
           numReadPorts,
           numWritePorts,
           numReadwritePorts,
           size.intValue,
           tpe.getWidth,
-          tpe match {
-            case vec: Vec[_] if isVecMem => vec.sample_element.getWidth
-            case _ => 0
-          }
+          maskGranularity
         )
       )
     )
@@ -360,7 +361,7 @@ object SRAM {
     val includeMetadata = Builder.includeUtilMetadata
 
     val out = Wire(
-      new SRAMInterface(size, tpe, numReadPorts, numWritePorts, numReadwritePorts, isVecMem, includeMetadata)
+      new SRAMInterface(size, tpe, numReadPorts, numWritePorts, numReadwritePorts, enableMask, includeMetadata)
     )
 
     out.readPorts.zip(sramReadPorts).zip(readPortClocks).map { case ((intfReadPort, sramReadPort), readClock) =>
@@ -397,17 +398,11 @@ object SRAM {
       val descriptionInstance: Instance[SRAMDescription] = Instantiate(new SRAMDescription)
       descriptionInstance.depthIn           := Property(size)
       descriptionInstance.widthIn           := Property(tpe.getWidth)
-      descriptionInstance.maskedIn          := Property(isVecMem)
+      descriptionInstance.maskedIn          := Property(enableMask)
       descriptionInstance.readIn            := Property(numReadPorts)
       descriptionInstance.writeIn           := Property(numWritePorts)
       descriptionInstance.readwriteIn       := Property(numReadwritePorts)
-      descriptionInstance.maskGranularityIn := Property(
-        Option
-          .when(isVecMem)(tpe match {
-            case t: Vec[_] => t.sample_element.getWidth
-          })
-          .getOrElse(0)
-      )
+      descriptionInstance.maskGranularityIn := Property(maskGranularity)
       descriptionInstance.hierarchyIn       := Property(Path(mem.toTarget))
       description                           := descriptionInstance.getPropertyReference
     }