diff --git a/src/components/tl/ucp/allgather/allgather.c b/src/components/tl/ucp/allgather/allgather.c index f1c5a0477b..61a1c0b2d9 100644 --- a/src/components/tl/ucp/allgather/allgather.c +++ b/src/components/tl/ucp/allgather/allgather.c @@ -61,8 +61,7 @@ char *ucc_tl_ucp_allgather_score_str_get(ucc_tl_ucp_team_t *team) ucc_status_t loopback_self_copy(void* rbuf, void* sbuf, size_t data_size, ucc_memory_type_t rmem, ucc_memory_type_t smem, - ucc_tl_ucp_task_t *task, ucc_tl_ucp_team_t *team, ucc_rank_t rank) -{ + ucc_rank_t rank, ucc_tl_ucp_team_t *team, ucc_tl_ucp_task_t *task) { ucc_status_t status; status = ucc_tl_ucp_send_nb(sbuf, data_size, smem, rank, team, task); if (UCC_OK != status) { diff --git a/src/components/tl/ucp/allgather/allgather.h b/src/components/tl/ucp/allgather/allgather.h index 0d192e1060..9b98b259ce 100644 --- a/src/components/tl/ucp/allgather/allgather.h +++ b/src/components/tl/ucp/allgather/allgather.h @@ -7,6 +7,7 @@ #define ALLGATHER_H_ #include "../tl_ucp.h" #include "../tl_ucp_coll.h" +#include "tl_ucp_sendrecv.h" enum { UCC_TL_UCP_ALLGATHER_ALG_KNOMIAL, @@ -39,7 +40,7 @@ static inline int ucc_tl_ucp_allgather_alg_from_str(const char *str) ucc_status_t ucc_tl_ucp_allgather_init(ucc_tl_ucp_task_t *task); ucc_status_t loopback_self_copy(void* rbuf, void* sbuf, size_t data_size, ucc_memory_type_t rmem, ucc_memory_type_t smem, - ucc_tl_ucp_task_t *task, ucc_tl_ucp_team_t *team, ucc_rank_t rank) + ucc_rank_t rank, ucc_tl_ucp_team_t *team, ucc_tl_ucp_task_t *task); /* Ring */ ucc_status_t ucc_tl_ucp_allgather_ring_init(ucc_base_coll_args_t *coll_args,