@@ -159,29 +159,22 @@ void build_sorted_mst(
159159 cuvs::distance::DistanceType metric = cuvs::distance::DistanceType::L2SqrtExpanded,
160160 int max_iter = 10 )
161161{
162- RAFT_LOG_INFO (" nnz %zu" , nnz);
163162 auto stream = raft::resource::get_cuda_stream (handle);
164- RAFT_LOG_INFO (" initialize mst_coo" );
165163
166164 // We want to have MST initialize colors on first call.
167165 auto mst_coo = raft::sparse::solver::mst<value_idx, value_idx, value_t , double >(
168166 handle, indptr, indices, pw_dists, (value_idx)m, nnz, color, stream, false , true );
169167
170- RAFT_LOG_INFO (" initialized mst_coo" );
171-
172168 int iters = 1 ;
173169 int n_components = cuvs::sparse::neighbors::get_n_components (color, m, stream);
174170
175- RAFT_LOG_INFO (" doing iterations" );
176171 while (n_components > 1 && iters < max_iter) {
177172 connect_knn_graph<value_idx, value_t >(handle, X, mst_coo, m, n, color, reduction_op);
178173
179174 iters++;
180175
181176 n_components = cuvs::sparse::neighbors::get_n_components (color, m, stream);
182177 }
183- cudaDeviceSynchronize ();
184- RAFT_LOG_INFO (" done iterations" );
185178
186179 /* *
187180 * The `max_iter` argument was introduced only to prevent the potential for an infinite loop.
@@ -206,8 +199,6 @@ void build_sorted_mst(
206199 raft::sparse::op::coo_sort_by_weight (
207200 mst_coo.src .data (), mst_coo.dst .data (), mst_coo.weights .data (), mst_coo.n_edges , stream);
208201
209- RAFT_LOG_INFO (" mst_coo.n_edges %d" , mst_coo.n_edges );
210-
211202 raft::copy_async (mst_src, mst_coo.src .data (), mst_coo.n_edges , stream);
212203 raft::copy_async (mst_dst, mst_coo.dst .data (), mst_coo.n_edges , stream);
213204 raft::copy_async (mst_weight, mst_coo.weights .data (), mst_coo.n_edges , stream);
0 commit comments