Skip to content

Commit ceb404b

Browse files
committed
update scripts
1 parent 985dca9 commit ceb404b

File tree

4 files changed

+86
-57
lines changed

4 files changed

+86
-57
lines changed

.gitignore

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,11 @@ data
127127
results
128128

129129
third_party
130-
gscodec
130+
# gscodec
131131
setup_gscodec.py
132+
Readme_GSCodec.md
133+
134+
figs
135+
stats
132136

133137
!examples/benchmarks/compression/results/

examples/benchmarks/compression/final_exp/mcmc_mip_sim.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ CAP_MAX=1000000
2222

2323
RD_LAMBDA=0.01
2424

25-
GPU_LIST="2 3 4 5"
25+
GPU_LIST="0 1 2 3 4"
2626
# ------------------------- Settings ------------------------- #
2727

2828
# ------------------------- Args ----------------------------- #

examples/benchmarks/compression/final_exp/mcmc_tt_sim.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ run_single_scene() {
6666

6767

6868
# ----------------- Experiment Loop -------------- #
69-
GPU_LIST=(0 1)
69+
GPU_LIST=(5 6)
7070
GPU_COUNT=${#GPU_LIST[@]}
7171

7272
SCENE_IDX=-1
Lines changed: 79 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -1,58 +1,83 @@
1-
RD_LAMBDA_LIST="0.0025 0.005 0.01 0.02 0.04"
1+
#!/bin/bash
22

3-
# TT -> (0,1)
4-
# # 检查脚本是否存在且可执行
5-
if [ ! -x "benchmarks/compression/final_exp/mcmc_tt_sim.sh" ]; then
6-
echo "Error: mcmc_tt_sim.sh not found or not executable"
3+
RD_LAMBDA_LIST="0.002 0.004 0.0075 0.006 0.008"
4+
5+
# 定义TT执行函数
6+
run_tt() {
7+
if [ ! -x "benchmarks/compression/final_exp/mcmc_tt_sim.sh" ]; then
8+
echo "Error: mcmc_tt_sim.sh not found or not executable"
9+
return 1
10+
fi
11+
12+
for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
13+
echo "Processing TT with RD_LAMBDA = ${RD_LAMBDA}"
14+
benchmarks/compression/final_exp/mcmc_tt_sim.sh "${RD_LAMBDA}"
15+
16+
if [ $? -ne 0 ]; then
17+
echo "Error occurred in TT with RD_LAMBDA = ${RD_LAMBDA}"
18+
return 1
19+
fi
20+
done
21+
}
22+
23+
# 定义MIP执行函数
24+
run_mip() {
25+
if [ ! -x "benchmarks/compression/final_exp/mcmc_mip_sim.sh" ]; then
26+
echo "Error: mcmc_mip_sim.sh not found or not executable"
27+
return 1
28+
fi
29+
30+
for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
31+
echo "Processing MIP with RD_LAMBDA = ${RD_LAMBDA}"
32+
benchmarks/compression/final_exp/mcmc_mip_sim.sh "${RD_LAMBDA}"
33+
34+
if [ $? -ne 0 ]; then
35+
echo "Error occurred in MIP with RD_LAMBDA = ${RD_LAMBDA}"
36+
return 1
37+
fi
38+
done
39+
}
40+
41+
# 定义DB执行函数
42+
run_db() {
43+
if [ ! -x "benchmarks/compression/final_exp/mcmc_db_sim.sh" ]; then
44+
echo "Error: mcmc_db_sim.sh not found or not executable"
45+
return 1
46+
fi
47+
48+
for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
49+
echo "Processing DB with RD_LAMBDA = ${RD_LAMBDA}"
50+
benchmarks/compression/final_exp/mcmc_db_sim.sh "${RD_LAMBDA}"
51+
52+
if [ $? -ne 0 ]; then
53+
echo "Error occurred in DB with RD_LAMBDA = ${RD_LAMBDA}"
54+
return 1
55+
fi
56+
done
57+
}
58+
59+
# 并行执行三个函数
60+
run_tt &
61+
tt_pid=$!
62+
63+
# run_mip &
64+
# mip_pid=$!
65+
66+
# run_db &
67+
# db_pid=$!
68+
69+
# # 等待所有进程完成
70+
wait $tt_pid
71+
tt_status=$?
72+
# wait $mip_pid
73+
# mip_status=$?
74+
# wait $db_pid
75+
# db_status=$?
76+
77+
# # 检查是否所有进程都成功完成
78+
if [ $tt_status -ne 0 ] || [ $mip_status -ne 0 ] || [ $db_status -ne 0 ]; then
79+
echo "One or more processes failed"
780
exit 1
881
fi
982

10-
for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
11-
echo "Processing RD_LAMBDA = ${RD_LAMBDA}"
12-
benchmarks/compression/final_exp/mcmc_tt_sim.sh "${RD_LAMBDA}"
13-
14-
# 检查上一个命令是否成功执行
15-
if [ $? -ne 0 ]; then
16-
echo "Error occurred with RD_LAMBDA = ${RD_LAMBDA}"
17-
# 根据需要决定是继续还是退出
18-
# exit 1
19-
fi
20-
done
21-
22-
# # MIP -> (2,3,4,5)
23-
# # 检查脚本是否存在且可执行
24-
# if [ ! -x "benchmarks/compression/final_exp/mcmc_mip_sim.sh" ]; then
25-
# echo "Error: mcmc_mip_sim.sh not found or not executable"
26-
# exit 1
27-
# fi
28-
29-
# for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
30-
# echo "Processing RD_LAMBDA = ${RD_LAMBDA}"
31-
# benchmarks/compression/final_exp/mcmc_mip_sim.sh "${RD_LAMBDA}"
32-
33-
# # 检查上一个命令是否成功执行
34-
# if [ $? -ne 0 ]; then
35-
# echo "Error occurred with RD_LAMBDA = ${RD_LAMBDA}"
36-
# # 根据需要决定是继续还是退出
37-
# # exit 1
38-
# fi
39-
# done
40-
41-
# DB -> (6,7)
42-
# 检查脚本是否存在且可执行
43-
# if [ ! -x "benchmarks/compression/final_exp/mcmc_db_sim.sh" ]; then
44-
# echo "Error: mcmc_db_sim.sh not found or not executable"
45-
# exit 1
46-
# fi
47-
48-
# for RD_LAMBDA in ${RD_LAMBDA_LIST}; do
49-
# echo "Processing RD_LAMBDA = ${RD_LAMBDA}"
50-
# benchmarks/compression/final_exp/mcmc_db_sim.sh "${RD_LAMBDA}"
51-
52-
# # 检查上一个命令是否成功执行
53-
# if [ $? -ne 0 ]; then
54-
# echo "Error occurred with RD_LAMBDA = ${RD_LAMBDA}"
55-
# # 根据需要决定是继续还是退出
56-
# # exit 1
57-
# fi
58-
# done
83+
echo "All processes completed successfully"

0 commit comments

Comments
 (0)