From 50993996154d5857c0de329a9a21bbbfafb36c67 Mon Sep 17 00:00:00 2001
From: Brandon Butler <butlerbr@umich.edu>
Date: Wed, 8 Nov 2023 10:36:23 -0500
Subject: [PATCH] fix: Raise error on non-homogeneous MPI requests

---
 flow/templates/umich-greatlakes.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/flow/templates/umich-greatlakes.sh b/flow/templates/umich-greatlakes.sh
index 60c5838cf..efde67d3f 100644
--- 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!" %}