TL/UCP: Refactor, add alltoall

openucx · Jun 25, 2024 · 53b6b3c · 53b6b3c
1 parent a362467
commit 53b6b3c
Show file tree

Hide file tree

Showing 10 changed files with 606 additions and 490 deletions.
diff --git a/src/components/tl/ucp/Makefile.am b/src/components/tl/ucp/Makefile.am
@@ -29,6 +29,7 @@ alltoall =                       \
 	alltoall/alltoall.c          \
 	alltoall/alltoall_onesided.c \
 	alltoall/alltoall_pairwise.c \
+	alltoall/alltoall_xgvmi.c    \
 	alltoall/alltoall_bruck.c
 
 alltoallv =                        \

diff --git a/src/components/tl/ucp/allgather/allgather.h b/src/components/tl/ucp/allgather/allgather.h
@@ -76,9 +76,7 @@ ucc_status_t ucc_tl_ucp_allgather_sparbit_init(ucc_base_coll_args_t *coll_args,
                                                 ucc_coll_task_t     **task_h);
 
 /* XGVMI */
-ucc_status_t ucc_tl_ucp_allgather_xgvmi_init(ucc_base_coll_args_t *coll_args,
-                                                ucc_base_team_t      *team,
-                                                ucc_coll_task_t     **task_h);
+void ucc_tl_ucp_dpu_xgvmi_rdma_progress_allgather(ucc_coll_task_t *coll_task);
 
 /* Uses allgather_kn_radix from config */
 ucc_status_t ucc_tl_ucp_allgather_knomial_init(ucc_base_coll_args_t *coll_args,