fix materialized_arrays perf

inducer · Oct 14, 2024 · e8b5806 · e8b5806
1 parent 77f1bbd
commit e8b5806
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py
@@ -844,7 +844,7 @@ def find_distributed_partition(
     # processing below.
     materialized_arrays = {a: None
                            for a in materialized_arrays_collector.materialized_arrays
-                           if a not in received_arrays | sent_arrays}
+                           if a not in received_arrays and a not in sent_arrays}
 
     # "mso" for "materialized/sent/output"
     output_arrays = dict.fromkeys(outputs._data.values())