В Slurm возможно ли назначить разное количество процессоров для каждой задачи?

#mpi #slurm

#mpi #slurm

Вопрос:

Я запускаю задания MPI поверх openmp в кластере Slurm и пытаюсь найти способ предоставить разное количество процессоров для каждой сгенерированной задачи. Например, предположим, что мы запускаем это задание:

srun --nodes 1 --ntasks 2 --cpus-per-task 2 ./mpi_exe

Это сгенерировало бы 2 процесса MPI в одном узле, по 2 процессора в каждом. Однако я хотел бы, например, назначить 3 процессора первому процессу и 1 во втором процессе.

Знаете ли вы какой-либо способ добиться этого?

Ответ №1:

Взгляните на разнородные задания. Для вашего примера это должно сработать:

 srun -N1 -n1 -c3 : -N1 -n1 -c1 ./mpi_exe