util: parallel_sim.sh: support gz and zstd suffix checkpoints

Change-Id: I8315594f40e3b152368fd382057d67eb40be1642
OpenXiangShan · Jul 29, 2024 · 1a93940 · 1a93940
1 parent 13142f9
commit 1a93940
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/util/xs_scripts/parallel_sim.sh b/util/xs_scripts/parallel_sim.sh
@@ -84,8 +84,15 @@ function prepare_env() {
     all_args=("$@")
     task=${all_args[0]}
     task_path=${all_args[1]}
-    suffix="zstd"
-    checkpoint=$(find -L $cpt_dir -wholename "*${task_path}*${suffix}" | head -n 1)
+    # 同时匹配 gz zstd 后缀
+    suffixes=("gz" "zstd")
+    checkpoint=""
+    for suffix in "${suffixes[@]}"; do
+        checkpoint=$(find -L $cpt_dir -wholename "*${task_path}*${suffixes}" | head -n 1)
+        if [ -n "$checkpoint" ]; then
+            break
+        fi
+    done
     echo $checkpoint
 
     export work_dir=$full_work_dir/$task