Skip to content

Commit 625fa2b

Browse files
authored
Merge pull request #9712 from vspetrov/v5.0.x_coll_ucc_dt
V5.0.x: coll/ucc: adjust to UCC API change for datatype
2 parents 78b565d + c4b8a31 commit 625fa2b

File tree

3 files changed

+6
-13
lines changed

3 files changed

+6
-13
lines changed

ompi/mca/coll/ucc/coll_ucc_allreduce.c

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ static inline ucc_status_t mca_coll_ucc_allreduce_init(const void *sbuf, void *r
3131
goto fallback;
3232
}
3333
ucc_coll_args_t coll = {
34-
.mask = UCC_COLL_ARGS_FIELD_PREDEFINED_REDUCTIONS,
34+
.mask = 0,
3535
.coll_type = UCC_COLL_TYPE_ALLREDUCE,
3636
.src.info = {
3737
.buffer = (void*)sbuf,
@@ -45,9 +45,7 @@ static inline ucc_status_t mca_coll_ucc_allreduce_init(const void *sbuf, void *r
4545
.datatype = ucc_dt,
4646
.mem_type = UCC_MEMORY_TYPE_UNKNOWN
4747
},
48-
.reduce = {
49-
.predefined_op = ucc_op,
50-
},
48+
.op = ucc_op,
5149
};
5250
if (MPI_IN_PLACE == sbuf) {
5351
coll.mask |= UCC_COLL_ARGS_FIELD_FLAGS;

ompi/mca/coll/ucc/coll_ucc_reduce.c

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
/**
32
* Copyright (c) 2021 Mellanox Technologies. All rights reserved.
43
* $COPYRIGHT$
@@ -32,7 +31,7 @@ static inline ucc_status_t mca_coll_ucc_reduce_init(const void *sbuf, void *rbuf
3231
goto fallback;
3332
}
3433
ucc_coll_args_t coll = {
35-
.mask = UCC_COLL_ARGS_FIELD_PREDEFINED_REDUCTIONS,
34+
.mask = 0,
3635
.coll_type = UCC_COLL_TYPE_REDUCE,
3736
.root = root,
3837
.src.info = {
@@ -47,9 +46,7 @@ static inline ucc_status_t mca_coll_ucc_reduce_init(const void *sbuf, void *rbuf
4746
.datatype = ucc_dt,
4847
.mem_type = UCC_MEMORY_TYPE_UNKNOWN
4948
},
50-
.reduce = {
51-
.predefined_op = ucc_op,
52-
},
49+
.op = ucc_op,
5350
};
5451
if (MPI_IN_PLACE == sbuf) {
5552
coll.mask |= UCC_COLL_ARGS_FIELD_FLAGS;

oshmem/mca/scoll/ucc/scoll_ucc_reduce.c

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ static inline ucc_status_t mca_scoll_ucc_reduce_init(const void *sbuf, void *rbu
3333
}
3434

3535
ucc_coll_args_t coll = {
36-
.mask = UCC_COLL_ARGS_FIELD_PREDEFINED_REDUCTIONS,
36+
.mask = 0,
3737
.coll_type = UCC_COLL_TYPE_ALLREDUCE,
3838
.src.info = {
3939
.buffer = (void *)sbuf,
@@ -47,9 +47,7 @@ static inline ucc_status_t mca_scoll_ucc_reduce_init(const void *sbuf, void *rbu
4747
.datatype = ucc_dt,
4848
.mem_type = UCC_MEMORY_TYPE_UNKNOWN
4949
},
50-
.reduce = {
51-
.predefined_op = ucc_op,
52-
},
50+
.op = ucc_op,
5351
};
5452

5553
if (sbuf == rbuf) {

0 commit comments

Comments
 (0)