From 6d5e6f7ae64a434b9773eb4fa387926c5043dded Mon Sep 17 00:00:00 2001
From: fangyuedong <fangyuedong@qxslab.cn>
Date: Thu, 25 Apr 2024 01:05:10 +0800
Subject: [PATCH] update submit script for Pukou machine

---
 config/config_overall.yaml |  2 +-
 run_C6.pbs                 | 27 +++++++++------------------
 2 files changed, 10 insertions(+), 19 deletions(-)

diff --git a/config/config_overall.yaml b/config/config_overall.yaml
index 8506e57..ca4fc9b 100644
--- a/config/config_overall.yaml
+++ b/config/config_overall.yaml
@@ -11,7 +11,7 @@
 # can add some of the command-line arguments here as well;
 # ok to pass either way or both, as long as they are consistent
 work_dir: "/public/home/fangyuedong/project/workplace/"
-run_name: "QSO_50sqdeg_test"
+run_name: "magnification_test"
 
 # Project cycle and run counter are used to name the outputs
 project_cycle: 9
diff --git a/run_C6.pbs b/run_C6.pbs
index 520645f..37b8515 100755
--- a/run_C6.pbs
+++ b/run_C6.pbs
@@ -1,24 +1,15 @@
 #!/bin/bash
-#PBS -N C6_TEST
-#PBS -l walltime=70:00:00
-#PBS -j oe
-##PBS -l nodes=wcl-1:ppn=16+wcl-2:ppn=16
-#PBS -q batch
-#PBS -u fangyuedong
 
-cd $PBS_O_WORKDIR
+#SBATCH -J CSSTSim
+#SBATCH -N 1
+#SBATCH --ntasks-per-node=40
+#SBATCH -p debug
+#SBATCH --mem=240G
 
-NP=384
-hostfile=wcl-1,wcl-2,wcl-3,wcl-4,wcl-5,wcl-6
+module load mpi/hpcx/2.4.1/gcc-7.3.1
 date
 
-mpirun --oversubscribe -H $hostfile -np $NP python /share/home/fangyuedong/new_sim/csst-simulation/run_sim.py \
-    --config_file config_C6.yaml \
-    --catalog C6_Catalog \
-    -c /share/home/fangyuedong/new_sim/csst-simulation/config
-
-    # --config_file config_50sqdeg.yaml \
-    # --catalog C6_50sqdeg \
-    # -c /share/home/fangyuedong/new_sim/csst-simulation/config
-
+#限定单节点任务数
+srun hostname -s | sort -n | awk -F"-" '{print $2}' | uniq > pnodes
 
+mpirun -mca pml ucx -x  UCX_NET_DEVICES=mlx5_0:1 -machinefile pnodes -np 40 --map-by node python3 /public/home/fangyuedong/project/csst_msc_sim/run_sim.py  --config_file config_overall.yaml --catalog C6_50sqdeg -c /public/home/fangyuedong/project/csst_msc_sim/config
\ No newline at end of file
-- 
GitLab