Re-enable this snippet with fix - testing multivols

illinois-ceesd · Dec 12, 2024 · da25b8d · da25b8d
1 parent c9eb887
commit da25b8d
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/grudge/trace_pair.py b/grudge/trace_pair.py
@@ -916,12 +916,15 @@ def cross_rank_trace_pairs(
     # --------
 
     # Disabling this bit due to above error
-    # if isinstance(ary, Number):
-    #    # NOTE: Assumed that the same number is passed on every rank
-    #    return [TracePair(
-    #            volume_dd.trace(BTAG_PARTITION(remote_rank)),
-    #           interior=ary, exterior=ary)
-    #        for remote_rank in connected_parts(dcoll, volume_dd=volume_dd)]
+    if isinstance(ary, Number):
+        # NOTE: Assumed that the same number is passed on every rank
+        voltag = volume_dd.domain_tag.tag
+        return [TracePair(
+            volume_dd.trace(BTAG_PARTITION(remote_rank)),
+            interior=ary, exterior=ary)
+                for remote_rank in connected_parts(dcoll,
+                                                   self_volume_tag=voltag,
+                                                   other_volume_tag=voltag)]
 
     rank = dcoll.mpi_communicator.Get_rank()