Skip to content

Commit 006c4e9

Browse files
committed
coll/han: fix the count of alreduce t3 task's upper bcast operation
The upper bcast operaiton is applied to next seg, therefore the count can be last_seg_count. Prior to this patch, seg_count is use unconditionally. This patch used last_seg_count if next seg is last seg. Signed-off-by: Wei Zhang <wzam@amazon.com>
1 parent 0d7c8c3 commit 006c4e9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ompi/mca/coll/han/coll_han_allreduce.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ int mca_coll_han_allreduce_t3_task(void *task_args)
354354
if (t->cur_seg == t->num_segments - 2 && t->last_seg_count != t->seg_count) {
355355
tmp_count = t->last_seg_count;
356356
}
357-
t->up_comm->c_coll->coll_ibcast((char *) t->rbuf + extent * t->seg_count, t->seg_count,
357+
t->up_comm->c_coll->coll_ibcast((char *) t->rbuf + extent * t->seg_count, tmp_count,
358358
t->dtype, t->root_up_rank, t->up_comm, &(reqs[0]),
359359
t->up_comm->c_coll->coll_ibcast_module);
360360
req_count++;

0 commit comments

Comments
 (0)