From ab86b9a3101469ed8671b7e1868539271b8026b8 Mon Sep 17 00:00:00 2001
From: unlsycn <unlsycn@unlsycn.com>
Date: Mon, 2 Dec 2024 04:52:13 +0000
Subject: [PATCH] fix mask granularity

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

diff --git a/rocketv/src/SRAM.scala b/rocketv/src/SRAM.scala
index efef671c9..29836be71 100644
--- a/rocketv/src/SRAM.scala
+++ b/rocketv/src/SRAM.scala
@@ -346,7 +346,7 @@ object SRAM {
           size.intValue,
           tpe.getWidth,
           tpe match {
-            case vec: Vec[_] => vec.size
+            case vec: Vec[_] if isVecMem => vec.sample_element.getWidth
             case _ => 0
           }
         )
diff --git a/t1/src/SRAM.scala b/t1/src/SRAM.scala
index efef671c9..29836be71 100644
--- a/t1/src/SRAM.scala
+++ b/t1/src/SRAM.scala
@@ -346,7 +346,7 @@ object SRAM {
           size.intValue,
           tpe.getWidth,
           tpe match {
-            case vec: Vec[_] => vec.size
+            case vec: Vec[_] if isVecMem => vec.sample_element.getWidth
             case _ => 0
           }
         )