Skip to content

Commit 282be20

Browse files
authored
Merge pull request #8210 from vspetrov/master
coll/hcoll: svatterv inplace fix
2 parents 67421c5 + 9fa0015 commit 282be20

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ompi/mca/coll/hcoll/coll_hcoll_ops.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -413,6 +413,10 @@ int mca_coll_hcoll_scatterv(const void* sbuf, const int *scounts, const int *dis
413413
mca_coll_hcoll_module_t *hcoll_module = (mca_coll_hcoll_module_t*)module;
414414
stype = ompi_dtype_2_hcoll_dtype(sdtype, NO_DERIVED);
415415
rtype = ompi_dtype_2_hcoll_dtype(rdtype, NO_DERIVED);
416+
if (rbuf == MPI_IN_PLACE) {
417+
assert(root == comm->c_my_rank);
418+
rtype = stype;
419+
}
416420
if (OPAL_UNLIKELY(HCOL_DTE_IS_ZERO(stype) || HCOL_DTE_IS_ZERO(rtype))) {
417421
/*If we are here then datatype is not simple predefined datatype */
418422
/*In future we need to add more complex mapping to the dte_data_representation_t */

0 commit comments

Comments
 (0)