[t1rocket] fix wrong quit check

[t1rocket] add set scope for difftest quit

t1rocketemu/online_dpi/src/dpi.rs

@@ -330,7 +330,10 @@ pub(crate) fn dump_wave(scope: crate::svdpi::SvScope, path: &str) {
   }
 }
 
-pub(crate) fn quit() {
+pub(crate) fn quit(scope: crate::svdpi::SvScope) {
+  use crate::svdpi;
+
+  svdpi::set_scope(scope);
   unsafe {
     dpi_export::quit();
   }

t1rocketemu/online_dpi/src/drive.rs

@@ -1,7 +1,7 @@
 use crate::dpi::*;
-use crate::{ get_t, EXIT_CODE, EXIT_POS };
 use crate::svdpi::SvScope;
 use crate::OfflineArgs;
+use crate::{get_t, EXIT_CODE, EXIT_POS};
 
 use anyhow::Context;
 use common::MEM_SIZE;
@@ -336,17 +336,19 @@ impl Driver {
     let data_hex = hex::encode(data);
     self.last_commit_cycle = get_t();
 
-    // exit with code
-    if addr == EXIT_POS && data.len() == 4 && data == &EXIT_CODE.to_le_bytes() {
-      info!("exit successfully");
-      quit();
-      return;
-    }
-
     trace!(
       "[{}] axi_write_load_store (addr={addr:#x}, size={size}, data={data_hex})",
       get_t()
     );
+
+    // check exit with code
+    if addr == EXIT_POS {
+      let exit_data_slice = data[..4].try_into().expect("slice with incorrect length");
+      if u32::from_le_bytes(exit_data_slice) == EXIT_CODE {
+        info!("exit successfully");
+        self.quit()
+      }
+    }
   }
 
   pub(crate) fn axi_read_instruction_fetch(&mut self, addr: u32, arsize: u64) -> AxiReadPayload {
@@ -397,4 +399,8 @@ impl Driver {
   fn start_dump_wave(&mut self) {
     dump_wave(self.scope, &self.wave_path);
   }
+
+  fn quit(&mut self) {
+    quit(self.scope)
+  }
 }