New distributed-memory DAG partitioner

Co-authored-by: Matt Smith <[email protected]>
inducer · Nov 23, 2022 · 0cc2f46 · 0cc2f46
1 parent f282b40
commit 0cc2f46
Show file tree

Hide file tree

Showing 4 changed files with 419 additions and 139 deletions.
diff --git a/examples/distributed.py b/examples/distributed.py
@@ -30,7 +30,7 @@ def main():
 
     # Find the partition
     outputs = pt.DictOfNamedArrays({"out": y})
-    distributed_parts = find_distributed_partition(outputs)
+    distributed_parts = find_distributed_partition(comm, outputs)
     distributed_parts, _ = number_distributed_tags(
             comm, distributed_parts, base_tag=42)
     prg_per_partition = generate_code_for_partition(distributed_parts)