Ejecute programas de configuración de trabajadores en el script sbatch de SLURM.

Answer

1. 

   davy.ai

   June 5, 2023 at 9:58 pm

   Para ejecutar el código de configuración en un script SLURM sbatch, puede utilizar el comando srun, tal como se muestra en el fragmento de código. El primer comando srun copia los archivos de datos necesarios al directorio temporal ($TMPDIR) en el nodo donde se ejecutará el trabajo. El segundo comando srun ejecuta el trabajo real (a.out) utilizando los archivos de datos del directorio temporal.

   Sin embargo, usar dos líneas srun no garantiza que se ejecute en el mismo nodo y puede resultar en pérdida de datos. Para asegurarse de que el código de configuración y el trabajo se ejecuten en el mismo nodo, puede utilizar el indicador --exclusive con el comando salloc para asignar acceso exclusivo al nodo para el trabajo y luego utilizar un solo comando srun tanto para la configuración como para el trabajo.

   #!/bin/bash
   
   # Parámetros
   #SBATCH ...
   
   # asignar nodos
   salloc --exclusive
   
   # configuración y trabajo
   srun cp /nfs/data $TMPDIR && srun a.out $TMPDIR
   

   Esto asegurará que el código de configuración y el trabajo se ejecuten en el mismo nodo sin limpieza en el medio.

Comments are closed.