CL/HIER: Add allgatherv

nsarka · Dec 2, 2024 · 198da51 · 198da51
1 parent 4c5ac3e
commit 198da51
Show file tree

Hide file tree

Showing 15 changed files with 650 additions and 60 deletions.
diff --git a/src/components/cl/hier/Makefile.am b/src/components/cl/hier/Makefile.am
@@ -1,49 +1,56 @@
 #
-# Copyright (c) 2020-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 
+allgatherv =                         \
+    allgatherv/unpack.h              \
+    allgatherv/unpack.c              \
+    allgatherv/allgatherv.h          \
+    allgatherv/allgatherv.c
+
 allreduce =                          \
-	allreduce/allreduce.h            \
-	allreduce/allreduce.c            \
-	allreduce/allreduce_rab.c        \
-	allreduce/allreduce_split_rail.c
-
-alltoallv =                   \
-	alltoallv/alltoallv.h     \
-	alltoallv/alltoallv.c
-
-alltoall =                    \
-	alltoall/alltoall.h       \
-	alltoall/alltoall.c
-
-barrier =                     \
-	barrier/barrier.h         \
-	barrier/barrier.c
-
-bcast =                       \
-	bcast/bcast.h             \
-	bcast/bcast.c             \
-	bcast/bcast_2step.c
-
-reduce =                      \
-	reduce/reduce.h           \
-	reduce/reduce.c           \
-	reduce/reduce_2step.c
-
-sources =             \
-	cl_hier.h         \
-	cl_hier.c         \
-	cl_hier_lib.c     \
-	cl_hier_context.c \
-	cl_hier_team.c    \
-	cl_hier_coll.c    \
-	cl_hier_coll.h    \
-	$(allreduce)      \
-	$(alltoallv)      \
-	$(alltoall)       \
-	$(barrier)        \
-	$(bcast)          \
-	$(reduce)
+    allreduce/allreduce.h            \
+    allreduce/allreduce.c            \
+    allreduce/allreduce_rab.c        \
+    allreduce/allreduce_split_rail.c
+
+alltoallv =                          \
+    alltoallv/alltoallv.h            \
+    alltoallv/alltoallv.c
+
+alltoall =                           \
+    alltoall/alltoall.h              \
+    alltoall/alltoall.c
+
+barrier =                            \
+    barrier/barrier.h                \
+    barrier/barrier.c
+
+bcast =                              \
+    bcast/bcast.h                    \
+    bcast/bcast.c                    \
+    bcast/bcast_2step.c
+
+reduce =                             \
+    reduce/reduce.h                  \
+    reduce/reduce.c                  \
+    reduce/reduce_2step.c
+
+sources =                            \
+    cl_hier.h                        \
+    cl_hier.c                        \
+    cl_hier_lib.c                    \
+    cl_hier_context.c                \
+    cl_hier_team.c                   \
+    cl_hier_coll.c                   \
+    cl_hier_coll.h                   \
+    $(allgatherv)                    \
+    $(allreduce)                     \
+    $(alltoallv)                     \
+    $(alltoall)                      \
+    $(barrier)                       \
+    $(bcast)                         \
+    $(reduce)
 
 module_LTLIBRARIES         = libucc_cl_hier.la
 libucc_cl_hier_la_SOURCES  = $(sources)