From a098b7099f7652807e1713109a5943b087ee6d0a Mon Sep 17 00:00:00 2001
From: Derek Christ <christ.derek@gmail.com>
Date: Sat, 6 Jan 2024 17:41:10 +0100
Subject: [PATCH] Increase HBM2 memory size to 2 GiB

---
 ext/dramsys/SConscript                       | 2 +-
 src/python/gem5/components/memory/dramsys.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ext/dramsys/SConscript b/ext/dramsys/SConscript
index d5a690c2a1..e0d34434ef 100644
--- a/ext/dramsys/SConscript
+++ b/ext/dramsys/SConscript
@@ -50,7 +50,7 @@ subprocess.run(
         "-DCMAKE_BUILD_TYPE=Release",
         f"-DSCONS_SOURCE_DIR:STRING={scons_root}",
         "-DDRAMSYS_BUILD_CLI=OFF",
-        "-DDRAMSYS_SHARED_PIM_UNITS=ON"
+        "-DDRAMSYS_SHARED_PIM_UNITS=OFF"
     ],
     check=True
 )
diff --git a/src/python/gem5/components/memory/dramsys.py b/src/python/gem5/components/memory/dramsys.py
index a09d2fa0fc..4cdd5a8528 100644
--- a/src/python/gem5/components/memory/dramsys.py
+++ b/src/python/gem5/components/memory/dramsys.py
@@ -183,6 +183,6 @@ class DRAMSysHBM2(DRAMSysMem):
             configuration=(
                 DEFAULT_DRAMSYS_DIRECTORY / "configs/hbm2-example.json"
             ).as_posix(),
-            size="1GB",
+            size="2GB",
             recordable=recordable,
         )