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,