Refactor RMAT

cpp/CMakeLists.txt

@@ -192,7 +192,6 @@ set(CUGRAPH_SOURCES
     src/community/detail/mis_mg.cu
     src/detail/utility_wrappers.cu
     src/structure/graph_view_mg.cu
-    src/utilities/cython.cu
     src/utilities/path_retrieval.cu
     src/structure/legacy/graph.cu
     src/linear_assignment/legacy/hungarian.cu

python/cugraph/CMakeLists.txt

@@ -87,7 +87,6 @@ add_subdirectory(cugraph/community)
 add_subdirectory(cugraph/components)
 add_subdirectory(cugraph/dask/comms)
 add_subdirectory(cugraph/dask/structure)
-add_subdirectory(cugraph/generators)
 add_subdirectory(cugraph/internals)
 add_subdirectory(cugraph/layout)
 add_subdirectory(cugraph/linear_assignment)

python/cugraph/cugraph/dask/comms/comms.pxd

@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2022, NVIDIA CORPORATION.
+# Copyright (c) 2020-2023, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -18,8 +18,6 @@
 
 from pylibraft.common.handle cimport *
 
-
-cdef extern from "cugraph/utilities/cython.hpp" namespace "cugraph::cython":
-
-   cdef void init_subcomms(handle_t &handle,
-                           size_t row_comm_size)
+cdef extern from "cugraph/partition_manager.hpp" namespace "cugraph::partition_manager":
+   cdef void init_subcomm(handle_t &handle,
+                          size_t row_comm_size)

python/cugraph/cugraph/dask/comms/comms_wrapper.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2022, NVIDIA CORPORATION.
+# Copyright (c) 2020-2023, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -18,10 +18,10 @@
 
 
 from pylibraft.common.handle cimport *
-from cugraph.dask.comms.comms cimport init_subcomms as c_init_subcomms
+from cugraph.dask.comms.comms cimport init_subcomm as c_init_subcomm
 
 
 def init_subcomms(handle, row_comm_size):
     cdef size_t handle_size_t = <size_t>handle.getHandle()
     handle_ = <handle_t*>handle_size_t
-    c_init_subcomms(handle_[0], row_comm_size)
+    c_init_subcomm(handle_[0], row_comm_size)