Merge pull request #9352 from vspetrov/master

janjust · web-flow · commit 7c1bd89d80af · 2021-09-07T12:49:14.000-05:00
coll/ucc: fill count/dtype in dst.info for allreduce/reduce
diff --git a/ompi/mca/coll/ucc/coll_ucc_allreduce.c b/ompi/mca/coll/ucc/coll_ucc_allreduce.c
@@ -41,6 +41,8 @@ static inline ucc_status_t mca_coll_ucc_allreduce_init(const void *sbuf, void *r
         },
         .dst.info = {
             .buffer   = rbuf,
+            .count    = count,
+            .datatype = ucc_dt,
             .mem_type = UCC_MEMORY_TYPE_UNKNOWN
         },
         .reduce = {
diff --git a/ompi/mca/coll/ucc/coll_ucc_reduce.c b/ompi/mca/coll/ucc/coll_ucc_reduce.c
@@ -43,6 +43,8 @@ static inline ucc_status_t mca_coll_ucc_reduce_init(const void *sbuf, void *rbuf
         },
         .dst.info = {
             .buffer   = rbuf,
+            .count    = count,
+            .datatype = ucc_dt,
             .mem_type = UCC_MEMORY_TYPE_UNKNOWN
         },
         .reduce = {
