Update Python to coll

Author: pentschev

python/rapidsmpf/rapidsmpf/CMakeLists.txt

@@ -21,7 +21,7 @@ rapids_cython_create_modules(
 )
 
 add_subdirectory(_detail)
-add_subdirectory(allgather)
+add_subdirectory(coll)
 add_subdirectory(memory)
 add_subdirectory(communicator)
 add_subdirectory(integrations/cudf)

python/rapidsmpf/rapidsmpf/allgather/CMakeLists.txt renamed to python/rapidsmpf/rapidsmpf/coll/CMakeLists.txt

@@ -16,16 +16,16 @@ from rapidsmpf.progress_thread cimport cpp_ProgressThread
 from rapidsmpf.statistics cimport cpp_Statistics
 
 
-cdef extern from "<rapidsmpf/allgather/allgather.hpp>" namespace \
-        "rapidsmpf::allgather::AllGather" nogil:
+cdef extern from "<rapidsmpf/coll/allgather.hpp>" namespace \
+        "rapidsmpf::coll::AllGather" nogil:
     cpdef enum class Ordered(bool):
         NO
         YES
 
-cdef extern from "<rapidsmpf/allgather/allgather.hpp>" nogil:
+cdef extern from "<rapidsmpf/coll/allgather.hpp>" nogil:
     ctypedef int64_t milliseconds_t "std::chrono::milliseconds"
 
-    cdef cppclass cpp_AllGather "rapidsmpf::allgather::AllGather":
+    cdef cppclass cpp_AllGather "rapidsmpf::coll::AllGather":
         cpp_AllGather(
             shared_ptr[cpp_Communicator] comm,
             shared_ptr[cpp_ProgressThread] progress_thread,

python/rapidsmpf/rapidsmpf/allgather/__init__.pxd renamed to python/rapidsmpf/rapidsmpf/coll/__init__.pxd

@@ -8,8 +8,7 @@ from libcpp.memory cimport make_unique
 from libcpp.utility cimport move
 from libcpp.vector cimport vector
 
-from rapidsmpf.allgather.allgather cimport (Ordered, cpp_AllGather,
-                                            milliseconds_t)
+from rapidsmpf.coll.allgather cimport Ordered, cpp_AllGather, milliseconds_t
 from rapidsmpf.communicator.communicator cimport Communicator
 from rapidsmpf.memory.buffer_resource cimport (BufferResource,
                                                cpp_BufferResource)
@@ -23,7 +22,7 @@ cdef class AllGather:
     """
     AllGather communication service for distributed operations.
 
-    The `rapidsmpf.allgather.AllGather` class provides a communication service
+    The `rapidsmpf.coll.AllGather` class provides a communication service
     where each rank contributes data and all ranks receive all inputs from all ranks.
 
     The implementation uses a ring broadcast algorithm where each rank receives a

python/rapidsmpf/rapidsmpf/allgather/__init__.py renamed to python/rapidsmpf/rapidsmpf/coll/__init__.py

@@ -12,7 +12,7 @@
 import cudf
 from pylibcudf.contiguous_split import pack
 
-from rapidsmpf.allgather import AllGather
+from rapidsmpf.coll import AllGather
 from rapidsmpf.integrations.cudf.partition import unpack_and_concat
 from rapidsmpf.memory.buffer_resource import BufferResource
 from rapidsmpf.memory.packed_data import PackedData