Skip to content

Commit 61c9403

Browse files
committed
communicator: add assertion on OMPI_COMM_DISJOINT flag
This patch introduces assertions to verify that sub-communicators are created with the expected OMPI_COMM_DISJOINT* flags. Signed-off-by: Wenduo Wang <wenduwan@amazon.com>
1 parent a1cf411 commit 61c9403

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ompi/mca/coll/han/coll_han_subcomms.c

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,7 @@ int mca_coll_han_comm_create(struct ompi_communicator_t *comm,
282282
opal_info_set(&comm_info, "ompi_comm_coll_preference", "tuned,^han");
283283
ompi_comm_split_type(comm, MPI_COMM_TYPE_SHARED, 0,
284284
&comm_info, &(low_comms[0]));
285+
assert(OMPI_COMM_IS_DISJOINT_SET(low_comms[0]) && !OMPI_COMM_IS_DISJOINT(low_comms[0]));
285286

286287
/*
287288
* Get my local rank and the local size
@@ -296,6 +297,7 @@ int mca_coll_han_comm_create(struct ompi_communicator_t *comm,
296297
opal_info_set(&comm_info, "ompi_comm_coll_preference", "sm,^han");
297298
ompi_comm_split_type(comm, MPI_COMM_TYPE_SHARED, 0,
298299
&comm_info, &(low_comms[1]));
300+
assert(OMPI_COMM_IS_DISJOINT_SET(low_comms[1]) && !OMPI_COMM_IS_DISJOINT(low_comms[1]));
299301

300302
/*
301303
* Upgrade libnbc module priority to set up up_comms[0] with libnbc module
@@ -304,15 +306,16 @@ int mca_coll_han_comm_create(struct ompi_communicator_t *comm,
304306
*/
305307
opal_info_set(&comm_info, "ompi_comm_coll_preference", "libnbc,^han");
306308
ompi_comm_split_with_info(comm, low_rank, w_rank, &comm_info, &(up_comms[0]), false);
307-
308309
up_rank = ompi_comm_rank(up_comms[0]);
310+
assert(OMPI_COMM_IS_DISJOINT_SET(up_comms[0]) && OMPI_COMM_IS_DISJOINT(up_comms[0]));
309311

310312
/*
311313
* Upgrade adapt module priority to set up up_comms[0] with adapt module
312314
* This sub-communicator contains one process per node.
313315
*/
314316
opal_info_set(&comm_info, "ompi_comm_coll_preference", "adapt,^han");
315317
ompi_comm_split_with_info(comm, low_rank, w_rank, &comm_info, &(up_comms[1]), false);
318+
assert(OMPI_COMM_IS_DISJOINT_SET(up_comms[1]) && OMPI_COMM_IS_DISJOINT(up_comms[1]));
316319

317320
/*
318321
* Set my virtual rank number.
@@ -350,5 +353,3 @@ int mca_coll_han_comm_create(struct ompi_communicator_t *comm,
350353
OBJ_DESTRUCT(&comm_info);
351354
return OMPI_SUCCESS;
352355
}
353-
354-

0 commit comments

Comments
 (0)