compiled

openucx · Dec 18, 2024 · 7453381 · 7453381
1 parent 5ba0039
commit 7453381
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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,