Update corresponding inline comment.

rapidsai · Feb 12, 2025 · 02be3c3 · 02be3c3
1 parent b218c94
commit 02be3c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/include/raft/cluster/detail/kmeans_balanced.cuh b/cpp/include/raft/cluster/detail/kmeans_balanced.cuh
@@ -967,7 +967,8 @@ void build_hierarchical(const raft::resources& handle,
   IdxT n_mesoclusters = std::min(n_clusters, static_cast<IdxT>(std::sqrt(n_clusters) + 0.5));
   RAFT_LOG_DEBUG("build_hierarchical: n_mesoclusters: %u", n_mesoclusters);
 
-  // TODO: Remove the explicit managed memory- we shouldn't be creating this on the user's behalf.
+  // Need to use explicit managed_memory here since corresponding allocations
+  // must be both host and device accessible.
   rmm::mr::managed_memory_resource managed_memory;
   rmm::device_async_resource_ref device_memory = resource::get_large_workspace_resource(handle);
   auto [max_minibatch_size, mem_per_row] =