ram: overwrite gcpt

OpenXiangShan · Mar 22, 2024 · d496ffc · d496ffc
1 parent f8df21a
commit d496ffc
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 4 deletions.
diff --git a/src/test/csrc/common/ram.cpp b/src/test/csrc/common/ram.cpp
@@ -432,6 +432,13 @@ void LinearizedFootprintsMemory::save_linear_memory(const char *filename) {
   out_file.close();
 }
 
+void overwrite_ram(char *gcpt_restore, uint64_t overwrite_nbytes) {
+  InputReader *reader = new FileReader(gcpt_restore);
+  int overwrite_size = reader->read_all(simMemory->as_ptr(), overwrite_nbytes);
+  Info("Overwrite %d bytes from file %s.\n", overwrite_size,gcpt_restore);
+  delete reader;
+}
+
 #ifdef WITH_DRAMSIM3
 void dramsim3_init() {
 #if !defined(DRAMSIM3_CONFIG) || !defined(DRAMSIM3_OUTDIR)

diff --git a/src/test/csrc/common/ram.h b/src/test/csrc/common/ram.h
@@ -214,6 +214,7 @@ class LinearizedFootprintsMemory : public FootprintsMemory {
 extern SimMemory *simMemory;
 // This is to initialize the common mmap RAM
 void init_ram(const char *image, uint64_t n_bytes);
+void overwrite_ram(char *gcpt_restore, uint64_t overwrite_nbytes);
 
 #ifdef WITH_DRAMSIM3
 

diff --git a/src/test/csrc/vcs/vcs_main.cpp b/src/test/csrc/vcs/vcs_main.cpp
@@ -32,9 +32,11 @@
 static bool has_reset = false;
 static char bin_file[256] = "/dev/zero";
 static char *flash_bin_file = NULL;
+static char *gcpt_bin_file = NULL;
 static bool enable_difftest = true;
 static uint64_t max_instrs = 0;
 static char *workload_list = NULL;
+static uint64_t overwrite_nbytes = 0xe00;
 
 enum {
   SIMV_RUN,
@@ -53,6 +55,11 @@ extern "C" void set_flash_bin(char *s) {
   strcpy(flash_bin_file, s);
 }
 
+extern "C" void set_gcpt_bin_file(char *s) {
+  gcpt_bin_file = (char *)malloc(256);
+  strcpy(gcpt_bin_file, s);
+}
+
 extern "C" void set_max_instrs(uint64_t mc) {
   printf("set max instrs: %lu\n", mc);
   max_instrs = mc;
@@ -117,6 +124,9 @@ extern "C" uint8_t simv_init() {
     init_goldenmem();
     init_nemuproxy(DEFAULT_EMU_RAM_SIZE);
   }
+  if (gcpt_bin_file != NULL) {
+    overwrite_ram(gcpt_bin_file, overwrite_nbytes);
+  }
   return 0;
 }
 

diff --git a/src/test/csrc/verilator/emu.cpp b/src/test/csrc/verilator/emu.cpp
@@ -382,10 +382,7 @@ Emulator::Emulator(int argc, const char *argv[])
   }
 
   if (args.gcpt_restore) {
-    InputReader *reader = new FileReader(args.gcpt_restore);
-    int overwrite_size = reader->read_all(simMemory->as_ptr(), args.overwrite_nbytes);
-    Info("Overwrite %d bytes from file %s.\n", overwrite_size, args.gcpt_restore);
-    delete reader;
+    overwrite_ram(args.gcpt_restore, args.overwrite_nbytes);
   }
 
 #ifdef ENABLE_CHISEL_DB