fix: Raise error on non-homogeneous MPI requests

glotzerlab · Nov 8, 2023 · 5099399 · 5099399
1 parent ebdd1cd
commit 5099399
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flow/templates/umich-greatlakes.sh b/flow/templates/umich-greatlakes.sh
@@ -1,5 +1,8 @@
 {% extends "slurm.sh" %}
 {% set partition = partition|default('standard', true) %}
+{% if not (force or operations|homogeneous_openmp_mpi_config) %}
+    {% raise "Can only submit jobs with identical nranks and omp_num_threads" %}
+{% endif %}
 {% block tasks %}
     {% if resources.ngpu_tasks and 'gpu' not in partition and not force %}
         {% raise "Requesting GPUs requires a gpu partition!" %}