submit_jobs.slurm

#!/bin/bash

#SBATCH -J CSSTSim
#SBATCH -N 1
#SBATCH --ntasks-per-node=6
#SBATCH -p debug
#SBATCH --mem=60G

module load mpi/hpcx/2.4.1/gcc-7.3.1
date

#限定单节点任务数
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 6 --map-by node python3 /public/home/fangyuedong/project/csst_msc_sim/run_sim.py  --config_file config_overall.yaml --catalog C9_Catalog -c /public/home/fangyuedong/project/csst_msc_sim/config