From e1c6318edb0dd355f9ad4e3296930e7bd67460b2 Mon Sep 17 00:00:00 2001
From: Derek Christ <christ.derek@gmail.com>
Date: Tue, 28 Nov 2023 20:22:55 +0100
Subject: [PATCH] Integrate additional pim-vm library in DRAMSys linking

---
 ext/dramsys/SConscript | 3 +++
 src/mem/dramsys.cc     | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/ext/dramsys/SConscript b/ext/dramsys/SConscript
index bfe4ee5ec1..d976768e64 100644
--- a/ext/dramsys/SConscript
+++ b/ext/dramsys/SConscript
@@ -76,6 +76,9 @@ subprocess.run(
 env.Append(LIBS="DRAMSys_libdramsys")
 env.Append(LIBPATH=Dir("./DRAMSys/src/libdramsys").abspath)
 
+env.Append(LIBS=["libpim_vm", "libpim-vm-cxx"])
+env.Append(LIBPATH=Dir("./DRAMSys").abspath)
+
 env.Append(LIBS="DRAMSys_Configuration")
 env.Append(LIBPATH=Dir("./DRAMSys/src/configuration").abspath)
 
diff --git a/src/mem/dramsys.cc b/src/mem/dramsys.cc
index d1b6e9c372..fb1b508dd2 100644
--- a/src/mem/dramsys.cc
+++ b/src/mem/dramsys.cc
@@ -27,6 +27,8 @@
  */
 
 #include "dramsys.hh"
+#include "DRAMSys/common/Deserialize.h"
+#include "DRAMSys/common/Serialize.h"
 
 namespace gem5
 {