From d592e92575827526993aa3bb0a6d39f565dceac3 Mon Sep 17 00:00:00 2001
From: Jordan Carlin <jordanmcarlin@gmail.com>
Date: Tue, 10 Dec 2024 15:53:04 -0800
Subject: [PATCH] Update platform.yaml

---
 sim/questa/wally.do                |  1 -
 testbench/trek_files/platform.yaml | 11 +++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/sim/questa/wally.do b/sim/questa/wally.do
index 925889350..6c7a422c4 100644
--- a/sim/questa/wally.do
+++ b/sim/questa/wally.do
@@ -140,7 +140,6 @@ if {[lcheck lst "--breker"]} {
                     +incdir+${WALLY}/testbench/trek_files \
                     ${WALLY}/testbench/trek_files/uvm_output/trek_uvm_pkg.sv"
     set brekervopt "${WKDIR}.trek_uvm"
-    # may need to change this path
     set brekervsim "+TREK_TBX_FILE=${WALLY}/tests/breker/work/${TESTSUITE_NO_ELF}/${TESTSUITE_NO_ELF}.tbx"
     append SVLib " -sv_lib ${BREKER_HOME}/linux64/lib/libtrek "
 }
diff --git a/testbench/trek_files/platform.yaml b/testbench/trek_files/platform.yaml
index 376a25857..7d262f4f0 100644
--- a/testbench/trek_files/platform.yaml
+++ b/testbench/trek_files/platform.yaml
@@ -159,6 +159,17 @@ trek:
       doc: >-
         Verbatim code that will be put into the declaration section of the test.
       value: |-
+        volatile uint64_t tohost;
+
+        void __attribute__((noreturn)) tohost_exit(uintptr_t code)
+        {
+          tohost = (code << 1) | 1;
+          while (1);
+        }
+
+        #undef trek_exit
+        #define trek_exit(status) tohost_exit(status);
+
         int main(void)
         {
           return trek_main();