Distributed partitioning without partitioning

inducer · Dec 6, 2022 · 94b7d19 · 94b7d19
1 parent 2dd57be
commit 94b7d19
Show file tree

Hide file tree

Showing 5 changed files with 212 additions and 562 deletions.
diff --git a/examples/distributed.py b/examples/distributed.py
@@ -39,14 +39,17 @@ def main():
         from pytato.visualization import show_dot_graph
         show_dot_graph(distributed_parts)
 
-    # Sanity check
-    from pytato.visualization import get_dot_graph_from_partition
-    get_dot_graph_from_partition(distributed_parts)
+    if 0:
+        # Sanity check
+        from pytato.visualization import get_dot_graph_from_partition
+        get_dot_graph_from_partition(distributed_parts)
 
     # Execute the distributed partition
     ctx = cl.create_some_context()
     queue = cl.CommandQueue(ctx)
 
+    pt.verify_distributed_partition(comm, distributed_parts)
+
     context = execute_distributed_partition(distributed_parts, prg_per_partition,
                                              queue, comm)