File tree Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Original file line number Diff line number Diff line change 1818# SBATCH --qos=${quality_of_service}
1919% endif
2020% if  gpu:
21- # SBATCH -G${tasks_per_node}
21+ # SBATCH --gres=gpu:V100:${tasks_per_node}
22+ # SBATCH --mem-per-gpu=16G\
2223% endif
2324% if  email:
2425# SBATCH --mail-user=${email}
4041    % if  not mpi:
4142        (set -x;  ${profiler}  " ${target.get_install_binpath(case)} "  )
4243    % else:
43-          (set -x;  srun \
44-             % if  engine == ' interactive'  :
45-                 --nodes ${nodes}  --ntasks-per-node ${tasks_per_node}  \
46-                 --cpus-per-task 1                                    \
47-                 % if  gpu:
48-                     --gpus-per-task 1 --gpu-bind closest                 \
49-                 % endif
50-             % endif
51-             ${profiler}  " ${target.get_install_binpath(case)} "  )
44+         (set -x;  ${profiler}     \
45+             mpirun -np ${nodes* tasks_per_node}             \
46+                    --bind-to none                         \
47+                    " ${target.get_install_binpath(case)} "  )
5248    % endif
5349
5450    ${helpers.run_epilogue(target)} 
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments