From 8c228a6a8987f6f052d18f9de61009ee57a17fba Mon Sep 17 00:00:00 2001
From: Richard Yan <yrh@berkeley.edu>
Date: Thu, 30 Jan 2025 23:59:06 -0800
Subject: [PATCH] sanity script, bump radiance

---
 generators/radiance                 |  2 +-
 sims/vcs/scripts/compile_designs.sh |  1 -
 sims/vcs/scripts/pprint             |  2 +-
 sims/vcs/scripts/run_sims.sh        |  4 +--
 sims/vcs/scripts/sanity.sh          | 43 +++++++++++++++++++++++++++++
 5 files changed, 47 insertions(+), 5 deletions(-)
 create mode 100755 sims/vcs/scripts/sanity.sh

diff --git a/generators/radiance b/generators/radiance
index 34c33278..4a0b1c05 160000
--- a/generators/radiance
+++ b/generators/radiance
@@ -1 +1 @@
-Subproject commit 34c33278d2e674bd3ddc11ff6e76f5649e4c075e
+Subproject commit 4a0b1c05cd13da417875642408b8121ab8dbc624
diff --git a/sims/vcs/scripts/compile_designs.sh b/sims/vcs/scripts/compile_designs.sh
index 75dccbba..583900ba 100755
--- a/sims/vcs/scripts/compile_designs.sh
+++ b/sims/vcs/scripts/compile_designs.sh
@@ -18,5 +18,4 @@ make CONFIG=VirgoHopperConfig
 make CONFIG=VirgoHopperConfig debug
 echo -e "\nCompiling flash designs"
 make CONFIG=VirgoFlashConfig
-make CONFIG=VirgoFlashConfig debug
 echo -e "\nCompilation completed"
diff --git a/sims/vcs/scripts/pprint b/sims/vcs/scripts/pprint
index e3c67fae..fce3bb0e 100755
--- a/sims/vcs/scripts/pprint
+++ b/sims/vcs/scripts/pprint
@@ -156,7 +156,7 @@ def main():
             print(translated_line, end='')
             sys.stdout.flush()
 
-    print("\033[s", end='')
+    print("\033[2J\033[s", end='')
     print("\033[" + str(lineno) + "H\033[2K\033[1m" + run_label, "DONE", "\033[0m", end='')
     print("\033[u", end='', flush=True)
 
diff --git a/sims/vcs/scripts/run_sims.sh b/sims/vcs/scripts/run_sims.sh
index c032bfe6..daa34a06 100755
--- a/sims/vcs/scripts/run_sims.sh
+++ b/sims/vcs/scripts/run_sims.sh
@@ -56,8 +56,8 @@ for dim in "${dims[@]}"; do
     suffix=""
 done
 
-start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.ampere.seqlen1024.headdim64.elf  "ampere" "${suffix}"
-start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.virgo.seqlen1024.headdim64.elf   "virgo"  "${suffix}"
+start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.ampere.seqlen1024.headdim64.elf  "flash ampere" ""
+start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.virgo.seqlen1024.headdim64.elf   "flash virgo " ""
 
 wait
 
diff --git a/sims/vcs/scripts/sanity.sh b/sims/vcs/scripts/sanity.sh
new file mode 100755
index 00000000..447d8835
--- /dev/null
+++ b/sims/vcs/scripts/sanity.sh
@@ -0,0 +1,43 @@
+#!/bin/bash
+
+set -e
+
+echoerr() { echo "$@" 1>&2; }
+
+CURRENT_DIR="${PWD##*/}"
+if [[ "$CURRENT_DIR" != "vcs" ]]; then
+    echoerr "Error: This script must be run from chipyard/sims/vcs."
+    exit 1
+fi
+
+source ./scripts/env.sh > /dev/null
+
+check_exists() {
+    if ! [ -f "$1" ]; then
+        echo "Error: looked for file $1 that does not exist."
+        exit 1
+    fi
+}
+
+echo "Checking if all kernels have been compiled.."
+
+dims=(256 512 1024)
+for dim in "${dims[@]}"; do
+    check_exists "$KERNELS_PATH/sgemm_tcore/kernel.radiance.gemm.tcore.volta.dim${dim}.elf"
+    check_exists "$KERNELS_PATH/sgemm_tcore/kernel.radiance.gemm.tcore.ampere.dim${dim}.elf"
+    check_exists "$KERNELS_PATH/sgemm_tcore/kernel.radiance.gemm.tcore.hopper.dim${dim}.elf"
+    check_exists "$KERNELS_PATH/sgemm_gemmini_dma/kernel.radiance.gemm.virgo.hopper.dim${dim}.elf"
+done
+
+check_exists "$KERNELS_PATH/flash_attention/kernel.radiance.flash.ampere.seqlen1024.headdim64.elf"
+check_exists "$KERNELS_PATH/flash_attention/kernel.radiance.flash.virgo.seqlen1024.headdim64.elf"
+
+echo "Checking if all simulation binaries have been compiled.."
+
+check_exists "simv-chipyard.harness-VirgoFP16Config"
+check_exists "simv-chipyard.harness-VirgoFP16Config-debug"
+check_exists "simv-chipyard.harness-VirgoHopperConfig"
+check_exists "simv-chipyard.harness-VirgoHopperConfig-debug"
+check_exists "simv-chipyard.harness-VirgoFlashConfig"
+
+echo "Sanity check passed!"