From f7a9f47df6d1b743ee29218f775ad780f42f951c Mon Sep 17 00:00:00 2001
From: Mike Campbell <mtcampbe@illinois.edu>
Date: Wed, 2 Mar 2022 18:41:34 -0600
Subject: [PATCH] Get_attr from world
It is OK to use COMM_WORLD here, and this continues to work even for array contexts that don't have the mpi_communicator attribute. This fixes many failing tests in MIRGE-Com.
---
grudge/trace_pair.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/grudge/trace_pair.py b/grudge/trace_pair.py
index 3e627849..872028d7 100644
--- a/grudge/trace_pair.py
+++ b/grudge/trace_pair.py
@@ -505,7 +505,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 = actx.mpi_communicator.Get_attr(MPI.TAG_UB)
+ tag_ub = MPI.COMM_WORLD.Get_attr(MPI.TAG_UB)
key_builder = _TagKeyBuilder()
digest = key_builder(comm_tag)
num_tag = sum(ord(ch) << i for i, ch in enumerate(digest)) % tag_ub
--
GitLab