fixup! Log UMAP arrays at trace verbosity level.

rapidsai · Jan 29, 2025 · ec50cbf · ec50cbf
1 parent d7efec6
commit ec50cbf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cpp/src/umap/fuzzy_simpl_set/naive.cuh b/cpp/src/umap/fuzzy_simpl_set/naive.cuh
@@ -342,11 +342,11 @@ void launcher(int n,
                                                                               n_neighbors);
   RAFT_CUDA_TRY(cudaPeekAtLastError());
 
-  if (ML::default_logger().should_log(ML::level_enum::debug)) {
-    CUML_LOG_DEBUG("Compute Membership Strength");
+  if (ML::default_logger().should_log(ML::level_enum::trace)) {
+    CUML_LOG_TRACE("Compute Membership Strength");
     std::stringstream ss;
     ss << in;
-    CUML_LOG_DEBUG(ss.str().c_str());
+    CUML_LOG_TRACE(ss.str().c_str());
   }
 
   /**