Skip to content

Commit fdc618c

Browse files
dalcinlabouteiller
andcommitted
ulfm: Support non-ft shrink for intercommunicators
Co-authored-by: Aurelien Bouteiller <[email protected]> Signed-off-by: Lisandro Dalcin <[email protected]>
1 parent cba47b6 commit fdc618c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

ompi/communicator/comm_cid.c

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1578,14 +1578,17 @@ static int ompi_comm_ft_allreduce_intra_nb(int *inbuf, int *outbuf, int count,
15781578
static int ompi_comm_ft_allreduce_inter_nb(int *inbuf, int *outbuf, int count,
15791579
struct ompi_op_t *op, ompi_comm_cid_context_t *cid_context,
15801580
ompi_request_t **req) {
1581-
return MPI_ERR_UNSUPPORTED_OPERATION;
1581+
//TODO: CID_INTER_FT needs an implementation, using the non-ft for now...
1582+
int rc = ompi_comm_allreduce_inter_nb(inbuf, outbuf, count, op, cid_context, req);
1583+
return (rc == MPI_ERR_REVOKED || rc == MPI_ERR_PROC_FAILED) ? MPI_ERR_UNSUPPORTED_OPERATION: rc;
15821584
}
15831585

15841586
static int ompi_comm_ft_allreduce_intra_pmix_nb(int *inbuf, int *outbuf, int count,
15851587
struct ompi_op_t *op, ompi_comm_cid_context_t *cid_context,
15861588
ompi_request_t **req) {
15871589
//TODO: CID_INTRA_PMIX_FT needs an implementation, using the non-ft for now...
1588-
return ompi_comm_allreduce_intra_pmix_nb(inbuf, outbuf, count, op, cid_context, req);
1590+
int rc = ompi_comm_allreduce_intra_pmix_nb(inbuf, outbuf, count, op, cid_context, req);
1591+
return (rc == MPI_ERR_REVOKED || rc == MPI_ERR_PROC_FAILED) ? MPI_ERR_UNSUPPORTED_OPERATION: rc;
15891592
}
15901593

15911594
#endif /* OPAL_ENABLE_FT_MPI */

0 commit comments

Comments
 (0)