Merge pull request #6516 from hzhou/2305_ch3_split

ch3: fix MPIR_Get_node_id for intercomm Approved-by: Ken Raffenetti
pmodels · May 15, 2023 · bba85c7 · bba85c7
2 parents 0f72280 + 1029595
commit bba85c7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/mpid/ch3/src/mpid_vc.c b/src/mpid/ch3/src/mpid_vc.c
@@ -804,7 +804,11 @@ char MPIU_hostname[MAX_HOSTNAME_LEN] = "_UNKNOWN_"; /* '_' is an illegal char fo
 
 int MPID_Get_node_id(MPIR_Comm *comm, int rank, int *id_p)
 {
-    *id_p = comm->dev.vcrt->vcr_table[rank]->node_id;
+    if (comm->comm_kind == MPIR_COMM_KIND__INTRACOMM) {
+        *id_p = comm->dev.vcrt->vcr_table[rank]->node_id;
+    } else {
+        *id_p = comm->dev.local_vcrt->vcr_table[rank]->node_id;
+    }
     return MPI_SUCCESS;
 }