Skip to content

Commit 5426b35

Browse files
committed
rm debug statements
1 parent 1dd9039 commit 5426b35

1 file changed

Lines changed: 0 additions & 9 deletions

File tree

cpp/src/cluster/detail/mst.cuh

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)