use actx mpi_comm

Co-authored-by: Andreas Klöckner <[email protected]>
inducer · Mar 2, 2022 · 2caeff5 · 2caeff5
1 parent 42b4fd3
commit 2caeff5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/grudge/trace_pair.py b/grudge/trace_pair.py
@@ -498,7 +498,7 @@ def cross_rank_trace_pairs(
                 # - https://github.com/illinois-ceesd/mirgecom/issues/617#issuecomment-1057082716  # noqa
                 # - https://github.com/inducer/grudge/pull/222
                 from mpi4py import MPI
-                tag_ub = MPI.COMM_WORLD.Get_attr(MPI.TAG_UB)
+                tag_ub = actx.mpi_communicator.Get_attr(MPI.TAG_UB)
                 num_tag = hash(comm_tag) % tag_ub
 
                 from warnings import warn