From 0b61c571c27304c05ed9e4c7dc24eb442b938be8 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Wed, 14 Dec 2022 15:57:02 +0100
Subject: [PATCH] Pytato+distributed: Verify DAG before partitioning

---
 grudge/array_context.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/grudge/array_context.py b/grudge/array_context.py
index 541c83b34..c17d684b9 100644
--- a/grudge/array_context.py
+++ b/grudge/array_context.py
@@ -194,6 +194,13 @@ def _dag_to_compiled_func(self, dict_of_named_arrays,
 
         from pytools import ProcessLogger
 
+        if __debug__:
+            # pylint-ignore-reason:
+            # '_BasePytatoArrayContext' has no 'mpi_communicator' member
+            pt.verify_distributed_dag_pre_partition(
+                self.actx.mpi_communicator,  # pylint: disable=no-member
+                dict_of_named_arrays)
+
         self.actx._compile_trace_callback(self.f, "pre_deduplicate_data_wrappers",
                 dict_of_named_arrays)