如何让VASP 从 197 秒跑到 145 秒？

#!/usr/bin/env bash
#SBATCH --job-name=vasp-96r
#SBATCH --partition=YOUR_PARTITION
#SBATCH --nodes=1
#SBATCH --ntasks=96
#SBATCH --exclusive
#SBATCH --time=01:00:00

set -euo pipefail

# ===== 提交前修改这三处 =====
NRANKS=96
SOCKET_ID=0
VASP_LAUNCH=/path/to/site-validated-vasp-launch-command

# 单节点纯 MPI：每个 rank 使用一个线程
export OMP_NUM_THREADS=1
export MKL_NUM_THREADS=1
export MKL_DYNAMIC=FALSE

# 节点内共享内存通信，通过 Slurm PMI2 启动
export I_MPI_FABRICS=shm
export I_MPI_PMI=pmi2

# 关闭 Intel MPI 二次绑核，由 Slurm 统一负责
export I_MPI_PIN=0
export I_MPI_PIN_RESPECT_CPUSET=1

# 从指定 socket 中选择互不重复的物理核
CPU_MAP=$(
    lscpu -p=CPU,CORE,SOCKET,NODE |
    awk -F, -v socket="$SOCKET_ID" -v limit="$NRANKS" '
        $1 !~ /^#/ && $3 == socket && count < limit && !seen[$2]++ {
            print $1
            count++
        }
    ' |
    paste -sd, -
)

# CPU 数量必须与 MPI ranks 一致
MAP_COUNT=$(printf '%s\n' "$CPU_MAP" | awk -F, '{print NF}')

if [ "$MAP_COUNT" -ne "$NRANKS" ]; then
    echo "错误：socket ${SOCKET_ID} 只找到 ${MAP_COUNT} 个物理核，需要 ${NRANKS} 个。" >&2
    exit 1
fi

echo "使用 socket：${SOCKET_ID}"
echo "MPI ranks：${NRANKS}"
echo "CPU map：${CPU_MAP}"

unset SLURM_CPUS_PER_TASK

srun --mpi=pmi2 \
    --nodes=1 \
    --ntasks="$NRANKS" \
    --cpu-bind="verbose,map_cpu:${CPU_MAP}" \
    --mem-bind=verbose,local \
    --kill-on-bad-exit=1 \
    "$VASP_LAUNCH"