TL/UCP: Add xgvmi allgather

openucx · Jun 24, 2024 · ecbd9e1 · ecbd9e1
1 parent a360c7a
commit ecbd9e1
Show file tree

Hide file tree

Showing 6 changed files with 573 additions and 1 deletion.
diff --git a/src/components/tl/ucp/Makefile.am b/src/components/tl/ucp/Makefile.am
@@ -16,7 +16,8 @@ allgather = \
  allgather/allgather_neighbor.c \
  allgather/allgather_bruck.c \
  allgather/allgather_sparbit.c \
- allgather/allgather_knomial.c
+ allgather/allgather_knomial.c \
+ allgather/allgather_xgvmi.c
 
 allgatherv = \
  allgatherv/allgatherv.h \

diff --git a/src/components/tl/ucp/allgather/allgather.c b/src/components/tl/ucp/allgather/allgather.c
@@ -31,6 +31,10 @@ ucc_base_coll_alg_info_t
  {.id = UCC_TL_UCP_ALLGATHER_ALG_SPARBIT,
  .name = "sparbit",
  .desc = "O(log(N)) SPARBIT algorithm"},
+ [UCC_TL_UCP_ALLGATHER_ALG_XGVMI] =
+ {.id = UCC_TL_UCP_ALLGATHER_ALG_XGVMI,
+ .name = "xgvmi",
+ .desc = "Offloaded xgvmi algorithm"},
  [UCC_TL_UCP_ALLGATHER_ALG_LAST] = {
  .id = 0, .name = NULL, .desc = NULL}};
 

diff --git a/src/components/tl/ucp/allgather/allgather.h b/src/components/tl/ucp/allgather/allgather.h
@@ -14,6 +14,7 @@ enum {
  UCC_TL_UCP_ALLGATHER_ALG_NEIGHBOR,
  UCC_TL_UCP_ALLGATHER_ALG_BRUCK,
  UCC_TL_UCP_ALLGATHER_ALG_SPARBIT,
+ UCC_TL_UCP_ALLGATHER_ALG_XGVMI,
  UCC_TL_UCP_ALLGATHER_ALG_LAST
 };
 
@@ -74,6 +75,11 @@ ucc_status_t ucc_tl_ucp_allgather_sparbit_init(ucc_base_coll_args_t *coll_args,
  ucc_base_team_t *team,
  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);
+
 /* Uses allgather_kn_radix from config */
 ucc_status_t ucc_tl_ucp_allgather_knomial_init(ucc_base_coll_args_t *coll_args,
  ucc_base_team_t *team,