chn/nixos
1
0
Fork 0
mirror of https://github.com/CHN-beta/nixos.git synced 2026-01-12 04:39:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

packages.sbatch-tui: nvidia mpi bind to none

Browse Source
This commit is contained in:
陈浩南
2025-01-12 13:17:26 +08:00
parent 4cc5e63248
commit e053215d95
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/sbatch-tui/src/main.cpp
View File

@@ -223,12 +223,12 @@ int main()
if (state.gpu_scheme_entries[state.gpu_scheme_selected] == "any single GPU")
state.submit_command =
"sbatch --partition={}\n--ntasks=1 --cpus-per-gpu=1 --gpus=1\n--job-name='{}' --output='{}'\n"
"--wrap=\"vasp-nvidia mpirun vasp-{}\""_f
"--wrap=\"vasp-nvidia mpirun --bind-to none vasp-{}\""_f
(device.GpuPartition, state.job_name, state.output_file, state.vasp_entries[state.vasp_selected]);
else
state.submit_command =
"sbatch --partition={}\n--ntasks=1 --cpus-per-gpu=1 --gpus={}:1\n--job-name='{}' --output='{}'\n"
"--wrap=\"vasp-nvidia mpirun vasp-{}\""_f
"--wrap=\"vasp-nvidia mpirun --bind-to none vasp-{}\""_f
(
device.GpuPartition, state.gpu_entries[state.gpu_selected],
state.job_name, state.output_file, state.vasp_entries[state.vasp_selected]