From fd2f88ef33503313fbf7756df9c5d13cfc39229d Mon Sep 17 00:00:00 2001
From: Kunlin Han <hi@iwktd.com>
Date: Sat, 6 Apr 2024 17:06:06 -0700
Subject: [PATCH] Add fallback on site-setup.sh if no RISCV/site-setup.sh
 exists.

---
 setup.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/setup.sh b/setup.sh
index f609c1e3d..d8063b2aa 100644
--- a/setup.sh
+++ b/setup.sh
@@ -27,6 +27,10 @@ export PATH=$WALLY/bin:$PATH
 ulimit -s 100000
 
 # load site licenses and tool locations
-source $RISCV/site-setup.sh
+if [ -f ${RISCV}/site-setup.sh ]; then
+    source ${RISCV}/site-setup.sh
+else
+    source ${WALLY}/site-setup.sh
+fi
 
 echo "setup done"